Getting Array Input In Python Syntax For If Else Statements