如何基于多分隔符分割DataFrame列值并按分割结果是否相等分组
解决Pandas多分隔符分割后分组的问题
你之前的代码只判断了Value列是否包含/,这显然没触碰到需求的核心——我们需要先把Value列用**两种分隔符(/和|)**分割成两部分,再判断这两部分内容是否相等,最后给每行分配对应的分组。
正确实现方案
方法1:用str.split()处理多分隔符
Pandas的str.split()支持用正则表达式指定多个分隔符,我们可以用[/|]来匹配/或|,直接分割后取两部分做比较:
import pandas as pd data = { 'Name': ['AVY', 'SCOT', 'JYN', 'MAZ', 'SAS', 'LEN', 'PAUL'], 'Value': ['A/A', 'A|G', 'GG|GG', 'G/GGGG', 'G|GG', 'AA/AA', 'TTT|TTT'] } df = pd.DataFrame(data) # 分割后比较两部分内容,生成Group列 df['Group'] = df['Value'].str.split('[/|]', expand=True).apply(lambda x: 'G1' if x[0] == x[1] else 'G2', axis=1) print(df)
方法2:用str.extract()提取内容后比较
如果觉得分割的方式不够直观,也可以用正则表达式直接提取分隔符前后的内容,再做比较:
import pandas as pd data = { 'Name': ['AVY', 'SCOT', 'JYN', 'MAZ', 'SAS', 'LEN', 'PAUL'], 'Value': ['A/A', 'A|G', 'GG|GG', 'G/GGGG', 'G|GG', 'AA/AA', 'TTT|TTT'] } df = pd.DataFrame(data) # 提取分隔符前后的两个部分到临时列 df[['part1', 'part2']] = df['Value'].str.extract(r'^(.+)[/|](.+)$') # 根据两部分是否相等分配分组 df['Group'] = df.apply(lambda row: 'G1' if row['part1'] == row['part2'] else 'G2', axis=1) # 可选:删除临时列 df = df.drop(['part1', 'part2'], axis=1) print(df)
最终输出结果
两种方法都会得到你预期的结果:
| Name | Value | Group | |
|---|---|---|---|
| 0 | AVY | A/A | G1 |
| 1 | SCOT | A | G |
| 2 | JYN | GG | GG |
| 3 | MAZ | G/GGGG | G2 |
| 4 | SAS | G | GG |
| 5 | LEN | AA/AA | G1 |
| 6 | PAUL | TTT | TTT |
原代码的问题所在
你的代码df.Value.apply(lambda col:'G1' if col.find('/')!=-1 else 'G2')只是简单判断了Value中是否包含/,完全没有涉及分割后的值是否相等的逻辑,所以会错误地把所有含/的行归为G1,含|的归为G2,这和你的需求完全不符。我们的核心逻辑应该是分割后比较两部分内容,而不是判断分隔符的类型。
内容的提问来源于stack exchange,提问作者user2110417
相关产品推荐
相关产品推荐

