def plot_histogram(self, o, hist): plt.rcParams['grid.linestyle'] = '--' edges = tools.get_edges(o) plt.plot(edges, hist, linewidth=3) for c in self.cl: plt.plot(edges, self.pdfs[o][c], label=TRADUCTION_TABLE[c], linewidth=3) plt.xlabel('{}({})'.format(PARAM_NAME_TABLE[o], PARAM_UNIT_TABLE[o])) plt.ylabel('p({})'.format(PARAM_NAME_TABLE[o])) plt.xlim(PLOT_PARAM_TABLE[o]) plt.grid() plt.show()
def save_pdfs(self): for o in self.obs: edges = tools.get_edges(o) for c in self.cl: filename = 'data/pdfs/{}_{}.txt'.format(o, c) np.savetxt(filename, np.c_[edges, self.pdfs[o][c]])