python自定义函数拟合