python找出np.array中大于某个值的
时间: 2024-01-16 08:03:32 浏览: 226
可以使用NumPy中的比较运算符和布尔索引来实现。
例如,假设我们有一个名为arr的NumPy数组,我们想要找到其中大于5的元素,可以使用以下代码:
```python
import numpy as np
arr = np.array([1, 6, 2, 8, 3, 0, 9])
result = arr[arr > 5]
print(result)
```
输出:
```
[6 8 9]
```
其中,`arr > 5` 返回一个布尔类型的NumPy数组,表示arr中每个元素是否大于5。然后,我们可以将这个布尔数组作为索引,从arr中筛选出对应的元素。
相关问题
numpy找出大于某个数的索引
可以使用numpy的where函数来找出大于某个数的索引,示例如下:
```python
import numpy as np
a = np.array([1, 2, 3, 4, 5])
idx = np.where(a > 3)
print(idx)
```
输出结果为:
```
(array([3, 4]),)
```
其中,`np.where(a > 3)` 返回的是一个元组,第一个元素是符合条件的元素的索引。在本例中,大于3的元素索引为3和4。
a = np.array([[1,5],[3,4],[8,6]]) print(a) #取出大于5并以数组的形式输出
要取出数组 `a` 中大于 5 的元素,并以数组的形式输出,可以使用以下代码:
```python
import numpy as np
a = np.array([[1, 5], [3, 4], [8, 6]])
output = a[a > 5]
print(output) # 输出 [8, 6]
```
在这个例子中,我们使用条件索引 `a > 5` 来找出满足条件大于 5 的元素。然后,我们将满足条件的元素存储在一个新的数组 `output` 中,并通过打印 `output` 输出结果 `[8, 6]`。