python编写自定义函数