plotting.py

# Include this only if running in a Jupyter notebook
# %matplotlib inline
 
import matplotlib.pyplot as plt
from root_pandas import read_root
 
df = read_root("ContinuumSuppression_applied.root")
 
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")