如何查找值大于numpy中某个阈值的数组的索引?

时间:2019-06-26 08:27:12

标签: python

在python(3.5.2)中,我有一个越来越多的值数组,我想找到一个值大于该阈值的索引。

我尝试了以下代码

f = [0.0, 9.7, 19.5, 29.2, 39.0, 48.8, 58.5, 68.3, 78.1, 87.8, 97.6, 107.4, 117.1, 126.9]
index = min(np.argwhere(f>100))

产生错误

TypeError: unorderable types: list() > int()

但是,预期结果为index=11,因为f[11]是数组f的第一个大于100的元素。

如何修复此代码?有更好的方法吗?

1 个答案:

答案 0 :(得分:3)

这可以做到:

np.argwhere(np.array(f)>100).min()

或者:

np.argmax(np.array(f)>100)

请注意,argmax在将条件评估为True的第一个索引处停止。