Taking Input Array In Python