导语：
本文主要介绍了关于Python中numpy.where()函数的使用的相关知识，希望可以帮到处于编程学习途中的小伙伴
本文教程操作环境：
windows7系统、Python 3.9.1，DELL G3电脑。
1、概念
numpy.where(condition[,x,y]) 函数返回输入数组中满足给定条件的元素的索引。
2、参数
condition
3、返回值
返回out
代码1：
<p><span># Python program explaining
# where() function
import numpy as np
np.where([[True, False], [True, True]],
[[1, 2], [3, 4]], [[5, 6], [7, 8]])<br/></span></p>
输出：
<p><span>array([[1, 6],
[3, 4]])<br/></span></p>
代码2：
<p><span># 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])<br/></span></p>
输出：
<p><span>[[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])<br/></span></p>
以上就是Python中numpy.where()函数的使用方法。
本文为原创文章，版权归知行编程网所有，欢迎分享本文，转载请保留出处！
内容反馈