結果 : how to do a math function in python