Pep8ify examples

This PR converts the animation, api and axes_grid examples.

(Retry of #3182/#3183)

I addressed @ianthomas23' comments from #3183
Can you re-factor the other test this way as well?