python中的np.argmax() 返回最大值索引号
作者：rainie1003
这篇文章主要介绍了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()的用法
用法：求的是最大值对应的索引
案例如下
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。