python - 使用 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) 返回每行的最小值。