python分段函数如何编写