任务描述:
现在有一个二维数组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)
运行结果
现在有一个二维数组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)
运行结果