figures.py

import matplotlib.pyplot as plt 
import numpy as np
import sys 

def make_figure(arr):
    fig = plt.figure()
    ax = fig.add_subplot(111)
    imag = ax.imshow(arr)
    plt.savefig("output2.png")
    plt.close(fig)
    return None

def main():
    iterations = int(sys.argv[1])
    for i in range(iterations):
        g = np.random.randn(1000, 1000)
        make_figure(g)
    
    return 0


if __name__ == "__main__": 
    main()