【python】np.argmax() 返回最大值索引号 2022-10-09 12:05:21 import numpy as np a = np.array([[2, 4, 6, 1], [1, 5, 2, 9]]) print(np.argmax(a)) print(np.argmax(a, axis=0)) #竖着比较,返回行号 print(np.argmax(a, axis=1)) #横着比较,返回列号输出:7 [0 1 0 1] [2 3]