How to write a binary array as an image in Python?

Question

I have an array of binary numbers in Python:

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

I would like to extract this data and save it as bitmap images, and “0” is white and “1” is black. I know that there are 2500 numbers in the array, which corresponds to a 50x50 bitmap. I downloaded and installed PIL, but I'm not sure how to use it for this purpose. How to convert this array to the corresponding image?

Answer 1

numpy and matplotlib way to do this:

 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) 

See this

Answer 2

You can use Image.new with 1 and put each integer as a pixel in the original image:

 >>> 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 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)