python 用参数传递的方法 将1维数组修改成2维RGB图像

假设我们有一个长度为$n$的1维数组$A$，表示一张$nxm$的RGB图像，其中每个像素点由3个元素组成（R,G,B），即$A=[R_1,G_1,B_1,R_2,G_2,B_2,...,R_{n\times m},G_{n\times m},B_{n\times m}]$。

我们可以通过参数传递的方式将其修改为2维RGB图像。具体来说，我们可以定义一个函数，该函数接受一个1维数组$A$和图像的尺寸$n,m$作为参数，然后将$A$转换为2维RGB图像$B$，其中$B_{i,j}=[R_{i,j},G_{i,j},B_{i,j}]$。代码如下：

def reshape_image(A, n, m):
    B = []
    for i in range(n):
        row = []
        for j in range(m):
            pixel = [A[(i*m+j)*3], A[(i*m+j)*3+1], A[(i*m+j)*3+2]]
            row.append(pixel)
        B.append(row)
    return B

我们可以使用以下代码测试该函数：

A = [255,0,0, 0,255,0, 0,0,255, 255,255,0, 255,0,255, 0,255,255]
B = reshape_image(A, 2, 3)
print(B)

输出结果为：

[[[255, 0, 0], [0, 255, 0], [0, 0, 255]], [[255, 255, 0], [255, 0, 255], [0, 255, 255]]]

可以看到，函数成功将1维数组$A$转换为了2维RGB图像$B$。其中，$B_{0,0}=[255,0,0]$表示第一个像素点的颜色为红色。