假设我有一个 numpy 数组 A,形状为 (5,24,1)。
我想沿 axis = 0 进行五个单独的转置,这样转置后的结果形状是 (5,1,24)。
我如何使用一些 numpy 函数来做到这一点?
请您参考如下方法:
可以建议三种方法-
A.swapaxes(1,2)
A.transpose(0,2,1)
np.rollaxis(A,2,1)
对于swapaxes 和transpose,请遵循他们的文档。
与 np.rollaxis ,我们正在滚动轴以将第三个轴带到第二个位置,从而将第二个轴推回到第三个轴位置。链接文档中的更多信息。
利用最后一个轴是单例的事实 -
A[:,None,:,0]
