python - 创建具有固定颜色的三 channel 图像的最佳方法？

Question

我想创建一个尺寸为 10x5 且颜色固定为 [0, 1, 2] 的 numpy 三 channel 图像。我目前正在使用以下代码进行操作:

x = np.array([0, 1, 2])
x = np.array((x,) * 10)
x = np.array((x,) * 5)

这可行，但不是很优雅。用更少的代码实现相同目标的最好/最有效的方法是什么？

Answer 1

或者，您可以使用 np.full:

np.full((10, 5, 3), [0, 1, 2])

它创建一个给定形状的数组 (10, 5, 3) 并用常量值 [0, 1, 2] 填充它。