获取第一个元素的索引,验证numpy数组中的条件

时间:2017-04-19 14:49:53

标签: python arrays numpy optimization

在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元素。

不需要循环遍历整个数组,只要满足验证条件的第一个元素,迭代就会停止。

如果不迭代整个数组,有没有一种简单的方法呢?

0 个答案:

没有答案