結果 : how to write mathematical functions in python