Take User Input Array In Python Syntax If Then Else