在此博客中,我将向您展示如何在数组中搜索特定项目。
搜索
numpy具有一种称为where
的方法,可以在数组中找到一个项目。此方法需要其参数的表达式。让我向您展示如何实施它。
首先,让我们创建一个数组
import numpy as np
arr = np.array([1, 2, 3, 4, 5, 2])
然后,使用方法where
在数组中查找项目的索引并将其存储在变量x
中
x = np.where(arr == 2)
让我们打印x
的值以查看其输出
print(x)
以下是输出
(array([1, 5]),)
从上面的输出中,您可以看到2
在索引1
和5
中发生了两次。
恭喜,这就是您可以在numpy数组中搜索项目的方式。谢谢您的阅读,祝您有美好的一天!