python - 以简洁的方式填充 Numpy 数组的第三维

Question

我有一个表示 RGB 图像的 3D numpy 数组。我想用特定的 RGB 值填充整个图像。 numpy.fill 仅采用标量作为参数——是否有比循环更简洁的方法将相同的三维 RGB 三元组分配给二维网格中的每个点？

Answer 1

也许:

>>> m = np.zeros((2,2,3))
>>> m[:] = [10,20,3]
>>> m
array([[[ 10.,  20.,   3.],
        [ 10.,  20.,   3.]],

       [[ 10.,  20.,   3.],
        [ 10.,  20.,   3.]]])
>>> m[0,0]
array([ 10.,  20.,   3.])
>>> m[0,1]
array([ 10.,  20.,   3.])
>>> m[1,0]
array([ 10.,  20.,   3.])
>>> m[1,1]
array([ 10.,  20.,   3.])