二次元リストの列の合計

Question

Python3で教えてください

二次元リスト　例

[[1, 2, 3], [1, 10, 5], [2, 22, 222], [3, 4, 5], [3, 45, 40]]

で各インデックス0の値毎に、該当インデックス2の合計をリスト

[[1, 8], [2, 222], [3, 45]]

のような結果を得たいのですが、どうすればいいのでしょうか

Answer 1

pandasの取り扱いに慣れていれば、ラクができます。

>>> import pandas as pd
>>> df = pd.DataFrame([[1, 2, 3], [1, 10, 5], [2, 22, 222], [3, 4, 5], [3, 45, 40]])
>>> df.groupby(0)[2].sum()
0
1      8
2    222
3     45
Name: 2, dtype: int64

pandasを使わずただのpython＋標準ライブラリでやるのも、まあ簡単なものです。

>>> lst = [[1, 2, 3], [1, 10, 5], [2, 22, 222], [3, 4, 5], [3, 45, 40]]
>>> from collections import defaultdict
>>> d = defaultdict(list)
>>> for k, _, v in lst:
...     d[k].append(v)
... 
>>> [[k, sum(v)] for k,v in d.items()]
[[1, 8], [2, 222], [3, 45]]  # 外側のリストの順序は保証されないので注意

Comments

[aki777]

早々のご回答ありがとうございます。ゆっくり咀嚼してみます。

Answer 2

参考までに別解をば。

from itertools import groupby
from operator import itemgetter

lst = [[1, 2, 3], [1, 10, 5], [2, 22, 222], [3, 4, 5], [3, 45, 40]] 

dst = []
for k, it in groupby(lst, key=itemgetter(0)):
    total = sum(e[-1] for e in it)
    dst.append([k, total])
    
print(dst)

実行結果 Wandbox

[[1, 8], [2, 222], [3, 45]]

---

若干読みづらくはなりますが、内包表記で書いても良いです。

dst = [
    [k, sum(e[-1] for e in it)]
    for k, it in groupby(lst, key=itemgetter(0))
]

Wandbox

Comments

[aki777]

別解のご回答ありがとうございます。こちらもゆっくり咀嚼してみます。
内包表記まで書いていただき、非常に勉強になりました。