看代码吧~

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() 返回最大值索引号

以上为个人经验，希望能给大家一个参考，也希望大家多多支持免费资源网。

声明：本站所有文章，如无特殊说明或标注，均为本站原创发布。任何个人或组织，在未征得本站同意时，禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益，可联系我们进行处理。