plot_darkies.py 712 B

123456789101112131415161718192021222324252627
  1. import matplotlib.pyplot as plt
  2. import numpy as np
  3. import os
  4. import glob
  5. darkies = []
  6. idx = 0
  7. for darkie in glob.glob('log/darkie[0-9]*.log'):
  8. with open(darkie) as f:
  9. buf = f.read()
  10. lines = buf.split('\n')
  11. apr = float(lines[2].split(':')[1].strip())
  12. initial_stake = [float(item) for item in lines[0].split(':')[1].split(',')]
  13. idx +=1
  14. if sum(initial_stake)!=0 :
  15. darkies += [(initial_stake, apr, idx)]
  16. for darkie in darkies:
  17. plt.plot(darkie[0])
  18. legends = []
  19. for darkie in darkies:
  20. legend = ["darkie{}, apr: {}".format(darkie[2], format(darkie[1]))]
  21. legends +=[legend]
  22. plt.legend(legends)
  23. plt.show()
  24. plt.savefig("log/plot_darkies.png")