二次元リスト内で足し算を行いたいです。

Question

前提
以下のようなdataのdata[0][1][2]をそれぞれで足し算したいです。

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

最終的には、
data[0] = 1 + 2 + 3 = 6
data[1] = 4 + 5 + 6 = 15
data[2] = 7 + 8 + 9 =24
data_1 = [6, 15, 24]のような結果を得たいです。

試したこと
以下のサイトを参考に算出することはできているのですが、dataのリストが100個など増えたときに汎用性に欠けるので質問しました。
https://knuth256.com/2021/05/06/python3

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

s = (sum(row) for row in data)
print(next(s))
print(next(s))
print(next(s))

#出力結果
#6
#15
#24

Answer 1

Numpy を使う場合。

import numpy as np

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

data_1 = np.array(data).sum(axis=1).tolist()
print(data_1)

# [6, 15, 24]

Answer 2

data_1 = list(map(sum, data))

参考 組み込み関数 — Python 3.10.4 ドキュメント

Answer 3

以下のようにしてください。

data = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
data_1 = [ sum(x) for x in data ]
print(data_1)

結果
[6, 15, 24]

内包表記を使わないなら

data = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
data_1 = []
for x in data:
    data_1.append(sum(x))

print(data_1)

Comments

[退会済みユーザー]

2つも紹介してくださりありがとうございます。内包表記便利ですね。

Answer 4

data_1 = [sum(row) for row in data]

Comments

[退会済みユーザー]

ありがとうございます。内包表記便利ですね。