python - Numpy:如何获得属性为真的最小索引

Question

这个问题已经有答案了:

Numpy first occurrence of value greater than existing value (8 个回答)

已关闭 3 年前。

我有一个一维的 numpy 数组。我想获得属性真实的最大和最小索引。

例如，

A = np.array([0, 3, 2, 4, 3, 6, 1, 0])

我想知道 A 的值的最小索引大于或等于4 。

我能做到

i = 0
while A[i] < 4:
    i += 1
print("smallest index", i)

i = -1
while A[i] <4:
    i -= 1
print("largest index", len(A)+i)

有更好的方法吗？

<小时/>

按照 this answer 中的建议,

np.argmax(A>=4)

返回3 ，这确实是最小的索引。但这并没有给我最大的索引。

Answer 1

你可以尝试类似的事情。根据评论，如果 A 是。

A = np.array([0, 3, 2, 4, 3, 6, 1, 4])

idx_values = np.where(A >= 4)[0]
min_idx, max_idx = idx_values[[0, -1]]

print(idx_values)
# array([3, 5, 7], dtype=int64)

idx_values 返回满足您的条件的所有索引值。然后您可以访问最小和最大的索引位置。

print(min_idx, max_idx)
# (3, 7)