Pythonで複数のリスト間の要素(数)の和を新たなリストとして表示したいです。

Question

for loopを利用して(importを使わずに)、[x1, y1, z1],[x2, y2, z2], [x3, y3, z3]という３つのリストに基づいて、新たな[x1+x2+x3, y1+y2+y3, z1+z2+z3]というリストを作りたいです。
例えば、リストが[1, 6, 3], [6, 9, 3], [4, 7, 9]というふうにあった場合、[11, 22, 15]という新たなリストを作りたいです。

Answer 1

a, b, c = [1, 6, 3], [6, 9, 3], [4, 7, 9]
d = [sum(v) for v in zip(a, b, c)]
print(d)

Comments

[snowhawk]

とても簡潔なコードにしてくださいまして、ありがとうございます！

Answer 2

2つの方法を書いてみました。
まずは、すなおに 2 重の for で。
つぎは、 lists を行列として考えて、それの行・列を 入れ替えて、行ごとに合計集計をするものです。
sums.py

def sum_lists(lists):
    sums = []
    for i in range(len(lists[0])):
        sum = 0
        for row in range(len(lists)):
            sum += lists[row][i]
        sums.append(sum)
    return sums

def sum_lists_x(lists):
    return [sum(ls) for ls in map(list, zip(*lists))]

lists = [
    [1, 6, 3],
    [6, 9, 3],
    [4, 7, 9]
]
print(sum_lists(lists))
print(sum_lists_x(lists))

実行例

参考情報

• Transpose this matrix(python3)

https://stackoverflow.com/questions/23461843/

• 【Python3】sum()でリスト要素の合計を計算する

https://pycarnival.com/sum/

Comments

[snowhawk]

２つも回答を用意していただき、ありがとうございます！