小编给大家分享一下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()函数有了一定的了解，想了解更多相关知识，欢迎关注亿速云行业资讯频道，感谢各位的阅读！
亿速云「云服务器」，即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘，价格低至29元/月。点击查看>>
免责声明：本站发布的内容（图片、视频和文字）以原创、转载和分享为主，文章观点不代表本网站立场，如果涉及侵权请联系站长邮箱：is@yisu.com进行举报，并提供相关证据，一经查实，将立刻删除涉嫌侵权内容。
原文链接：https://www.py.cn/jishu/jichu/20433.html
