Array Input In Python Using Dictionaries