在数据框的列中将列表拆分为多个列

Question

我有一个Pandas DataFrame的列，其中包含多个列表。类似于这样：

df
     col1
0    [[1,2], [2,3]]
1    [[a,b], [4,5], [x,y]] 
2    [[6,7]]

我希望将列表分成多列，以便输出结果类似于：

    col1    col2     col3
0   [1,2]   [2,3]   
1   [a,b]   [4,5]    [x,y]
2   [6,7]

请帮我解决这个问题。非常感谢您的帮助。

- Ronnie

Answer 1

您可以使用pd.Series.apply函数：

df = pd.DataFrame({'col1': [[[1, 2], [2, 3]],
                            [['a', 'b'], [4, 5], ['x', 'y']],
                            [[6, 7]]]})

res = df['col1'].apply(pd.Series)

print(res)

        0       1       2
0  [1, 2]  [2, 3]     NaN
1  [a, b]  [4, 5]  [x, y]
2  [6, 7]     NaN     NaN

- jpp

Answer 2

如果性能很重要，我认为需要使用DataFrame构造函数：

df = pd.DataFrame(df['col1'].values.tolist())
print (df)
        0       1       2
0  [1, 2]  [2, 3]    None
1  [a, b]  [4, 5]  [x, y]
2  [6, 7]    None    None

如果需要删除 NaN - 缺失值，首先添加 dropna：

df = pd.DataFrame(df['col1'].dropna().values.tolist())

- jezrael