python3 リスト内で同じ要素をもつ数値の合計

Question

リストの操作方法で困っていることがあります。
同じ要素ごとに数値を合計したいと思っています。

[入力]
鈴木さん 2500
加藤さん 2680
武田さん 1200
武田さん 6500

上記のように標準入力から各メンバーの名前とガソリン代を受け取ってこれをメンバーごとにまとめたいと考えてます。

[期待する出力]
武田さん 7700
加藤さん 2680
鈴木さん 2500

というように２つあった武田さんのデータをまとめ、かつガソリン代を降順で並べたいと考えています。

試したこと

[コード]

#メンバー数の分だけ0の入ったリストを用意
result = [0]*3
member = ["鈴木さん","加藤さん","武田さん"] 
 
#
for i in range(3):                       #3 → メンバー数
    for j in range(4):                   #4 → データ数
        if member[i] == data[j][0]:
            result[i] += int(data[j][1])
print(result)

[出力]
[2500,2680,7700]

上記のように数値のみをまとめたリストを作成するところまではできました。

Comments

[mather]

このままでは丸投げの質問なので、まずはご自身で試してみて、具体的にどの部分の書き方がわからないのか、どの部分でエラーがでるのか、を書いてください。

Answer 1

result = [0]*3
member = ["鈴木さん","加藤さん","武田さん"]

としたのでは、 名前や人数が変化したときに対応できません。
もっと汎用的にする必要があります。
data.txt

鈴木さん 2500
加藤さん 2680
武田さん 1200
武田さん 6500

a.py

import sys

result = {}
for line in sys.stdin:
    name, val = line.split()
    result[name] = result.get(name, 0) + int(val)

for name, val in sorted(result.items(), key=lambda x: x[1], reverse=True):
    print(name, "\t", val)

実行例

(名前, val) で dict に保存するようにしていきます。
１行ずつ読み込んだら、 Space で区切って名前と値に分割します。
dict[name] に val を足し算して登録していきます。(val は integer に変換して処理)
名前が登録されていない場合は、 val の値は 0 が得られるようにしておきます。
入力が終了したら、val で降順ソートしてから、各項目を 1 行ずつ整形して print します。

Answer 2

標準入力から各メンバーの名前とガソリン代を受け取ってこれをメンバーごとにまとめたい

名前と金額を紐付ける辞書を用意して、一行ずつ処理していけば良いです。

ガソリン代を降順で並べたい

辞書は原則順序付けできないので、dict#items でタプルのシーケンスを取得してソートします。
わりあい良く使う処理なので、検索すればサンプルは充分得られます。

---

collections.defaultdict や collections.Counter を使ってみても良いでしょう。

Comments

[Alter_9524]

ありがとうございます。
一旦、教えていただいたcollectionsや辞書について勉強してからこの処理に再挑戦しようと思います。