コード例 #1
0
ファイル: eval_flow.py プロジェクト: Sangbaek/clas12-nflows
                     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()
コード例 #2
0
##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)])