Python用列表推导式计算嵌套列表正元素和报错如何解决

报错原因 #

• 直接语法错误原因：sum_l += y 是赋值语句，而列表推导式的元素位置只能放置可返回值的表达式，不允许直接写语句，因此触发语法报错。
• 额外逻辑错误：过滤条件 if y > 0 写在了外层循环 for x in data 的后面，此时内层循环的变量y还未定义，即使解决语法问题也会触发NameError；且使用全局累加变量sum_l会将所有子列表的正数合并求和，无法得到每个子列表单独的正数和。

修复后的代码 #

直接用内置sum()函数结合生成器表达式实现需求，代码更简洁高效：

data = [[10, 6, -1, 2, -4, 3, -10, -3, -2, 4],
[0, -3, -9, -1, 2, 6, -5, 8, -7, 0],
[-2, -10, -2, -7, 8, 0, 1, 0, 8, -5]]

# 遍历每个子列表，对每个子列表内的正数求和
sum_list = [sum(num for num in sub_list if num > 0) for sub_list in data]
print(sum_list) # 输出结果：[25, 16, 17]

如果需要自定义累加逻辑（不使用内置sum函数），也可以用常规嵌套循环实现，可读性更高：

sum_list = []
for sub_list in data:
    sub_sum = 0
    for num in sub_list:
        if num > 0:
            sub_sum += num
    sum_list.append(sub_sum)

Python列表推导式的核心作用是快速生成列表，遵循以下结构和规则：

基础结构 #

• 无过滤的单循环结构：[求值表达式 for 循环变量 in 可迭代对象]
  等价于普通循环写法：

  结果列表 = []
  for 循环变量 in 可迭代对象:
      结果列表.append(求值表达式)
  

• 带过滤的单循环结构：[求值表达式 for 循环变量 in 可迭代对象 if 过滤条件]
  只有满足过滤条件的循环变量才会参与表达式求值，加入结果列表
• 多层嵌套循环结构：[求值表达式 for 外层变量 in 外层可迭代对象 for 内层变量 in 内层可迭代对象]
  循环顺序和普通嵌套for循环完全一致，外层循环在前，内层循环在后

注意事项 #

• 推导式内所有位置只能使用表达式，不允许出现赋值、break、continue、return这类语句
• 过滤条件的变量作用域和其所属的循环层级绑定，不能使用未在当前/外层循环定义的变量
• 避免在推导式内修改外部全局变量，不仅可读性差，还容易产生不可预期的副作用

内容的提问来源于stack exchange，提问作者hello_w