Python 获取数组最大值、获取数组最大值索引

任务描述：

现在有一个二维数组test_y，我想要获取每一行的最大值，以及最大值所在索引值

代码：

随机生成一个3行4列的数组，利用np.argmax（）函数获取最大值所在索引，amax（）获取最大值的值。

import numpy as np

test_y = np.random.rand(3,4)
idx = np.argmax(test_y, axis=1)
pro = np.amax(test_y, axis=1)

print(idx)
print(pro)

运行结果