Take User Input Array In Python Syntax If Loop