您可以尝试类似。根据评论,如果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)