PEP8 fixes in plot_affine_transformation.py

Fixes a number of PEP8 violations in affine matrix example.