> @rhattersley I remember writing a comment about this list comprehension
>  ... A few array operations ...

I agree, but I don't think it would be on this PR.
All I was attempting here was reformatting it to pass PEP8 requirements (mainly the 80-column limit).
I don't think functional changes or refactoring belong here.