在dataframe中格式化字符串

2024-04-25 13:02:09 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个(简化的)数据框,看起来像:

col1 col2
1     a
2     b
3     c,ddd,ee,f,5,hfsf,a

在col2中,我需要能够删除最后2个逗号之后的所有内容,如果没有逗号,请保持值不变:

    col1 col2
1     a
2     b
3     c,ddd,ee

同样,这是简化的,解决方案需要放大到1000行,并且每个逗号之间的空间并不总是相同的

编辑: 这让我走上了正确的轨道

 df.col2 = df.col2.str.split(',').str[:2].str.join(',')