Solving Ode Using Python Dictionaries Multiple Keys