python - 带有嵌套列表的字典中的 DataFrame

Question

我有一个带有嵌套列表的 python 字典，我想把它变成一个 pandas DataFrame

a = {'A': [1,2,3], 'B':['a','b','c'],'C':[[1,2],[3,4],[5,6]]}

我希望最终的 DataFrame 看起来像这样:

> A  B  C
> 1  a  1
> 1  a  2
> 2  b  3
> 2  b  4
> 3  c  5
> 3  c  6

当我使用 DataFrame 命令时，它看起来像这样:

pd.DataFrame(a)

>   A   B     C
>0  1   a   [1, 2]
>1  2   b   [3, 4]
>2  3   c   [5, 6]

有没有办法用 C 的元素使数据变长？

Answer 1

这是我想出的:

In [53]: df
Out[53]: 
   A  B       C
0  1  a  [1, 2]
1  2  b  [3, 4]
2  3  c  [5, 6]
In [58]: s = df.C.apply(Series).unstack().reset_index(level=0, drop = True)

In [59]: s.name = 'C2'

In [61]: df.drop('C', axis = 1).join(s)
Out[61]: 
   A  B  C2
0  1  a   1
0  1  a   2
1  2  b   3
1  2  b   4
2  3  c   5
2  3  c   6

apply(Series) 给我一个包含两列的 DataFrame。为了在保留原始索引的同时将它们合二为一，我使用了 unstack。 reset_index 删除索引的第一级，它基本上包含 C 中原始列表中的值的索引。然后我将它加入到 df 中。