python - 如何将 Pandas 数据框转换为分层字典

Question

我有以下 Pandas 数据框:

df1 = pd.DataFrame({'date': [200101,200101,200101,200101,200102,200102,200102,200102],'blockcount': [1,1,2,2,1,1,2,2],'reactiontime': [350,400,200,250,100,300,450,400]})

我正在尝试创建一个分层字典，将嵌入字典的值作为列表，如下所示:

{200101: {1:[350, 400], 2:[200, 250]}, 200102: {1:[100, 300], 2:[450, 400]}}

我该怎么做？我得到的最接近的是使用此代码:

df1.set_index('date').groupby(level='date').apply(lambda x: x.set_index('blockcount').squeeze().to_dict()).to_dict()

返回:

{200101: {1: 400, 2: 250}, 200102: {1: 300, 2: 400}}

Answer 1

这是使用 pivot_table 的另一种方法:

d = df1.pivot_table(index='blockcount',columns='date',
     values='reactiontime',aggfunc=list).to_dict()

print(d)

{200101: {1: [350, 400], 2: [200, 250]},
 200102: {1: [100, 300], 2: [450, 400]}}