python数组求和函数