Python3 辞書内の同一keyがあった場合 購入費管理

Question

皆さまいつもお世話になっております。

あるグループ内での社員それぞれの昼食で食べた定食メニューの管理をしたいです。

・1 行目に　グループの人数 N が整数で与えられます。

・2 行目に、半角英小文字で構成される文字列が N 個スペース区切りで与えられます。それぞれの文字列は、グループの人の名前です。

・3 行目に、社員たちが昼食で食べた、定食の合計の数 が整数　M　で与えられます。

・4 行目から続く M 行には、購入した人の名前を表す半角英小文字で構成される文字列 と

　彼らが食べた定食の価格 がこの順にスペース区切りで与えられます。

・入力は合計で M + 3 行となり、入力値最終行の末尾に改行が１つ入ります。

・全ての社員に対して、定食購入費が同一である別の社員は存在しない

入力例1
2
ando noda
2
noda 1000
ando 500

出力例1
noda
ando

入力例2
3
A B C
4
A 1000
B 1000
B 2000
C 2000

出力例2
B
C
A

コード


n = int(input())
name_lst = list(map(str,input().split()))

m = int(input())
data = [list(map(str,input().split())) for _ in range(m)]

dic = {}
for name, amount in data:
    dic[name] = int(amount)

dic_trans = sorted(dic, key=lambda x:x[1], reverse=True)

for x in dic_trans:
    print(x)

#質問

入力例2 だと、同じ人物が二回別の価格の定食を購入している。
これを辞書で表示すると以下のようになる

dic = {A:1000, B:1000, B:2000, C:2000}

Bを一つにまとめて、辞書を作りなおしたいのですが、方法がわかりません。

アドバイス等頂きたいです。宜しくお願い致します。

Answer 1

dic = {A:1000, B:1000, B:2000, C:2000}

同じキーだと上書きされるので、こうはなりません。

>>> d = {}
>>> d["A"] = 1000
>>> d["B"] = 1000
>>> d["B"] = 2000
>>> d["C"] = 2000
>>> d
{'C': 2000, 'A': 1000, 'B': 2000}

ほら、ならない。

けっきょく合計額だけほしいならLouiS0616さんの回答のように加算していくのも手ですが、正攻法で同じキーに複数の値を持たせようとすると、listが値の辞書を作って、appendしていくのが一つの手です。

defaultdictを使ってやると簡単にできます。

from collections import defaultdict

n = int(input())
users = input().split()

m = int(input())

d = defaultdict(list)
for _ in range(m):
    k, v = input().split()
    d[k].append(int(v))

for k,v in sorted([(k, sum(v)) for k,v in d.items()], 
                  key=lambda x:x[1], reverse=True):
    print(k)

Comments

[python3_beginer]

返信遅れて申し訳ありません。

いろんな方法の辞書の作成があるんですね。
教えて頂きありがとうございます。
何度も書いて復習します。

ご指導ありがとうございました。

Answer 2

前以てメンバーのリストが分かっているならば、バリューが0の辞書を作っておけば良いかと。

lunch_dict = {name: 0 for name in name_lst}

そうすれば、あとはデータを読み込んで各要素に足すだけです。

data = [input().split() for _ in range(m)]
for datum in data:
    name, amount = datum
    lunch_dict[name] += int(amount)

Comments

[python3_beginer]

返信遅れて申し訳ありません。

辞書の作成方法を理解していませんでした。
教えて頂きありがとうございます。
これからは辞書の知識を駆使して、スマートにコードを書いていきたいです。

ご指導ありがとうございました。