Sorry for the comments from the peanut gallery.  As a mpl developer, I think the original version is much better.   If you want to shorten it, use `fig, ax = plt.subplots(1, 1)` instead of relying on the state-machine magic.