data.py 303 B

123456789101112
  1. #!/usr/bin/env python3
  2. import numpy as np
  3. import sys
  4. n = int(sys.argv[1])
  5. f = open('data.txt', 'wt')
  6. f.write('%d\n'%n)
  7. a = np.random.random(size=(n, n))
  8. b = np.random.random(size=(n, n))
  9. np.savetxt(f, a, fmt='%.5f')
  10. np.savetxt(f, b, fmt='%.5f')
  11. np.savetxt(f, np.mat(a) * np.mat(b), fmt='%.5f')
  12. f.close()