在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

我会这样做：

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