Loop Through Numpy Array Python Code For Snake