import sys import pickle import numpy as np output_file_name, dimension, count = sys.argv[1:] pickle.dump(np.random.random((int(count), int(dimension))), open(output_file_name, 'w'))