如何在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.new使用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)

请参阅this

Answer 3

import scipy.misc
import numpy as np
data = [1,0,1,0,1,0...]
data = np.array(data).reshape(50,50)
scipy.misc.imsave('outfile.bmp', data)

Answer 4

不适用于二进制数组

如果您使用此格式的二进制数据 - b'\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00...，则可以使用此方法将其写为图像文件：

with open("image_name.png", "wb") as img:
    img.write(binary_data)