Take User Input Array In Python Interview