当列表类型列在熊猫数据框中具有多个值时创建行

Question

我有一个这样的数据框，

df
col1    col2 (the type is list) 
 1      [a]
 2      [a,b]
 3      [b]
 4      [a,b,c]

这里 col1 的值是唯一的。现在我想创建更多的行，其中列表的长度大于 1，并且在这些情况下 col1 值将相同。所以最终的数据框应该是这样的，

df
col1    col2
 1       [a]
 2       [a]
 2       [b]
 3       [b]
 4       [a]
 4       [b]
 4       [c]

我可以使用 for 循环来执行此操作并遍历列表的长度，但这需要更多时间，因此请寻找一些 pandas 捷径来更有效地完成此操作。

Answer 1

首先使用DataFrame.explode，然后对一个元素列表使用 lambda 函数：

df = df.explode('col2')
df['col2'] = df['col2'].apply(lambda x: [x])