根据二维坐标对二维数组进行排序

Question

给定一个2D数组，如何对其元素进行argsort并获得数组元素从低到高的坐标列表

例如:

1 5
2 3

我们得到:

[[0,0], [1,0], [1,1], [0,1].

可以推广到n维数组吗?

Answer 1

对于2D，准确的答案是:

A_2d_argsort = np.argsort(A, None)
ver = np.floor(A_2d_argsort // A.shape[1])
hor = A_2d_argsort % A.shape[1]
np.stack([ver,hor]).transpose()