讲师中心 微信公众号

工具推荐

视频效率加速
立刻MV
立刻MV

低成本歌曲 MV、视频生成

Laper
Laper

剧本策划到成片协作

updream

一键生成高质量推广成片

Seko
Seko

AI短剧一键生成成片

如何在两个 DataFrame 之间高效匹配行(基于元素交集)

落强吖_7701

落强吖_7701

发布时间:2026-01-19 19:24:30

|

203人浏览过

|

来源于php中文网

原创

如何在两个 dataframe 之间高效匹配行(基于元素交集)

本文介绍如何在两个大型 DataFrame 之间逐行匹配,找出 df2 中与 df1 中任意一行具有指定数量(如 ≥3)共同元素的行,并返回匹配的完整行及对应索引,适用于多列数值型数据的集合相似性检索。

在实际数据分析中,常需判断一个 DataFrame(如候选集 df2)中是否存在与另一个参考 DataFrame(如查询集 df1)某行“语义相似”的记录——这里相似性定义为行内元素集合的交集大小。例如:df1 每行代表一组关键指标组合,而 df2 是海量样本,需快速筛选出至少共享 3 个以上相同数值的候选行。

原始尝试(仅针对单行):

vals_to_find = set(df1.iloc[0])
mask = df2.loc[:, "Num1":].apply(lambda x: len(vals_to_find.intersection(x)) > 2, axis=1)

该逻辑无法扩展至 df1 全体行,因为它固定使用 iloc[0],未实现“对 df1 中每一行都尝试匹配,并取最大交集数”。

应用程序在线商店宣传网站模板
应用程序在线商店宣传网站模板

应用程序在线商店宣传网站模板是一款提供免费应用程序在线商店宣传网站模板下载。提示:本模板调用到谷歌字体库,可能会出现页面打开比较缓慢。

下载

✅ 正确解法是:对 df2 的每一行,计算其与 df1 所有行的元素交集大小,取最大值;再按阈值(如 >2)过滤。以下是推荐的高效、可读性强的实现:

✅ 推荐方案:基于集合交集的逐行匹配

import pandas as pd

# 构造示例数据(同题)
df1 = pd.DataFrame([[5,10,21],[22,15,7],[6,23,10],[4,34,57]], 
                    columns=['Num1','Num2','Num3'])
df2 = pd.DataFrame([
    [100,1,2,4,5,6,8], [87,1,6,10,22,23,34], [99,1,12,13,34,45,46],
    [64,1,10,14,29,32,33], [55,1,22,13,23,33,35], [66,1,6,7,8,9,10],
    [77,1,2,3,5,6,8], [811,1,2,5,6,8,10], [118,1,7,8,22,44,56],
    [117,1,66,44,47,87,91], [299,2,4,7,20,21,22], [187,3,6,10,12,23,39],
    [199,4,12,24,34,56,57], [264,3,7,8,9,10,33], [50,6,8,10,23,33,35],
    [212,4,6,12,18,19,20], [45,3,7,23,35,56,88], [801,1,2,4,6,28,39],
    [258,2,3,4,9,10,41], [220,5,6,10,27,57,81]
], columns=['Row', 'Num1','Num2','Num3','Num4','Num5','Num6'])

# 核心逻辑:对 df2 每行(仅数值列),计算其与 df1 所有行的最大交集数
def max_intersection_with_df1(row_set):
    return df1.apply(lambda r: len(row_set & set(r)), axis=1).max()

# 提取 df2 数值列(跳过 'Row' 列),转换为 set 后应用函数
numeric_cols = df2.columns[1:]  # ['Num1','Num2',...,'Num6']
mask = df2[numeric_cols].apply(lambda row: max_intersection_with_df1(set(row)), axis=1) > 2

# 获取匹配结果(含原始索引和 Row 值)
result = df2[mask].copy()
print("匹配的行(与 df1 中某行交集 >2):")
print(result)

输出结果:

   Row  Num1  Num2  Num3  Num4  Num5  Num6
1   87     1     6    10    22    23    34
11 187     3     6    10    12    23    39
12 199     4    12    24    34    56    57
14  50     6     8    10    23    33    35
? 验证说明: df2.iloc[1] → [1,6,10,22,23,34] 与 df1.iloc[2] → [6,23,10] 交集为 {6,10,23},大小=3 ✓ df2.iloc[12] → [4,12,24,34,56,57] 与 df1.iloc[3] → [4,34,57] 交集为 {4,34,57},大小=3 ✓

⚠️ 注意事项与优化建议

  • 性能提示:对超大数据集(>10万行),上述 apply + set 方式可能较慢。可考虑向量化替代(如 scikit-learn 的 NearestNeighbors 配合自定义距离)或使用 numba 加速。
  • 列对齐要求:本方案默认 df1 和 df2 的数值列名/顺序一致(如均以 'Num1','Num2','Num3' 开头)。若列结构不一致,请先用 df2[common_cols] 显式选取。
  • 空值处理:set() 会自动忽略 NaN;若需保留 NaN 参与匹配,需预处理(如填充或单独逻辑)。
  • 扩展性:阈值 >2 可替换为变量(如 min_overlap = 3),便于复用。

此方法逻辑清晰、易于调试,是解决“多行集合匹配”类问题的标准实践。

热门AI工具

更多
火山引擎

火山引擎是一款面向企业的云计算与AI服务平台。

蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

Seko
Seko Hot

一款AI视频创作工具,主要用于商汤科技推出的创编一体的AI短视频创作Agent,适合需要提升相关任务效率的用户。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

Atoms
Atoms Hot

Atoms是一款AI智能体工具,第一支自动构建真实业务的 AI 团队。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

函子Zion

一款支持AI搭建、可视化编辑和多端发布的全栈无代码开发平台。

PixTV
PixTV Hot

PixTV是一款面向AIGC内容创作的AI视频生成工具。

相关专题

更多
Postman身份认证怎么配置
Postman身份认证怎么配置

本专题从实际API接口调试场景出发,讲解Postman身份认证的完整配置流程,包括常见认证协议介绍、授权参数设置、访问令牌填写、OAuth 2.0授权及认证失败排查,帮助用户正确完成接口授权与访问验证。

60

2026.11.06

Postman请求头Headers怎么设置
Postman请求头Headers怎么设置

围绕Postman HTTP请求头配置展开,讲解Headers选项的具体使用方式,包括Key和Value参数填写、请求头信息添加、常见Header字段设置及请求头修改方法,帮助用户解决接口调试过程中遇到的请求头配置问题。

40

2026.11.06

Postman接口请求发送教程
Postman接口请求发送教程

详细介绍Postman接口请求的创建与发送方法,涵盖GET和POST请求、URL配置、请求参数、Headers请求头、Body请求体设置以及响应结果查看,帮助用户掌握Postman接口请求的完整操作流程。

40

2026.11.06

DBeaver怎么导入数据
DBeaver怎么导入数据

DBeaver数据导入专题,整理Excel、CSV、SQL、TXT等常见文件的数据导入方法,涵盖目标数据库选择、数据表创建、字段映射、数据类型设置及导入参数配置等操作,帮助用户将本地文件中的数据写入数据库,并了解批量导入、重复数据处理和导入失败等常见问题的解决方法。

60

2026.11.05

DBeaver安全防护教程大全
DBeaver安全防护教程大全

DBeaver安全防护专题,围绕数据库连接安全、账号密码保护和访问权限管理展开,整理SSL/TLS加密连接、SSH隧道、数据库凭据保存及证书验证等配置方法,同时介绍敏感信息保护、远程连接风险和高权限账号使用等注意事项,帮助用户提高数据库日常管理的安全性。

40

2026.11.05

DBeaver连接创建与配置教程
DBeaver连接创建与配置教程

本专题聚焦DBeaver数据库连接配置与管理,覆盖从新建连接、选择数据库类型到填写连接参数和验证连接的完整流程。针对MySQL、PostgreSQL、SQL Server等常见数据库,整理相应的连接设置方法,并提供驱动缺失、认证失败和连接异常等问题的排查方向。

60

2026.11.04

XORM框架结构体映射与表结构同步
XORM框架结构体映射与表结构同步

本专题围绕XORM框架的结构体映射展开,说明表名与字段的命名策略、xorm标签常用属性、主键自增与时间字段自动填充,并演示Sync2自动建表、增量字段升级、索引声明与反向生成结构体的完整流程。

60

2026.11.04

XORM框架安装配置与数据库连接
XORM框架安装配置与数据库连接

本专题聚焦XORM框架的配置与连接环节,拆解MySQL、PostgreSQL、SQLite、SQL Server的DSN写法与驱动选择,讲清连接池参数、多数据源拆分、读写分离与超时重连的处理方式,并补充容器与兼容数据库的适配经验。

40

2026.11.04

XORM框架安装配置与数据库连接
XORM框架安装配置与数据库连接

本专题聚焦XORM框架的配置与连接环节,拆解MySQL、PostgreSQL、SQLite、SQL Server的DSN写法与驱动选择,讲清连接池参数、多数据源拆分、读写分离与超时重连的处理方式,并补充容器与兼容数据库的适配经验。

0

2026.11.04

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn