pandas - 将 Dataframe 变成字典 Pandas

Question

我有一个数据框

Index       Cat        Prod1      Prod2     Prod2
'Pens'   'Writing'     'Red'      'Blue'   'Green'

我试图使用 to.dict()以这种格式放入字典:

dic= {'Pens':['Writing',['Red','Blue','Green']]}

这可能吗？

Answer 1

想法是将前 2 列转换为 MultiIndex然后使用 dict comprehension :

a = df.set_index(['Index','Cat'])

d = {i: [j, list(v)] for (i, j), v in a.T.items()}
print (d)
{'Pens': ['Writing', ['Red', 'Blue', 'Green']]}