NumPy 2D 数组获取指定列最小值方法
可以使用 NumPy 中的 argmin 函数来找到某列的最小值所在的行数,然后再通过索引来获取该列的最小值。
例如,假设有一个二维数组 arr,想要获取第 2 列的最小值:
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
min_row = np.argmin(arr[:, 1])
min_val = arr[min_row, 1]
print(min_val) # 输出 2
这里使用了切片操作 [:, 1] 来获取第 2 列的所有元素,再通过 argmin 函数找到最小值所在的行数,最后使用索引 [min_row, 1] 来获取该列的最小值。
原文地址: https://www.cveoy.top/t/topic/mv3X 著作权归作者所有。请勿转载和采集!