import pyhande
import matplotlib.pyplot as plt
(metadata, qmc_data) = pyhande.extract.extract_data('calcs/fciqmc/hubbard_fciqmc.out')[0]
plt.plot(qmc_data['iterations'], qmc_data['Shift'], label=r'$S(\tau)$')
plt.plot(qmc_data['iterations'], qmc_data['\sum H_0j N_j']/qmc_data['N_0'], label=r'$E(\tau) = \sum_j H_{0j} N_j(\tau)/N_0(\tau)$')
plt.legend()
plt.xlabel('iteration')
plt.ylabel('Correlation energy / $t$')