python - "Transform" NumPy 数组 : Move Dimension

Question

我正在创建数组 a:

import numpy as np
a = np.zeros((3, 10, 10), np.uint8)
a[1,5,5] = 255

中间有一个红点，其中RGB是第一维。然后我使用 matplotlib 绘制它:

import matplotlib.pyplot as plt
plt.imshow(a)

但这当然行不通，因为 imshow 需要一个维度为 (10, 10, 3) 的数组，而我正在为它提供一个维度为 (3, 10, 10) 的数组。我怎样才能“翻转”数组，使 RGB 成为第三维，而不是第一维？

Answer 1

你需要的是 swapaxes。

import numpy as np
a = np.zeros((3, 10, 10), np.uint8)

print(a.shape) #(3,10,10)

print(np.swapaxes(a,0,2).shape) #(10,10,3)

参见 documentation .

---

np.swapaxes(a,0,2) 等于 np.transpose(a, (2,1,0))。

还有一个选项是np.transpose(a, (1,2 0))。

与往常一样，转置矩阵可以有两个版本，它们产生相似的结果但具有不同的 3 维旋转对称性。

这取决于镜像矩阵是否影响你的结果，你应该仔细测试它是否有所不同。