Python中实现子列表对应元素累加的正确方法

你代码里的A[i]+A[i+1]是Python列表的拼接操作，会把两个子列表直接连在一起，这就是输出不符合预期的核心原因——你需要的是对应元素相加，而不是列表拼接。

要生成你预期的累加子列表，有两种实用写法：

方式一：循环迭代生成结果 #

先保留第一个子列表，再依次将后续子列表与前一个结果子列表的对应元素相加：

A = [[0, 0, 0, 0], [1, 2, 3, 4], [5, 6, 7, 8]]
result = [A[0].copy()]  # 复制第一个子列表，避免修改原数据
for i in range(1, len(A)):
    # 对每个位置的元素逐一相加
    new_sub_list = [prev + curr for prev, curr in zip(result[i-1], A[i])]
    result.append(new_sub_list)
print(result)

运行输出：

[[0, 0, 0, 0], [1, 2, 3, 4], [6, 8, 10, 12]]

方式二：用itertools.accumulate简化代码 #

借助标准库的accumulate函数，自定义元素级相加的逻辑：

import itertools

A = [[0, 0, 0, 0], [1, 2, 3, 4], [5, 6, 7, 8]]

def add_elements(a, b):
    return [x + y for x, y in zip(a, b)]

result = list(itertools.accumulate(A, add_elements))
print(result)

这个方法会自动按顺序累加子列表，每次调用add_elements完成对应元素的加法，输出结果和预期一致。

• Python列表的+运算符是拼接操作，不是元素级加法；要做元素级运算必须逐个处理每个元素。
• 使用zip可以方便地将两个子列表的对应元素配对，再通过列表推导式完成相加。

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