Python Numpy: reverse transpose

Question

假设我有一个形状为(2,2,3)的数组:x=np.array([[[1,2,3],[4,5,6]],[[7,8,9],[10,11,12]]])，并且我将该数组沿坐标轴(2,0,1)转置为y:y=np.transpose(x,(2,0,1))。

我的问题是，如何把y转置回x?我知道我可以使用np.transpose(y,(1,2,0))将其变回x，但我希望有一种简洁的方法来找到正确的轴(1,2,0)。

谢谢你的帮助!

Answer 1

可以使用

np.argsort((2, 0, 1))

它给:

array([1, 2, 0])

它适用于任何轴的排列。