python编写函数求平均值