python - 解压 DataFrame 的列表元素

Question

我有这个 df:

l1 = ['a', 'b', 'c']
l2 = ['x', ['y1', 'y2', 'y3'], 'z']
df = pd.DataFrame(list(zip(l1, l2)), columns = ['l1', 'l2'])

结果:

  l1            l2
0  a             x
1  b  [y1, y2, y3]
2  c             z

我需要的是解压 l2 中的内部列表并像这样在 l1 中传播相应的值:

  l1  l2
0  a   x
1  b  y1
2  b  y2
3  b  y3
4  c   z

执行此操作的正确方法是什么？谢谢。

Answer 1

您可以使用 itertools.zip_longest 的嵌套列表推导式.

import pandas as pd

from itertools import zip_longest

l1 = ['a', 'b', 'c']
l2 = ['x', ['y1', 'y2', 'y3'], 'z']

expanded = [(left, right) for outer in zip(l1, l2) 
                          for left, right in zip_longest(*outer, fillvalue=outer[0])]

pd.DataFrame(expanded)

结果是……

   0   1
0  a   x
1  b  y1
2  b  y2
3  b  y3
4  c   z

对我来说，这是列表组合太长的边缘。还假定 l1 中没有列表并将进行填充。