Numpy Loop Through Array In Python Definition