如何交换数组中的二维

Question

我有一组来自 Tiff 文件的 4 维数据。

image_stack = io.imread(path, plugin='tifffile')
print(image_stack.shape)
>>> (21, 10, 1331, 1126) 

所以最后 2 个维度是图像分辨率。 10 是因为我在给定时间有 10 个相同图像的切片（在 z 轴上） 而 21 是因为所有这些图像都是每秒拍摄的。

如何交换 21 和 10 维度？

Answer 1

你可以用swapaxes交换前两个轴：

a = np.random.rand(21, 10, 1331, 1126)
a.swapaxes(1,0).shape
# (10, 21, 1331, 1126)

或者用rollaxis向后移动第二个轴：

np.rollaxis(a, 1).shape
# (10, 21, 1331, 1126)