Loop Through Array In Python Definition Function