python设计函数计算