我有以下數據框import pandas as pdd = {'value': ['value1', 'value2', 'value3'], 'key': ['key1', 'key1', 'key2']}df = pd.DataFrame(data=d)我需要將其轉換為以下形式的字典{'key1': ['value1', 'value2'], 'key2': ['value2']}我嘗試過df.to_dict()，但字典中的條目格式不正確。我怎樣才能做到這一點？
1 回答
夢里花落0921
TA貢獻1772條經驗 獲得超6個贊
讓我們嘗試一下groupby：
df.groupby('key')['value'].agg(list).to_dict()
# run time
# 897 μs ± 50.2 μs per loop (mean ± std. dev. of 7 runs, 100 loops each)
或者：
{k:list(x) for k,x in df.groupby('key')['value']}
# run time
# 376 μs ± 14.8 μs per loop (mean ± std. dev. of 7 runs, 100 loops each)
輸出：
{'key1': ['value1', 'value2'], 'key2': ['value3']}
添加回答
舉報
0/150
提交
取消