将熊猫系列中的嵌套列表解压缩到新的DataFrame中

Question

我有一个数据框df，它的column [“ shares”]似乎是列表中的一个列表

0     [[10], [9], [9]]
1     [[3], [3], [2]]
2     [[17], [17], [18]]

如何将此列分为3列：

col1 col2 col3
10   9     9
3    3     2
17   17    18

我尝试过df["shares"].apply(literal_eval) 但是它给我错误的节点或字符串格式错误：

Answer 1

您将需要解压缩您的列表并重新构建您的框架。

cols = ['col1', 'col2', 'col3']
pd.DataFrame(([c[0] for c in r] for r in df.shares.tolist()), columns=cols)

   col1  col2  col3
0    10     9     9
1     3     3     2
2    17    17    18

要推广到具有 3 个以上子列表的列表，您可以使用

pd.DataFrame(
    [c[0] for c in r] for r in df.shares.tolist()
).rename(columns=lambda x: f'col{x+1}')

   col1  col2  col3
0    10     9     9
1     3     3     2
2    17    17    18

Answer 2

展平list的列表，然后使用DataFrame构造函数

import itertools
pd.DataFrame(list(map(lambda x : list(itertools.chain(*x)),df.shares.tolist())))
    0   1   2
0  10   9   9
1   3   3   2
2  17  17  18