將DataFrame轉換為字典包含列表

Question

堆疊團隊！ 我有一個數據框，我想將其轉換成字典，其中一列作為鍵，另一列的值附加到字典中的列表中。

我已經能夠將數據幀轉換為具有數據幀每一行鍵值對的字典，但是我只需要在屬於每個字典鍵的序列后附加一個序列或列表即可。

DataFrame:

   Team Name      ID
0  Idaho      1234
1  Idaho      5678
2  Tokyo      5432

嘗試了這個，我得到了包含字典的字典。 將ID設置為鍵，並將數據幀中的行記錄作為值。

代碼嘗試：

team_id_df.set_index('Team Name').to_dict(orient='index')

結果：

{'Idaho': {'ID': 1234, 'ID': 5678}, 'Tokyo': { 'ID': 5432}}

但是我試圖得到這個：

{'Idaho': [1234, 5678],'Tokyo': [5432]}

我有些困惑，對進行此轉換有什么建議嗎？

Answer 1

我將分兩步完成。 首先，我將使用group by，然后使用for循環，我將得到想要的結果，遍歷結果。

b = a.groupby(by=['Team Name','ID']).agg('count')
final_dict = {}
for team_name, ID in b.index:
    final_dict.setdefault(team_name,[]).append(ID)
print(final_dict)

這給了我以下輸出。