文章目录[隐藏]
看代码吧~
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中最大值所对应的行和列
本文由 华域联盟 原创撰写:华域联盟 » python中的np.argmax() 返回最大值索引号
转载请保留出处和原文链接:https://www.cnhackhy.com/26722.htm