結果 : how to write math functions in python