主页 / 问题 / 1274678
Accepted
bluesky
Asked:2022-04-25 23:05:24 +0000 UTC2022-04-25 23:05:24 +0000 UTC

如何合并 NumPy 数组

  • 772

有一个分辨率为 (220, 220) 的图像，即 三个 RGB 层的数组 (220, 220, 3)。如何在不干扰图像顺序的情况下向该数组添加四层（假设为零）。据我了解，应该获得一个大小为 (220, 220, 7) 的数组，还是我弄错了

import numpy as np
a = np.arange(220 * 220 * 3).reshape(220 * 220, 3)
a = np.array(np.vsplit(a, 220))

b = np.zeros(220 * 220 * 4).reshape(220 * 220, 4)
b = np.array(np.vsplit(b, 220))

c = np.vstack((a, b))
python

1 个回答

  1. Best Answer

    在 numpy 中，图片传统上存储在высота * ширина * глубина. 深度负责图片中的层数。

    我们从三层创建图像。图像尺寸 4 x 5：

    @>>> a = np.arange(4 * 5 * 3).reshape(4, 5, 3)
@>>> a[:, :, 0]
array([[ 0,  3,  6,  9, 12],
       [15, 18, 21, 24, 27],
       [30, 33, 36, 39, 42],
       [45, 48, 51, 54, 57]])
@>>> a[:, :, 1]
array([[ 1,  4,  7, 10, 13],
       [16, 19, 22, 25, 28],
       [31, 34, 37, 40, 43],
       [46, 49, 52, 55, 58]])
@>>> a[:, :, 2]
array([[ 2,  5,  8, 11, 14],
       [17, 20, 23, 26, 29],
       [32, 35, 38, 41, 44],
       [47, 50, 53, 56, 59]])

    创建两个附加层：

    @>>> b = np.zeros((4, 5, 2), dtype=int)
@>>> b[:, :, 0]
array([[0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0]])
@>>> b[:, :, 1]
array([[0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0]])

    沿最后一个轴粘贴：

    @>>> c = np.concatenate((a, b), axis=2)
@>>> c[:, :, 2]
array([[ 2,  5,  8, 11, 14],
       [17, 20, 23, 26, 29],
       [32, 35, 38, 41, 44],
       [47, 50, 53, 56, 59]])
@>>> c[:, :, 3]
array([[0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0]])
    • 4

相关问题