嗨，我正在努力解开这个Pandas 的嵌套列表.有人能帮帮忙吗？
这是数据帧
df = pd.DataFrame({"a" : [1,2], "b" : [['a','b',['c','d']],['a',['b','c']]]})
我需要有多余的行，方法是只出栈列表中的列表，而不是整个列表.
输出数据帧应如下所示
df2 = pd.DataFrame({"a" : [1,1,2,2], "b" : [['a','b','c'],['a','b','d'],['a','b'],['a','c']]})
提前谢谢你
您可以使用itertools.product和explode:
from itertools import product
out = (df
.assign(b=[list(map(list, product(*(x if isinstance(x, list) else [x]
for x in l))))
for l in df['b']])
.explode('b')
)
输出:
a b
0 1 [a, b, c]
0 1 [a, b, d]
1 2 [a, b]
1 2 [a, c]
其他示例:
a b
0 1 [Burger, Fries]
0 1 [Burger, Steak]
0 1 [Burger, Chicken]
0 1 [Burger, Salad]
1 2 [Fries, Chicken, Steak]
1 2 [Fries, Salad, Steak]
1 2 [Fries, Burger, Steak]
2 3 [Burger, Chicken, Fries, Salad, Steak]