将彩色图像转换为单行 numpy 数组 (r1, g1, b1; r2, g2, b2; ...)

Question

谁能告诉我如何使用 (r1, g1, b1; r2, g2, b2; ...) 将彩色图像转换为单行 NumPy 数组？

到目前为止，我已经将尺寸为 512x512px 的图像加载到 (512, 512, 3) numpy 数组中。现在我想把它转换成 (1 x 786432) 数组。

Answer 1

ndarray.flatten() 应该可以解决问题：

In [15]: image = np.arange(4*4*3).reshape((4, 4, 3))

In [16]: image
Out[16]: 
array([[[ 0,  1,  2],
        [ 3,  4,  5],
        [ 6,  7,  8],
        [ 9, 10, 11]],

       [[12, 13, 14],
        [15, 16, 17],
        [18, 19, 20],
        [21, 22, 23]],

       [[24, 25, 26],
        [27, 28, 29],
        [30, 31, 32],
        [33, 34, 35]],

       [[36, 37, 38],
        [39, 40, 41],
        [42, 43, 44],
        [45, 46, 47]]])

In [17]: image.flatten()
Out[17]: 
array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15, 16,
       17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33,
       34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47])

Answer 2

要将所有内容都放在一行中，您可以使用 reshape。

array.reshape(1, -1)

这将 return 数组的二维视图，形状为 (1, 786432)，单行。默认情况下，该方法将数组视为 C 顺序，因此较高的维度首先是 "flattened"（在这种情况下这是您想要的）。

或者，您可以使用 ravel 返回平面阵列：

array.ravel()

这个区别是数组的 returned 视图是一维的：形状 (786432,).

将彩色图像转换为单行 numpy 数组 (r1, g1, b1; r2, g2, b2; ...)

Convert a color image into single-row numpy array (r1, g1, b1; r2, g2, b2; ...)

python

arrays

numpy