I am creating matplotlib plots using data processed by a function. I would like to plot the data and co