numpy大于某个值的索引

可以使用NumPy中的布尔索引来查找大于某个值的元素的索引。下面是一个例子：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
idx = arr > 3
print(idx)  # [False False False  True  True]

# 可以使用np.where函数查找满足条件的元素的索引
idx = np.where(arr > 3)
print(idx)  # (array([3, 4]),)

上面的代码中，`idx`是一个布尔数组，`True`表示该位置的元素大于3，`False`表示该位置的元素小于等于3。然后，可以使用`np.where`函数查找满足条件的元素的索引。在这个例子中，满足条件的元素是4和5，它们的索引是3和4。`np.where`函数返回一个元组，其中第一个元素是一个数组，包含满足条件的元素的索引。