將包含列表作為值的字典轉換為DataFrame

Question

我有一個像這樣的python字典：

tag_dict = {'001':['apple', 'orange'], '002':['banana', 'strawberry', 'peach']}

我想將此字典轉移到如下所示的DataFrame中：

id  value
001 apple
001 orange
002 banana
002 strawberry
002 peach

有兩列：“ id”和“ value”。

我有兩行ID為001的行，但“值”列中的值不同：“蘋果”和“橙色”。 編號002同樣如此。

Answer 1

如果你想使用DataFrame.from_dict()你可以跟進與stack()然后重置索引：

>>> tag_dict = {'001':['apple','orange'], '002':['banana','strawberry','peach']}
>>> df = pd.DataFrame.from_dict(tag_dict, orient='index').stack()
>>> df.reset_index(level=0)
  level_0           0
0     001       apple
1     001      orange
0     002      banana
1     002  strawberry
2     002       peach

然后，您可以重命名列，並為其提供所需的任何索引：

>>> df.columns = ['id', 'value']
>>> df.set_index('id')
          value
id             
001       apple
001      orange
002      banana
002  strawberry
002       peach