Binary string to pixel matrix visualization

main.py

 # Simple script to visualize a binary string as a pixel matrix
 # Example input: "01100001110101011100010[...]"
 # Note that the data length should be some n^2 to produce a square image
 
import numpy as np
from math import sqrt

file = open('input.txt', 'r')
cont = file.read()[:-1]
data = [_char == '1' for _char in cont]
edge = int(sqrt(len(data)))
data = np.array(data).reshape(edge, edge)

import matplotlib.pyplot as plt
plt.imshow(data, aspect=1, interpolation='none')
plt.savefig('out.png')