Numpy Loop Through Array In Python Code For Snake