Array Input In Function Python Syntax If Else Statement