可以使用 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] 来获取该列的最小值。

NumPy 2D 数组获取指定列最小值方法

原文地址: https://www.cveoy.top/t/topic/mv3X 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录