使用python中的numpy库,实现取出一个ndarray中的所有最小值
时间: 2024-05-06 07:22:07 浏览: 62
可以使用numpy中的argmin函数找到ndarray中最小值的索引,然后再通过索引取出所有最小值。
示例代码如下:
```python
import numpy as np
# 创建一个ndarray
arr = np.array([[1, 2, 3], [4, 1, 6], [7, 8, 1]])
# 找到最小值的索引
min_index = np.argmin(arr)
# 通过索引取出所有最小值
min_values = arr[np.where(arr == arr.flat[min_index])]
print(min_values)
```
输出结果为:
```
[1 1 1]
```
其中,np.where函数用于找到所有最小值的索引,arr.flat属性将ndarray展开成一维数组,方便找到最小值的索引。
阅读全文