您当前的位置:首页 > IT编程 > python
| C语言 | Java | VB | VC | python | Android | TensorFlow | C++ | oracle | 学术与代码 | cnn卷积神经网络 | gnn | 图像修复 | Keras | 数据集 | Neo4j | 自然语言处理 | 深度学习 | 医学CAD | 医学影像 | 超参数 | pointnet | pytorch | 异常检测 | Transformers | 情感分类 | 知识图谱 |

自学教程:python中的np.argmax() 返回最大值索引号

51自学网 2021-10-30 22:33:34
  python
这篇教程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()如何将参数转换为字符串类型
万事OK自学网:51自学网_软件自学网_CAD自学网自学excel、自学PS、自学CAD、自学C语言、自学css3实例,是一个通过网络自主学习工作技能的自学平台,网友喜欢的软件自学网站。