在Python中重新组织数组元素

如何重新组织元素A (shape = (1,7,2))生成新数组A1 (shape = (1,7,2))按增加的顺序j? In A[0,0]= array([0, 1]), j指的是1 in [0,1].

如果有一些非常通用的东西那就太好了，这样代码就可以处理形状为A变得非常大，比如说(1,1000,2).

import numpy as np

A = np.array([[[0, 1],[0, 2],[1, 3],[2, 5],[3, 4],[4, 7],[5, 6]]])
print("A =", A.shape)

期望的输出是

A1 = np.array([[[0, 1],[0, 2],[1, 3],[3, 4],[2, 5],[5, 6],[4, 7]]])

首先定义顺序，然后按顺序对数组进行排序。

A = np.array([[[0, 1],[0, 2],[1, 3],[2, 5],[3, 4],[4, 7],[5, 6]]])
order = A[0,:, 1].argsort()
A =np.array([A[0][order]])
print(A)