Thanks for the advice.
I had a bit of a play around with things but the fix is a bit beyond my skill level at the mo.
I keep running up against problems using Pythonista with these libraries (using matplotlib for animations doesn't seem to work either) so I think I'm going to give up on it for now and go back to using the PC.