python - 在numpy数组中查找最大元素比较的索引

Question

我有一个数组 a，我想找到 a 中给定值仍然大于的最大元素的位置。

在这个例子中:

a = np.array([0, 50, 5, 52, 60])
v = 55

v 大于的最大元素是 52(索引 3)所以我想返回 3。

numpy 函数 argmax() 不适用于此目的，因为它返回第一个元素。使用 numpy 执行此操作的快速且正确的方法是什么？

Answer 1

您可以结合argmax与 where :

>>> np.nanargmax(np.where(a < v, a, np.nan))
3

np.where替换 v 以上的所有值至 nan在它适用之前nanargmax (在计算中忽略 nan s):

>>> np.where(a < v, a, np.nan)
array([  0.,  50.,   5.,  52.,  nan])