Taking Input For Array In Python