如何在python中将二维数组转换为RGB图像？

Question

我想将每个二维数组转换为 RGB 图像并将这个 RGB 图像返回给另一个函数，我该怎么做。我尝试通过PIL 和plt 做到这一点，但它对我不起作用。我尝试了两个多星期来找到我能做到的方法。

任何帮助将不胜感激。

for c in [cD5,cD4,cD3,cD2,cD1]:
        x = np.linspace(0, 3844, len(c))
        f = interp1d(x, c)
        result.append(f(common_x))
        normalized_result = preprocessing.normalize(result)

Answer 1

我认为这已经解决了，记住如果你想要一个 RGB 图像，你需要 3 个通道，这意味着一个具有形状 (NxMx3) 的矩阵。

Convert 2d array to collored image in python

干杯！

编辑：

这是一个示例，说明如何将 2D 数组堆叠成具有所需形状的 3D 数组。

import numpy as np
#some random arrays i just created for test
r = np.array([3,3,3])
g = np.array([6,6,6])
b = np.array([9,9,9])
bl = np.array([12,12,12])

#create the stacked arrays
stacked = np.dstack((r,g,b,bl))

#check the shape
print(np.shape(stacked))