How To Take Array Input In Python Using Dictionaries