python编写函数求和