Skip to main content
 首页 » 编程设计

python之转置 3-d numpy 数组

2025年05月04日22duanxz

假设我有一个 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]