根据另一多维数组排序

例如我有

array1 = np.array([[3,2,1],[2,1,3]])

array2 = np.array([[4,5,6],[7,8,9]])

我对array1进行排序以获得[[1,2,3],[1,2,3]],并且我对array2进行排序以获得[[6,5,4],[8,7,9]]

我试图做的是以下几点:

index = np.argsort(array1,axis = 1)

array2[index]

但这是行不通的,非常感谢您的帮助