使用值作为键将数据帧转换为字典

Question

我有以下数据框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()，但字典中的条目格式不正确。我怎样才能做到这一点？

Answer 1

让我们尝试一下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']}