展开 python怎么编写函数计算