如何將帶有值列表的列轉換為 Pandas DataFrame 中的行，也包括前一列

Question

嗨，我有一個這樣的數據框：

    A             B 
0:  a           [[L1, L2]]
1:  b           [[L1, L2, L3]]

我想把它改成：

    A     B                 C 
0:  a    [[L1, L2]]         L1
1:  a    [[L1, L2]]         L2
2:  b    [[L1, L2, L3]]     L1
3:  b    [[L1, L2, L3]]     L2
4:  b    [[L1, L2, L3]]     L3

我怎樣才能做到這一點？

Answer 1

試試這樣的：

import pandas as pd
from io import StringIO
data = """
A B
a [[L1,L2]]
b [[L1,L2,L3]]
"""
df = pd.read_csv(StringIO(data),sep=' ')
df['C']=df['B']
df['C']=df.C.astype(str).replace(['\[','\]', "'", "\s+"], '', regex=True)
print(df.set_index(df.columns.drop('C',1).tolist()).C.str.split(',', expand=True).stack().reset_index().rename(columns={0:'C'}).loc[:, df.columns])

結果 ：

   A             B   C
0  a     [[L1,L2]]  L1
1  a     [[L1,L2]]  L2
2  b  [[L1,L2,L3]]  L1
3  b  [[L1,L2,L3]]  L2
4  b  [[L1,L2,L3]]  L3

Answer 2

使用itertools.chain一種解決方案：

import pandas as pd
from itertools import chain

# old dataframe:
df = pd.DataFrame({'A': ['a', 'b'],
                   'B': [  [['L1', 'L2']], [['L1', 'L2', 'L3']]   ]})

d = {'A':[], 'B':[], 'C': []}
for a, b in zip(df['A'], df['B']):
    for c in chain.from_iterable(b):
        d['A'].append(a)
        d['B'].append(b)
        d['C'].append(c)

# new dataframe:     
df = pd.DataFrame(d)
print(df)

印刷：

   A               B   C
0  a      [[L1, L2]]  L1
1  a      [[L1, L2]]  L2
2  b  [[L1, L2, L3]]  L1
3  b  [[L1, L2, L3]]  L2
4  b  [[L1, L2, L3]]  L3