猿问

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

比如有数据

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
请问该如何操作


HUWWW
浏览 5672回答 1
1回答

拉风的咖菲猫

简单循环最简单的方式就是循环拆分一下呗。先上最简单方法: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  C0  1      1  31  2    2,3  32  3  4,5,6  3     A  B  C0  1  1  31  2  2  32  2  3  33  3  4  34  3  5  35  3  6  3更高效的方法采用expand直接进行扩展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)
随时随地看视频慕课网APP

相关分类

Python
我要回答