我有一个数组Pr。我想打印每行最大值的索引。例如,对于行0,最大值是1.72731864e+003,它出现在索引0处。我给出了预期的输出。
Pr
0
1.72731864e+003
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]
lzfw57am1#
可以使用argmax()
Pr.argmax(axis=-1)
1条答案
按热度按时间lzfw57am1#
可以使用argmax()