結果 : how to make a math function in python