How To Assign A Variable With Equation Python