リスト内の要素の足し算について

Question

以下のリストにおいて、内包リストのインデックス０が同じ数字の間、インデックス１の数字を足し合わせて、新たなリストを作成したい。

num_list = [[1,2],[1,5],[1,4],[2,1],[2,3],[3,6],[3,7]]
....
欲しい出力
new_list = [11,4,13]

上記の場合、左から三番目まで内包リストのインデックス０の値が１であるので、その間のインデックス１の値を足し合わせて、2 + 5 + 4 = 11 となる。そして、得られた結果を新しいリストに追加する。以降も同様に処理していく。

ご回答よろしくお願いします。

Comments

[opyon]

聞き慣れない「内包リスト」というのは文脈から、ネスト＝入れ子にされた２次元配列のことだと思いますが、「内包表記」と混同されていませんか？

[quickquip]

チャレンジして失敗したコードとか、書きかけのコードとか、書きましょう。

[mather]

丸投げの質問です。どこまでご自身で試して、何がわからないのかを質問に書いてください。

[jun68ykt]

こんにちは。回答を書きましたが、一点、要件を確認させてください。たとえば num_list が、 [[1,2],[1,5],[1,4],[3,1],[3,3],[4,6],[4,7]] のときは、 インデクス0の数字は 1, 3, 4 となって、2 が抜けています。 確認したいのは (1)このようなインデクス0の数字に抜けがある場合もあり得ると考えたほうがよいでしょうか？(2) また、あり得る場合、この例でいうと、欲しい結果は、[11, 0, 4,13]　でしょうか？または抜けている数字の合計は結果に入れず、 [11,4,13] が望ましいのでしょうか？

[fuzzball]

インデックス0の値は昇順であることが保証されていますか？

Answer 1

エレガントなコードは他の方に任せて・・・私は愚直にfor文で！
自分用の練習課題として辞書を使った例を追記

辞書を使った例

# k=key
# v=value
# d=dict
# data=[1,2] [1,5] ...
# n=newlist

k = sorted(list(dict(src).keys()))
v = [0 for i in range(len(k))]
d = dict(zip(k, v))
for data in src:
    d[data[0]] += data[1]
n = d.values()
print(n)

for文

# coding=UTF-8
src = [[1,2],[1,5],[1,4],[2,1],[2,3],[3,6],[3,7]]

x = src[0][0]##xyで比較
s = src[0][1]##sum
n = []##newlist

for i in range(len(src)-1):
    y = src[i+1][0]
    if x == y:
        s += src[i+1][1]
    else:
        n.append(s)
        x = y
        s = src[i+1][1]
n.append(s)
print(n)

Comments

[Sh0He1]

非常に簡潔でわかりやすいです。非常に参考になりました。ご回答ありがとうございました。

Answer 2

num_list = [[1, 2], [1, 5], [1, 4], [2, 1], [2, 3], [3, 6], [3, 7]]

result = [sum(d[1] for d in num_list if d[0]==i) for i in range(1,4)]
print(result)
#[11, 4, 13]

Comments

[opyon]

エレガントにサクっと内包表記出来る方が羨ましいです。

[magichan]

普通にfor文を使って実装できるのであれば、あとは書き方を変更するだけなので。
単に慣れの問題かと思います。

[Sh0He1]

ご回答ありがとうございます。内包表記にするとたった1行で収まることに驚きです。

[opyon]

@magichanさんへ
なるほど！言われてみるとfor文の入れ子のように見えてきますね。
地道にfor文書いたり内包表記に書き換えたりして練習してみます。

Answer 3

こんにちは。

defaultdict　を使ってみました。

from collections import defaultdict

num_list = [[1, 2], [1, 5], [1, 4], [2, 1], [2, 3], [3, 6], [3, 7]]

totals = defaultdict(int)

for e in num_list:
    totals[e[0]] += e[1]

result = [total[1] for total in sorted(totals.items())]

print(result)

動作確認用に、上記のコードを以下に上げておきました。

https://trinket.io/python/ad3381b5f6

参考になれば幸いです。

---

追記

上記の回答は、ご質問の文章とコード例

num_list = [[1, 2], [1, 5], [1, 4], [2, 1], [2, 3], [3, 6], [3, 7]]

から判断して、num_list の各要素の配列について、先頭の数字は（上記の例だと、 1, 2, 3 というように、）整数として連続しており、抜けている数字はないという前提での回答になります。

Comments

[quickquip]

0番目の要素が歯抜け(3は表れないが4はある、とか)になる可能性が考慮されていません。

[quickquip]

そういう規則性がありそうだ、という前提ということで了解しました。

[Sh0He1]

丁寧でわかりやすい回答ありがとうございます。リンク参考にさせていただきます。