Numpy Loop Through Array Python Example