Taking Input In Array In Python