python - 通过PIL将矩阵写入图像

Question

我有一个 2D 矩阵，我想将其写入 PBM 文件(只有 -1,1 值，黑白)。

我刚刚发现PIL，但以下方法不起作用:

im = Image.new('L', (self.nx,self.ny))  
data=[[255*(self.spins[i][j][0].m+1)/2 for j in range(0,self.ny)]for i in range(0,self.nx)]  
im.putdata(data)   
im.save('my_image.pbm')  

也就是说，我得到了带有标题的 PBM 文件，但没有数据:

P5  
4 3  
255  

如果有人可以帮助我...

谢谢!

亚历山德罗

Answer 1

Image.putdata 采用一维序列，而不是像代码中那样的多维序列。

即而不是使用类似的东西

[[v1, v2, v3],
 [v4, v5, v6],
 [v7, v8, v9]] 

要保存传递​​给 putdata 的像素数据，应该是

[v1, v2, v3, v4, v5, v6, v7, v8, v9]