Belle II Software  light-2205-abys
plotting_R2.py
1 # Include this only if running in a Jupyter notebook
2 # %matplotlib inline
3 
4 import matplotlib.pyplot as plt
5 from root_pandas import read_root
6 
7 df = read_root("ContinuumSuppression.root")
8 
9 fig, ax = plt.subplots()
10 
11 signal_df = df.query("(isContinuumEvent == 0.0)")
12 continuum_df = df.query("(isContinuumEvent == 1.0)")
13 
14 hist_kwargs = dict(bins=30, range=(0, 1), histtype="step")
15 ax.hist(signal_df["R2"], label="Not Continuum", **hist_kwargs)
16 ax.hist(continuum_df["R2"], label="Continuum", **hist_kwargs)
17 ax.set_xlabel("R2")
18 ax.set_ylabel("Total number of candidates")
19 ax.legend()
20 fig.savefig("R2.pdf")