python中的np.argmax() 返回最大值索引号

import numpy as npa = 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]

用法：求的是最大值对应的索引

案例如下

以上为个人经验，希望能给大家一个参考，也希望大家多多支持好吧啦网。