编写python函数