Moisture Budget Equation Python Programming