如何在numpy数组中打印特定行/列的最大值？

Question

#input

import numpy as np
arr = np.array([(1,2,3,4),(5,6,7,8),(9,10,11,12)])
print(arr)
print(np.max(arr))
print(np.max(arr,0))
print(np.max(arr,1))

#output

[[ 1  2  3  4]
[ 5  6  7  8]
[ 9 10 11 12]]
 12
[ 9 10 11 12]
[ 4  8 12]

我得到了整个矩阵的最大值，所有行和列的最大值，但是我怎样才能得到某一行的最大值假设我只想打印第 1 行的最大值（即 8 ) 不是每一行。

我该怎么做？

Answer 1

如果

np.max(arr, 1)

给你每行的最大值：

[ 4  8 12]

你可以简单地使用：

np.max(arr, 1)[i]

获取第i行的最大值

或者：

np.max(arr[i])