网络编程
位置:首页>> 网络编程>> Python编程>> python中的np.argmax() 返回最大值索引号

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

作者:rainie1003  发布时间:2022-08-07 14:22:57 

标签: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()的用法

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

案例如下

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

来源:https://blog.csdn.net/xykimred/article/details/84872897

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com