如何正确使用lambda+reduce处理带下标元素的求和操作?
错误原因
reduce的执行逻辑导致了这个问题:
- 第一次迭代:取列表前两个元组,执行
lambda得到80+120=200,返回的是整数。 - 第二次迭代:把上一步的整数
200作为第一个参数a,第三个元组作为b,此时调用a[1]就会报错——整数没有下标。
正确实现
方式一:先提取金额再累加
用map先把每个元组的金额字段提取出来,再用reduce做累加:
from functools import reduce expenses = [ ('Dinner', 80), ('Car repair', 120), ('csgo skins', 200) ] sum2 = reduce(lambda a, b: a + b, map(lambda x: x[1], expenses)) print(sum2) # 输出400
方式二:给reduce指定初始值
指定初始值0,这样每次迭代时第一个参数a始终是累加后的数字,第二个参数b是元组,直接取b[1]累加:
from functools import reduce expenses = [ ('Dinner', 80), ('Car repair', 120), ('csgo skins', 200) ] sum2 = reduce(lambda a, b: a + b[1], expenses, 0) print(sum2) # 输出400
更简洁的替代方案
其实Python内置的sum函数可以直接完成这个需求,写法更直观:
sum2 = sum(item[1] for item in expenses) print(sum2) # 输出400
内容的提问来源于stack exchange,提问作者Ceccoclat
相关产品推荐
相关产品推荐

