使用 numpy 提取每行的最小值

Question

我有一个问题，但在互联网和本网站上都找不到答案。我相信这很容易。假设我有一组 20 个数字，并将它们放在一个 5x4 矩阵中：

numbers = np.arange(20).reshape(5,4) 

这会产生以下矩阵：

[ 0,  1,  2,  3]
[ 4,  5,  6,  7]
[ 8,  9, 10, 11]
[12, 13, 14, 15]
[16, 17, 18, 19]

现在我想要每行的最小值，在本例中为 0,4,8,12,16。但是，我想补充一点，对于我的问题，最小值并不总是在第一列，它可以在矩阵中的随机位置（即每行的第一、第二、第三或第四列）。如果有人能对此有所了解，将不胜感激。

Answer 1

您只需要指定要取最小值的轴。要找到每一行中的最小值，您需要指定轴 1：

>>> numbers.min(axis=1)
array([ 0,  4,  8, 12, 16])

对于二维数组，numbers.min() 查找数组中的单个 最小值，numbers.min(axis=0) 返回每​​一列的最小值，numbers.min(axis=1) 返回每​​一行的最小值。