鍍金池/ 問答/數(shù)據(jù)分析&挖掘  Python/ python pandas dataframe拆分行(某些行一行拆成多行)

python pandas dataframe拆分行(某些行一行拆成多行)

比如有數(shù)據(jù)

A B C
1 1,2 3
2 3 3
3 4,5,6 3

這樣一張表,要按B將他拆分為:
A B C
1 1 3
1 2 3
2 3 3
3 4 3
1 5 3
1 6 3
請(qǐng)問該如何操作

回答
編輯回答
笨尐豬

簡(jiǎn)單循環(huán)

最簡(jiǎn)單的方式就是循環(huán)拆分一下唄。先上最簡(jiǎn)單方法:

import pandas as pd
df = pd.DataFrame({'A':['1','2','3'],'B':['1','2,3','4,5,6'],'C':['3','3','3']})
result = pd.DataFrame(columns=['A','B','C'])
print(df,'\n')
for i in df.itertuples():
    for j in i[2].split(','):
        result = result.append({'A':i[1],'B':j,'C':i[3]},ignore_index=True)        
print(result)

輸出:

   A      B  C
0  1      1  3
1  2    2,3  3
2  3  4,5,6  3 

   A  B  C
0  1  1  3
1  2  2  3
2  2  3  3
3  3  4  3
4  3  5  3
5  3  6  3

更高效的方法

采用expand直接進(jìn)行擴(kuò)展

df = pd.DataFrame({'A':['1','2','3'],'B':['1','2,3','4,5,6'],'C':['3','3','3']})
df = (df.set_index(['A','C'])['B']
       .str.split(',', expand=True)
       .stack()
       .reset_index(level=2, drop=True)
       .reset_index(name='B'))
print(df)
2017年12月12日 04:14