

新闻资讯
技术学院本文介绍一种高效、可靠的方法:通过正则提取 + 左连接,将 dataframe `a` 中的名称字段与 data
frame `b` 的 id 字段(含前缀格式如 `"name-123"`)进行子字符串匹配关联,并确保一对多关系正确展开。
在实际数据处理中,常遇到“模糊主键”场景:一个表(如 a)含规范名称("Name"),另一表(如 b)含复合标识符("ID",如 "Boomhaur-2345"),需按名称是否为 ID 的前缀子串建立关联。此时无法直接使用 merge(on=...),但可通过正则表达式精准提取并桥接。
核心思路分两步:
以下是完整可运行代码:
import pandas as pd
# 构造示例数据
a = pd.DataFrame({"Name": ["Boomhaur", "Dale", "Bill", "Hank"]})
b = pd.DataFrame({"ID": ["Boomhaur-2345", "Dale-999999", "Bill-000", "Bill-001", "Peggy-420"]})
# 步骤1:构建正则模式,确保单词边界匹配(避免 'Bill' 错配 'Billy')
names_pattern = r'\b(' + '|'.join(a['Name'].str.replace(r'[^\w]', r'\\$0', regex=True)) + r')\b'
b['_name'] = b["ID"].str.extract(names_pattern)
# 步骤2:左连接,保留 a 全部行,扩展匹配的 ID 及其他字段
result = a.merge(b, how='left', left_on='Name', right_on='_name').drop(columns=['_name'])
print(result)输出结果:
Name ID 0 Boomhaur Boomhaur-2345 1 Dale Dale-999999 2 Bill Bill-000 3 Bill Bill-001 4 Hank NaN
✅ 关键注意事项:
该方案时间复杂度为 O(n·m),适用于中等规模数据;若数据量极大(百万级),可考虑预构建 b 的映射字典或使用 swifter 加速正则提取。