如何在使用sort()[复制]将数组元素展平并按降序排序后保留其索引

这与前面的问题类似,我认为答案在[这个][1] https://stackoverflow.com/questions/5469286/how-to-get-the-index-of-a-maximum-element-in-a-numpy-array-along-one-axis

使用Numpy[argmax][1]有一个优雅的解决方案 1.https://numpy.org/doc/stable/reference/generated/numpy.argmax.html

argmax解决方案:

>>> import numpy as np
>>> a = np.array([[1,2,3],[4,3,1]])
>>> i,j = np.unravel_index(a.argmax(), a.shape)
>>> [i,j]
[1,0]
>>> a[i,j]
4