Taking Input Array In Python W3schools