Python Numpy:矩阵转置倒序

3
假设我有一个形状为(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)。
谢谢任何帮助!
1个回答

4

您可以使用

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

它提供:

array([1, 2, 0])

它可以适用于任何轴的排列。


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接