Is this a possibility? Because as_pyplot_figure just saves a bar graph and the output shown in the notebook seems to be much more intuitive.