import pyhande
import matplotlib.pyplot as plt
(metadata, qmc_data) = pyhande.extract.extract_data('calcs/dmqmc/heisenberg_reweight_single.out')[0]
tau = metadata['qmc']['tau']
for i in range(0,10):
    plt.plot(qmc_data['iterations']*tau, qmc_data['Excit. level %s'%i], label=r'Excit. level %s'%i)
plt.xlim([0,5000*tau])
plt.xlabel(r'$\beta J$')
plt.legend(numpoints=1, loc='best')
plt.ylabel('Weight of psips')