numpy argsort返回2d索引数组吗？

numpy argsort返回2d索引数组吗？

应用

numpy.argsort

在展平的数组上，然后将索引分解回（3，3）形状：

>>> arr = np.array([[5, 2, 4],[3, 3, 3],[6, 1, 2]])>>> np.dstack(np.unravel_index(np.argsort(arr.ravel()), (3, 3)))array([[[2, 1],        [0, 1],        [2, 2],        [1, 0],        [1, 1],        [1, 2],        [0, 2],        [0, 0],        [2, 0]]])