Taking Input Array In Python Example