plotting.py

# Include this only if running in a Jupyter notebook
# %matplotlib inline
 
import matplotlib.pyplot as plt
import uproot
 
var_list = ['isContinuumEvent', 'ContProb']
df = uproot.open("ContinuumSuppression_applied.root:tree").arrays(var_list, library='pd')
 
fig, ax = plt.subplots()
 
signal_df = df.query("(isContinuumEvent == 0.0)")
continuum_df = df.query("(isContinuumEvent == 1.0)")
 
hist_kwargs = dict(bins=30, range=(0, 1), histtype="step")
ax.hist(signal_df["ContProb"], label="Not Continuum", **hist_kwargs)
ax.hist(continuum_df["ContProb"], label="Continuum", **hist_kwargs)
ax.set_xlabel("ContinuumProbability")
ax.set_ylabel("Total number of candidates")
ax.legend()
fig.savefig("ContinuumProbability.pdf")