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

Question

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

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

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

标签： pythonnumpyscipy

Answer 1

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

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

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