利用Python-PIL库的Image模块将RGB矩阵还原成图片

利用Python-PIL库的Image模块将RGB矩阵还原成图片

最近在用Python模拟实现k-means算法时，遇到了一个大问题，如何将聚类完成后的RGB矩阵还原为图片？

下面用一个小示例记录一下实现的过程。

步骤

1.打开图像，并获得其对应的RGB矩阵

#读取图像
from PIL import Image
def getImgData():
    img = Image.open('example.jpg','r')
    print(img.size)
    lst = list(img.getdata())
    return lst

img.size为（500,358）

2.将RGB矩阵转为n*m的列表；n指的是图片的行数（500），m指的是图片的列数（358）

lst = getImgData()
lst1 = []
for i in range(0,500):
    lst1.append(lst[357*i:357*i+357])

3.验证一下是否转换成功

for val,i in zip(lst1,range(0,len(lst1))):
    print(val)
    print(len(val))

4.将每个RGB转为像素，然后保存成新的图片

newimg = Image.new('RGB',(500,358))
i=0
for val,i in zip(lst1,range(0,len(lst1))):
    for j in range(0,len(val)):
        newimg.putpixel((i,j),(int(val[j][0]),int(val[j][1]),int(val[j][2])))
newimg.save('flag.jpg')

完整代码

#读取图像
from PIL import Image
def getImgData():
    img = Image.open('example.jpg','r')
    print(img.size)
    lst = list(img.getdata())
    return lst
lst = getImgData()
lst1 = []
for i in range(0,500):
    lst1.append(lst[357*i:357*i+357])
for val,i in zip(lst1,range(0,len(lst1))):
    print(val)
    print(len(val))
print(i)
newimg = Image.new('RGB',(500,358))
i=0
for val,i in zip(lst1,range(0,len(lst1))):
    for j in range(0,len(val)):
        newimg.putpixel((i,j),(int(val[j][0]),int(val[j][1]),int(val[j][2])))
newimg.save('flag.jpg')

 运行结果

本人是Python菜鸟一个，请大家及时批评指正。