亲宝软件园·资讯

展开

python np.argmax最大值索引号 python中的np.argmax() 返回最大值索引号

rainie1003 人气:0
想了解python中的np.argmax() 返回最大值索引号的相关内容吗,rainie1003在本文为您仔细讲解python np.argmax最大值索引号的相关知识和一些Code实例,欢迎阅读和指正,我们先划重点: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()的用法

用法:求的是最大值对应的索引

案例如下

以上为个人经验,希望能给大家一个参考,也希望大家多多支持。

加载全部内容

相关教程
猜你喜欢
用户评论