Numpy Loop Over Array Python Example Code