我有一个数组,我想找到最大值的索引。
例如:
myarray = np.array([1,8,8,3,2])
我想要得到结果:[1,2]
,我该怎么做?
(实际上,我尝试过np.argmax(myarray)
,但是它只返回第一次出现的内容[1]
)
鉴于:
>>> myarray = np.array([1,8,8,3,2])
你可以做:
>>> np.where(myarray==myarray[np.argmax(myarray)])
(array([1, 2]),)
或者,
>>> np.where(myarray==max(myarray))
(array([1, 2]),)
或者,
>>> np.nonzero(myarray==max(myarray))
(array([1, 2]),)
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句