Pep8ify examples

This PR converts the animation, api and axes_grid examples.

(Retry of #3182/#3183)

I addressed @ianthomas23' comments from #3183
I personally prefer NO spaces around the arithmetic operator in this case. I'll remove this change if you are fine with that