numpy找大于某个值的索引
时间: 2023-06-19 22:09:56 浏览: 256
Python替换NumPy数组中大于某个值的所有元素实例
可以使用numpy的`where`函数来找到大于某个值的索引。例如,假设有一个numpy数组`arr`,我们想要找到大于5的元素的索引,可以使用以下代码:
```
import numpy as np
arr = np.array([1, 3, 5, 7, 9])
indices = np.where(arr > 5)
print(indices)
```
输出结果为:
```
(array([3, 4]),)
```
这意味着在原数组中,索引为3和4的元素大于5。注意,`where`函数返回的是一个元组,元组中的元素是满足条件的元素的索引。如果要获取索引数组,可以使用`indices[0]`。
阅读全文