看代码吧~
1
2
3
4
5
|
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()的用法
用法:求的是最大值对应的索引
案例如下
以上为个人经验,希望能给大家一个参考,也希望大家多多支持服务器之家。
原文链接:https://blog.csdn.net/xykimred/article/details/84872897