結果 : how to code a math problem in python