如何在Python中将二进制数组写为图像？

Question

我在Python中有一个二进制数组:

data = [0, 1, 1, 0, 1, 0, 1, 1, 0, 1, 0, 0, 0, 0, 0, 1...]

我想取出这些数据并将其保存为位图,其中"0"对应于白色,"1"对应于黑色.我知道数组中有2500个数字,对应于50x50位图.我已经下载并安装了PIL,但我不确定如何将它用于此目的.如何将此数组转换为相应的图像？

Answer 1

您可以使用Image.newwith 1模式并将每个整数作为像素放在初始图像中:

>>> from PIL import Image
>>> import random

>>> data = [random.choice((0, 1)) for _ in range(2500)]
>>> data[:] = [data[i:i + 50] for i in range(0, 2500, 50)]
>>> print data
[[0, 1, 0, 0, 1, ...], [0, 1, 1, 0, 1, ...], [1, 1, 0, 1, ...], ...]

>>> img = Image.new('1', (50, 50))
>>> pixels = img.load()

>>> for i in range(img.size[0]):
...    for j in range(img.size[1]):
...        pixels[i, j] = data[i][j]

>>> img.show()
>>> img.save('/tmp/image.bmp')

Answer 2

在numpy和matplotlib这样做将是这样:

import matplotlib.pyplot as plt
import matplotlib.cm as cm
import numpy as np
plt.imsave('filename.png', np.array(data).reshape(50,50), cmap=cm.gray)

看到这个