How To Take Array Input In Python Using Map