获取已排序矩阵的索引 [英] Get an index of a sorted matrix

问题描述

我有一个2D np.array：

I have a 2D np.array:

array([[ 1523.,   172.,  1613.],
       [ 3216.,   117.,  1999.],
       [   85.,  1271.,     4.]])

我想按值提取这个np.array的排序索引。

结果应该是（例如）：
[[2,2]，[2,0]，[1,1]，[0,1]，[2,1]，[0,0]，[0,2]，[1,2]，[ 1,0]]

I would to extract the sorted indexes of this np.array by value.
The results should be (for example) : [[2,2],[2,0],[1,1],[0,1],[2,1],[0,0],[0,2],[1,2],[1,0]]

我已经看到了如何提取分钟：

I already saw how to extract the min :

np.unravel_index(np.argmin(act),act.shape) #(2,2)

谢谢

推荐答案

使用 numpy.argsort axis = None （假设展平数组）：

Using numpy.argsort with axis=None (assuming flatten array):

>>> import numpy as np
>>> 
>>> act = np.array([[ 1523.,   172.,  1613.],
...                 [ 3216.,   117.,  1999.],
...                 [   85.,  1271.,     4.]])
>>> n = act.shape[1]
>>> zip(*np.argsort(act, axis=None).__divmod__(n))
[(2, 2), (2, 0), (1, 1), (0, 1), (2, 1), (0, 0), (0, 2), (1, 2), (1, 0)]

这篇关于获取已排序矩阵的索引的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！