import numpy as np
a = np.array([[1, 5, 5, 2],
[9, 6, 2, 10],
[3, 7, 9, 1]])
a = np.array([[1, 5, 5, 2],
[9, 6, 2, 10],
[3, 7, 9, 1]])
print(np.argmax(a, axis=1))
#输出(1,3,2)
axis=1时函数的作用为每一列进行比较,输出每行中数值最大的索引
print(np.argmax(a, axis=1))
#输出(1,3,2)
axis=1时函数的作用为每一列进行比较,输出每行中数值最大的索引