Tiny improvement for SoA over AoS in Python

I benchmarked SoA vs AoS in Python. While SoA was more efficient than AoS, its readability penalty outweighed the gains.

Happy to get some feedback about better ways to do this.