在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的元素。
如何修复此代码?有更好的方法吗?
答案 0 :(得分:3)
这可以做到:
np.argwhere(np.array(f)>100).min()
或者:
np.argmax(np.array(f)>100)
请注意,argmax在将条件评估为True
的第一个索引处停止。