这篇教程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]
Python np.argmax()的用法用法:求的是最大值对应的索引 案例如下 
以上为个人经验,希望能给大家一个参考,也希望大家多多支持51zixue.net。 Python3 类型标注支持操作 python str()如何将参数转换为字符串类型 |