如何将包含列表的列拆分为Dataframe中的不同条目?

时间:2017-10-25 10:44:14

标签: python pandas

我有以下数据框

    A   B   C
1  A1  B1  [C1, C2]
2  A2  B2  [C3, C4]

我希望将其转换为

    A   B   C
1  A1  B1  C1
2  A1  B1  C2
3  A2  B2  C3
4  A2  B2  C4

我该怎么办?感谢

1 个答案:

答案 0 :(得分:0)

一种非常简单的方法如下:

import pandas as pd
df = pd.DataFrame([['A1', 'B1', ['C1', 'C2']],['A2', 'B2', ['C3', 'C4']]], columns = ['A', 'B', 'C'])
df1 = df.copy()
df1['C'] = df['C'].apply(lambda x: x[0])
df2 = df.copy()
df2['C'] = df['C'].apply(lambda x: x[1])
pd.concat([df1, df2]).sort_values('A')