我想创建一个尺寸为 10x5 且颜色固定为 [0, 1, 2]
的 numpy 三 channel 图像。我目前正在使用以下代码进行操作:
x = np.array([0, 1, 2])
x = np.array((x,) * 10)
x = np.array((x,) * 5)
这可行,但不是很优雅。用更少的代码实现相同目标的最好/最有效的方法是什么?
最佳答案
或者,您可以使用 np.full
:
np.full((10, 5, 3), [0, 1, 2])
它创建一个给定形状的数组 (10, 5, 3)
并用常量值 [0, 1, 2]
填充它。
关于python - 创建具有固定颜色的三 channel 图像的最佳方法?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/51932766/