在numpy数组中检索验证某个条件的第一个元素的索引的最佳方法是什么?
我可以做以下
x = np.array([1, 2, 1, 2, 2, 2, 1, 1])
np.where(x == 2)[0][0]
这会给我数组中第一次出现数字2的索引。
循环整个数组,为x[i] == 2
中的所有i
评估条件range(len(x))
以创建一个布尔掩码,然后传递给np.where
以检索索引True
元素。
不需要循环遍历整个数组,只要满足验证条件的第一个元素,迭代就会停止。
如果不迭代整个数组,有没有一种简单的方法呢?