python - 将包含列表作为值的字典转换为 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，但“值”列中的值不同:“苹果”和“橙色”。 id 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