Loop Through Numpy Array Python Code Editor