小编给大家分享一下Python中如何使用numpy.where()函数,希望大家阅读完这篇文章后大所收获,下面让我们一起去探讨吧!
numpy.where(condition [,x,y])函数返回满足给定条件的输入数组中元素的索引。
参数:condition : When True, yieldx, otherwise yield y.x,y: Values from which to choose. x, y and
condition need to be broadcastable to some shape.
返回:out : [ndarray or tuple of ndarrays] If both x and y are specified, the output array contains elements of x where condition is True, and elements from y elsewhere.
If only condition is given, return the tuple condition.nonzero(), the indices where condition is True.
代码1:
# Python program explaining # where() function import numpy as np np.where([[True, False], [True, True]], [[1, 2], [3, 4]], [[5, 6], [7, 8]])
输出:
array([[1, 6], [3, 4]])
代码2:
# Python program explaining # where() function import numpy as np # a is an array of integers. a = np.array([[1, 2, 3], [4, 5, 6]]) print(a) print ('Indices of elements <4') b = np.where(a<4) print(b) print("Elements which are <4") print(a[b])
输出:
[[1 2 3] [4 5 6]] Indices of elements <4 (array([0, 0, 0], dtype=int64), array([0, 1, 2], dtype=int64)) Elements which are <4 array([1, 2, 3])
看完了这篇文章,相信你对Python中如何使用numpy.where()函数有了一定的了解,想了解更多相关知识,欢迎关注亿速云行业资讯频道,感谢各位的阅读!
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。