np.argmax()

import numpy as np


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时函数的作用为每一列进行比较，输出每行中数值最大的索引