Map Function In Python For Taking Input In Array