Numpy Loop Over Array Python Example