python - 将数据帧的索引设置为字典中的单个键

Question

我有一个数据框，例如:

df = {'index': [0, 0, 0, 0, 0, 1,1,1,1,1, 2,2,2,2], 'value': ['val1', 'val2', 'val3', 'val4', 'val5', 'val6','val7','val8','val9','val10', 'val11','val12','val13','val14']}

我想要一个字典，其中每个索引都将成为我字典中的一个键，这样:key=0 和 values = ['val1', 'val2', 'val3', 'val4', 'val5']

知道怎么做吗？我一直在使用“to_dict”，但看起来我没有做我需要做的事情。

Answer 1

我可以想到类似的东西:

import pandas as pd
df = pd.DataFrame({'index': [0, 0, 0, 0, 0, 1,1,1,1,1, 2,2,2,2], 'value': ['val1', 'val2', 'val3', 'val4', 'val5', 'val6','val7','val8','val9','val10', 'val11','val12','val13','val14']})
df.groupby(by='index').apply(lambda x: list(x['value'])).to_dict()

输出是:

{0: ['val1', 'val2', 'val3', 'val4', 'val5'],
 1: ['val6', 'val7', 'val8', 'val9', 'val10'],
 2: ['val11', 'val12', 'val13', 'val14']}