这是我的数据框(有更多的字母和~35.5k的长度)和其他相关字符串的东西。所有变量都是字符串,[' C1',' C2']是MultiIndex。
tmp
C1 C2 C3 C4 C5 Start End C8
A 1 - - - 12 14 -
A 2 - - - 1,4,7 3,6,10 -
A 3 - - - 16,19 17,21 -
A 4 - - - 22 24 -
我需要它成为这个(分隔包含逗号的每一行保留其他所有内容):
C1 C2 C3 C4 C5 Start End C8 Appearance
A 1 - - - 12 14 - 1
A 2 - - - 1 3 - 1
A 2 - - - 4 6 - 2
A 2 - - - 7 10 - 3
A 3 - - - 16 17 - 1
A 3 - - - 19 21 - 2
A 4 - - - 22 24 - 1
我试过这个脚本 pandas: How do I split text in a column into multiple rows?
作为
s = tmp['Start'].str.split(',').apply(Series, 1).stack()
s.index = s.index.droplevel(-1)
s.name = 'Start
del tmp['Start']
final = tmp.join(s)
但结果却比它应该大得多!我得到了数以千计的重复,这只是试图分裂'开始'。我甚至无法想象尝试为开始和结束这样做(每个逗号都在'开始'隐含逗号'结束'。
Lengths:
tmp = 35568
s = 35676
final = 293408
答案 0 :(得分:3)
您可以从df
和s1
以及join
创建新的s2
。另外更好的是在str.split
中使用参数expand=True
并按drop
删除多个列:
要创建列Appearance
,请groupby
使用index
cumcount
。{/ p>
s1 = tmp['Start'].str.split(',', expand=True).stack()
s1.index = s1.index.droplevel(-1)
s1.name = 'Start'
s2 = tmp['End'].str.split(',', expand=True).stack()
s2.index = s2.index.droplevel(-1)
s2.name = 'End'
tmp.drop(['Start', 'End'], inplace=True, axis=1)
df = pd.DataFrame({'s1':s1, 's2':s2}, index=s1.index)
final = tmp.join(df)
final['Appearance'] = final.groupby(final.index).cumcount() + 1
print (final)
C1 C2 C3 C4 C5 C8 s1 s2 Appearance
0 A 1 - - - - 12 14 1
1 A 2 - - - - 1 3 1
1 A 2 - - - - 4 6 2
1 A 2 - - - - 7 10 3
2 A 3 - - - - 16 17 1
2 A 3 - - - - 19 21 2
3 A 4 - - - - 22 24 1
通过评论编辑:
您可以先尝试reset_index
:
print (tmp)
C3 C4 C5 Start End C8
C1 C2
A 1 - - - 12 14 -
2 - - - 1,4,7 3,6,10 -
3 - - - 16,19 17,21 -
4 - - - 22 24 -
tmp.reset_index(inplace=True)
print (tmp)
C1 C2 C3 C4 C5 Start End C8
0 A 1 - - - 12 14 -
1 A 2 - - - 1,4,7 3,6,10 -
2 A 3 - - - 16,19 17,21 -
3 A 4 - - - 22 24 -
答案 1 :(得分:2)
我将展开的'Start'
和'End'
列连接起来,以确保它们匹配,即使它们没有相同数量的条目。
s = tmp.Start.str.split(',', expand=True).stack().rename('Start')
e = tmp.End.str.split(',', expand=True).stack().rename('End')
se = pd.concat([s, e], axis=1).reset_index(1, drop=True)
tmp.drop(['Start', 'End'], axis=1).merge(se, left_index=True, right_index=True)