pp_exp2/e/data.py

#!/usr/bin/env python3
import numpy as np
import sys
n = int(sys.argv[1])

f = open('data.txt', 'wt')
f.write('%d\n'%(n))

a = np.random.random(size=(n, n)) * 0.6
a = np.rint(a)

np.savetxt(f, a, fmt='%d')
f.close()