用python实现分段函数