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

看代码吧~

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]

python np.argmax()的用法

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

案例如下

以上为个人经验，希望能给大家一个参考，也希望大家多多支持华域联盟。

您可能感兴趣的文章:

• 详解numpy的argmax的具体使用
• 浅谈numpy.where() 的用法和np.argsort()的用法说明
• python中找出numpy array数组的最值及其索引方法
• Python 实现Numpy中找出array中最大值所对应的行和列