numpy Python中数组每行的最大值的索引

svdrlsy4 于 2023-03-23 发布在 Python

关注(0)|答案(1)|浏览(111)

我有一个数组Pr。我想打印每行最大值的索引。例如，对于行0，最大值是1.72731864e+003，它出现在索引0处。我给出了预期的输出。

import numpy as np 
Pr = np.array([[1.72731864e+003, 0.00000000e+000],
       [0.00000000e+000, 1.24439020e+003]])

MaxPr=Pr.max(axis=1)

预期输出为

indices=[0,1]

来源：https://stackoverflow.com/questions/75692561/index-of-maximum-value-of-each-row-of-an-array-in-python

1条答案

按热度按时间

可以使用argmax（）

Pr.argmax(axis=-1)

赞(0）回复(0）举报 2023-03-23

相关问题

微信公众号

热门标签

Java query python Node 开发语言 request Util 数据库 Table 后端算法 Logger Message Element Parser

最新问答

Python pillow/PIL无法识别对象“imagedraw”的属性“textsize”
回答(3) 发布于 5个月前
python-3.x ImportError：无法从“typing_extensions”导入名称“deprecated”
回答(2) 发布于 6个月前
python-3.x 我正尝试使用密码学对字典进行加密
回答(1) 发布于 5个月前
在Python的IDLE中导入audiolazy的问题
回答(1) 发布于 5个月前
如何将Python程序中的数据添加到Markdown？
回答(2) 发布于 6个月前