在python中查找Numpy数组中第一个带条件元素的索引

Question

我试图找到第一个元素的索引大于阈值，像这样：

index = 0
while timeStamps[index] < self.stopCount and index < len(timeStamps):
    index += 1

这可以在一个班轮内完成吗？我发现：

index = next((x for x in timeStamps if x <= self.stopCount), 0)

我不确定这个表达式的作用，它似乎总是返回0。。。有人能指出错误并解释一下表达方式吗？

Answer 1

这一班机可以用

sample_array = np.array([10,11,12,13,21,200,1,2])

# oneliner
print(sum(np.cumsum(arr>threshold)==0))

np。cumsum（样本数组）

Answer 2

我会这样做：

import numpy as np

threshold = 20

sample_array = np.array([10,11,12,13,21,200,1,2])

idx = np.array([np.where(sample_array > threshold)]).min()
print(idx)
#4

Answer 3

另一个选择是使用np.argmax（详见这篇文章）。所以你的代码会变成这样

(timeStamps > self.stopCount).argmax()

需要注意的是，如果从未满足该条件，argmax将返回0。