Take User Input Array In Python Syntax If Else Statement