python中的np.argmax() 返回最大值索引号
更新时间:2021年06月02日 10:42:28 作者: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()的用法
用法:求的是最大值对应的索引
案例如下
以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。
相关文章
Ubuntu20.04环境安装tensorflow2的方法步骤
这篇文章主要介绍了Ubuntu20.04环境安装tensorflow2的实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧2021-01-01Python对Excel按列值筛选并拆分表格到多个文件的代码
这篇文章主要介绍了Python对Excel按列值筛选并拆分表格到多个文件,本文通过代码给大家介绍的非常详细,需要的朋友可以参考下2019-11-11
最新评论