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

时间:2018-12-16 05:49:29

标签: python pandas dataframe

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

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

如何将此列分为三列:

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

我尝试了df["shares"].apply(literal_eval) 但这给了我错误的节点或字符串格式错误:

2 个答案:

答案 0 :(得分:5)

展开列表列表,然后使用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

答案 1 :(得分:4)

您需要打开列表包装并重新构建框架。

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