[英]iterate over df and output dictionary with list of values
我需要將數據幀轉換為字典,但無法從數據框中獲取所有值以顯示在字典中。
數據幀:
id| region | Num |
--|--------|-----|
2 | NYC |2344 |
3 | NYC |3243 |
4 | NYC |3253 |
5 | NYC |2345 |
6 | CHI |8756 |
7 | CHI |9786 |
8 | CHI |7674 |
9 | CHI |6678 |
10| ATL |1234 |
碼:
df.set_index('region').T.to_dict('list')
:
我需要的是這個:
{'NYC: [2344, 3243, 3253, 2345 ], 'CHI': [8756, 9786, 7674, 6678], 'ATL': [1234] }
但我得到的是這個:
{'NYC: [2345 ], 'CHI': [6678], 'ATL': [1234] }
我試過了:
num_dict = {}
for region, num in df:
num_dict.setdefault(region, []).append(num)
但是這給了我一個(ValueError: too many values to unpack (expected 2)
。
有一個更好的方法嗎?
提前致謝!
使用groupby.apply(list)
然后使用.to_dict
df.groupby('region')['Num'].apply(list).to_dict()
[OUT]
{'ATL': [1234],
'CHI': [8756, 9786, 7674, 6678],
'NYC': [2344, 3243, 3253, 2345]}
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.