我在一个具有不同输出的数据框中有以下列”
col1
MLB|NBA|NFL
MLB|NBA
NFL|NHL|NBA|MLB
我想使用split函数将列按最后一个管道拆分,如下所示:
col1 col2
MLB|NBA NFL
MLB NBA
NFL|NHL|NBA MLB
最佳答案
使用Series.str.rsplit
,限制拆分数。
df.col1.str.rsplit('|', 1, expand=True).rename(lambda x: f'col{x + 1}', axis=1)
如果上面的代码给了你一个语法错误,那就意味着你使用的是3.6之前的python版本(真丢脸!)。请改用
df.col1.str.rsplit('|', 1, expand=True)\
.rename(columns=lambda x: 'col{}'.format(x + 1))
col1 col2
0 MLB|NBA NFL
1 MLB NBA
2 NFL|NHL|NBA MLB
还有更快的loopy
str.rsplit
等价物。pd.DataFrame(
[x.rsplit('|', 1) for x in df.col1.tolist()],
columns=['col1', 'col2']
)
col1 col2
0 MLB|NBA NFL
1 MLB NBA
2 NFL|NHL|NBA MLB
是的,第二种解决方案更快:
df = pd.concat([df] * 100000, ignore_index=True)
%timeit df.col1.str.rsplit('|', 1, expand=True)
%timeit pd.DataFrame([x.rsplit('|', 1) for x in df.col1.tolist()])
473 ms ± 13.8 ms per loop (mean ± std. dev. of 7 runs, 1 loop each)
128 ms ± 1.29 ms per loop (mean ± std. dev. of 7 runs, 10 loops each)