pandas - 将 Dataframe 变成字典 Pandas

Question

我有一个数据框

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

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

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

这可能吗？

标签： pandas

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']]}