import matplotlib.pyplot as plt
import numpy as np
from sys import argv
from itertools import product
plot_n_tau = 400
results_file = argv[1]
float_cmp = lambda x,y: cmp(float(x), float(y))
arch = HDFArchive(results_file,'r')
beta = arch['beta']
move_global_prob = arch['move_global_prob']
all_mg = arch.keys()
all_mg.remove('beta')
all_mg.remove('move_global_prob')
# Plot occupations
pp = PdfPages('occupation_mg_beta%.0f_prob%.2f.pdf' % (beta,move_global_prob))
dat_N1_up = {mg: arch[mg]['N1_up'] for mg in all_mg}
dat_N1_dn = {mg: arch[mg]['N1_dn'] for mg in all_mg}
dat_N2_up = {mg: arch[mg]['N2_up'] for mg in all_mg}
dat_N2_dn = {mg: arch[mg]['N2_dn'] for mg in all_mg}
x = sorted(dat_N1_up.keys())
index = np.arange(len(x))
bar_width = 0.5
