Numpy Loop Through Array In Python Definition Function