将 2D 数组转换为重复现有值的 3D 数组

Question

我有一个形状数组 (360,480)，其中包含从 1 到 11 的值，

 Array([[ 1,  1,  1, ...,  1,  1,  1],
   [ 1,  1,  1, ...,  1,  1,  1],
   [ 1,  1,  1, ...,  1,  1,  1],
   ...,
   [ 4,  4,  4, ..., 11, 11, 11],
   [ 4,  4,  4, ..., 11, 11, 11],
   [ 4,  4,  4, ..., 11, 11, 11]])

我怎样才能将这个数组重塑成一个形状为 (360,480,3) 的数组

np.all(array[:,:,0]==array[:,:,1])

和

np.all(array[:,:,0]==array[:,:,2])

都是真的吗？ 预期的结果应该是

array([[[ 1,  1,  1],
    [ 1,  1,  1],
    [ 1,  1,  1],
    ...,
    [ 1,  1,  1],
    [ 1,  1,  1],
    [ 1,  1,  1]],

   [[ 4,  4,  4],
    [ 4,  4,  4],
    [ 4,  4,  4],
    ...,
    [11, 11, 11],
    [11, 11, 11],
    [11, 11, 11]],

   [[ 4,  4,  4],
    [ 4,  4,  4],
    [ 4,  4,  4],
    ...,
    [11, 11, 11],
    [11, 11, 11],
    [11, 11, 11]]])

Answer 1

你可以使用 numpy.repeat 函数

https://numpy.org/doc/stable/reference/generated/numpy.repeat.html

array3d = np.repeat(array2d[:, :, None], repeats=3, axis=2)