density=False,proton_line=0.135,first_color="blue",xlabel_1="NF Data") sys.exit() x_data = df["Mpi0"] y_data = df["pmass"] var_names = ["E Px","E Py"] saveplots = False output_dir = "figures/" title = "Pi Mass vs EPx" filename = title units = ["GeV","Gev"] ranges = [[0,.3,100],[.9,.96,100]] #interactive(True) make_histos.plot_2dhist(x_data,y_data,var_names,ranges,colorbar=False, saveplot=saveplots,pics_dir=output_dir,plot_title=title.replace("/",""), filename=filename.replace(" ",""),units=units) sys.exit() x_data = df2[1] y_data = df2[2] title = "Electron $P_X$ vs. $P_Y$, NF 4-Feat. Model, $M_e^2$ Cut" make_histos.plot_2dhist(x_data,y_data,var_names,ranges,colorbar=False, saveplot=saveplots,pics_dir=output_dir,plot_title=title.replace("/",""), filename=filename.replace(" ",""),units=units) #interactive(False) #plt.show() sys.exit()
##df_ps = df_ps.query("nf_Mpi0<.15") #df_ps = df_ps.query("nf_Mpi0>0.12") # df_ps = df_ps.query("nf_Pz_1-recon_Pz_1<0.01") # df_ps = df_ps.query("nf_Pz_1-recon_Pz_1>-0.01") # df_ps = df_ps.query("nf_Pz_2-recon_Pz_2<0.01") # df_ps = df_ps.query("nf_Pz_2-recon_Pz_2>-0.01") df = df_ps names = ["Px_1", "Py_1", "Pz_1", "Px_2", "Py_2", "Pz_2", "Mpi0"] for name in names: make_histos.plot_2dhist( df["gen_{}".format(name)], df["gen_{}".format(name)] - df["recon_{}".format(name)], ["gen_{}".format(name), "gen - recon: {}".format(name)], colorbar=True, saveplot=True, pics_dir="julypics/", plot_title="none", filename="gen-recon_{}".format(name), units=["GeV", "GeV"]) make_histos.plot_2dhist( df["gen_{}".format(name)], df["gen_{}".format(name)] - df["nf_{}".format(name)], ["gen_{}".format(name), "gen - nf: {}".format(name)], colorbar=True, saveplot=True, pics_dir="julypics/", plot_title="none", filename="gen-recon_{}", units=["GeV", "GeV"]) # plt.scatter(df["recon_{}".format(name)],df["recon_{}".format(name)]-df["nf_{}".format(name)])