python3 リスト ⇒ 辞書

Question

hour=[1,1,1,2,2]
paper=[50,50,90,55,40]
player_ans = {i:j for i,j in zip(hour,paper)}
#{1: 90, 2: 40}
#望む出力{1: 190, 2: 95}

2つのリストを、辞書型にしようと思っています。
その際、値を合計したいのですが、いい方法はありますでしょうか？

リストのインデックスは連動しています。

[1, 1, 1, 2, 2]
↓ ↓ ↓ ↓ ↓
[50,50,90,55,40]

１は50,50,90
２は55,40

よろしくお願いします。

Answer 1

collections.defaultdictが便利です。

from collections import defaultdict

hour = [1, 1, 1, 2, 2]
paper = [50, 50, 90, 55, 40]

player_ans = defaultdict(int)
for h, p in zip(hour, paper):
    player_ans[h] += p

for k, v in player_ans.items():
    print(f'{k}: {v}')

実行結果 Wandbox

1: 190
2: 95

---

collections.Counterを用いた別解。

from collections import Counter

hour = [1, 1, 1, 2, 2]
paper = [50, 50, 90, 55, 40]

player_ans = Counter(
    h for h, p in zip(hour, paper) for _ in range(p)
)

for k, v in player_ans.items():
    print(f'{k}: {v}')

Wandbox

Comments

[zeitaku_fire]

LouiS0616さま、いつもありがとうございます。
・import defaultdict
・player_ans = defaultdict(int)
上記をplayer_ansに定義したことにより、player_ans[h] += pで、計算ができるのだと解釈いたしました。
今後ともよろしくお願いします。

[LouiS0616]

defaultdictを用いることで、次のようなコードを簡潔に表しているのです。

player_ans = {}
for h, p in zip(hour, paper):
　　if h in player_ans:
　　　　player_ans[h] += p
　　else:
　　　　player_and[h] = int() # int()は0を返す

註： 成形のため全角スペースを用いています。

Answer 2

import pandas as pd

hour=[1,1,1,2,2]
paper=[50,50,90,55,40]

df = pd.DataFrame({"hour":hour, "paper":paper})
df1 = df.groupby("hour").sum()
print(df1)

出力結果は少し違いますが、pandasのDataFrameを使ってみました。

Comments

[zeitaku_fire]

おいそがしいとこ、ありがとうございます。
pandasというものを使ったことないのですが、これでも計算ができるのですね。
今後ともよろしくお願いします。

[退会済みユーザー]

参考になってよかったです。こちらこそ、よろしくお願いします。