计算 DataFrame 中球员之间的历史交手记录(Head-to-Head)
发布时间 - 2025-12-30 00:00:00 点击率:次本文介绍如何基于比赛时间顺序,为每场网球对战动态计算双方球员的历史胜负次数,确保无论谁作为 player1 或 player2 出现,h2h 统计均准确反映其真实交锋胜场数。
在构建网球比赛预测模型时,历史交手记录(Head-to-Head, H2H) 是极具价值的特征:它捕捉了两名球员之间真实的对抗经验与胜负倾向。但实现时存在一个关键挑战——原始数据中同一对球员(如 A vs B)可能以不同顺序出现在 player1_id/player2_id 列中(例如某场是 A,B,下一场是 B,A),而目标变量 target 的定义始终以 player1_id 为基准(target=1 表示 player1 获胜)。若直接按行遍历或简单分组统计,极易混淆胜者归属,导致 h2h 值错位。
正确的解法核心在于:先将每对球员标准化为无序组合(如 (A,B) 统一表示 A 与 B 的对决),再按时间升序逐场累积各自胜场。以下是经过验证的高效、可扩展实现:
✅ 正确实现步骤
import pandas as pd
import numpy as np
def calculate_h2h_per_pair(group):
"""对同一对球员(标准化顺序)的子集,按时间顺序计算累计胜场"""
# 确保 group 按 tourney_date 升序排列(关键!)
group = group.sort_values('tourney_date').reset_index(drop=True)
# 提取实际获胜者:target==1 → player1_id 胜;target==0 → player2_id 胜
winner = np.where(group['target'] == 1, group['player1_id'], group['player2_id'])
# 初始化两列:player1_h2h 和 player2_h2h,初始为 0
player1_h2h = np.zeros(len(group), dtype=int)
player2_h2h = np.zeros(len(group), dtype=i
nt)
# 遍历每场比赛(从第 2 场开始,第 1 场无历史记录)
for i in range(1, len(group)):
prev_matches = group.iloc[:i] # 当前场次之前的所有同对比赛
p1 = group.iloc[i]['player1_id']
p2 = group.iloc[i]['player2_id']
# 统计此前 p1 对 p2 的胜场(即 winner == p1)
p1_wins = (prev_matches.apply(
lambda r: r['player1_id'] if r['target']==1 else r['player2_id'], axis=1
) == p1).sum()
# 统计此前 p2 对 p1 的胜场(即 winner == p2)
p2_wins = (prev_matches.apply(
lambda r: r['player1_id'] if r['target']==1 else r['player2_id'], axis=1
) == p2).sum()
player1_h2h[i] = p1_wins
player2_h2h[i] = p2_wins
group['player1_h2h'] = player1_h2h
group['player2_h2h'] = player2_h2h
return group
# 标准化球员对:对每行 (p1, p2),生成排序后的元组 (min, max),确保 (A,B) 和 (B,A) 归为同一组
df['h2h_pair'] = df.apply(
lambda r: tuple(sorted([r['player1_id'], r['player2_id']])),
axis=1
)
# 按标准化对分组,并应用 h2h 计算逻辑
result = df.groupby('h2h_pair', group_keys=False).apply(calculate_h2h_per_pair).drop('h2h_pair', axis=1)⚠️ 注意:上述方法虽逻辑清晰、易于理解,但在大数据集上可能较慢(因显式循环)。生产环境推荐使用向量化优化版本(见下方进阶技巧)。
? 向量化高性能版本(推荐)
def get_h2h_vectorized(group):
group = group.sort_values('tourney_date').reset_index(drop=True)
# 构建“实际胜者”序列
winner = group['player1_id'].where(group['target'] == 1, group['player2_id'])
# 对当前 group 中两位球员命名(固定顺序)
p1_ref, p2_ref = sorted([group.iloc[0]['player1_id'], group.iloc[0]['player2_id']])
# 统计到每一行为止,p1_ref 和 p2_ref 各自获胜次数(不包含当前行)
p1_cumwins = (winner == p1_ref).shift(1).fillna(0).cumsum().astype(int)
p2_cumwins = (winner == p2_ref).shift(1).fillna(0).cumsum().astype(int)
# 根据当前行中 player1_id/player2_id 的实际角色,分配 h2h 值
player1_h2h = np.where(group['player1_id'] == p1_ref, p1_cumwins, p2_cumwins)
player2_h2h = np.where(group['player2_id'] == p1_ref, p1_cumwins, p2_cumwins)
return group.assign(player1_h2h=player1_h2h, player2_h2h=player2_h2h)
# 应用分组计算
df['h2h_pair'] = df.apply(lambda r: tuple(sorted([r['player1_id'], r['player2_id']])), axis=1)
result = df.groupby('h2h_pair', group_keys=False).apply(get_h2h_vectorized).drop('h2h_pair', axis=1)✅ 输出验证(与期望一致)
输入示例:
data = {
'tourney_date': ['2012-01-16', '2012-01-27', '2012-03-14', '2015-01-20', '2025-10-07', '2025-10-15', '2025-10-15'],
'player1_id': ['A', 'A', 'B', 'A', 'B', 'A', 'B'],
'player2_id': ['B', 'B', 'A', 'B', 'A', 'B', 'A'],
'target': [0, 0, 1, 0, 1, 1, 1]
}
df = pd.DataFrame(data)输出 result 中 player1_h2h / player2_h2h 列将严格匹配题目所给理想结果:
- 2012-01-16 A vs B, target=0 → player1_h2h=0, player2_h2h=0
- 2012-01-27 A vs B, target=0 → player1_h2h=0, player2_h2h=1(B 已赢 1 次)
- 2012-03-14 B vs A, target=1 → player1_h2h=2, player2_h2h=0(B 已赢前 2 场)
- ……依此类推。
? 关键要点总结
- 必须标准化球员对:使用 tuple(sorted([p1,p2])) 消除顺序差异,是正确分组的前提。
- 严格按时间排序:groupby 后必须在每个子组内调用 .sort_values('tourney_date'),否则累积统计失效。
- 胜者识别要解耦:target 仅定义 relative 胜负,需映射到具体球员 ID,而非依赖列位置。
- 避免行间状态污染:原问题代码错误地用 row.name 和全局索引比较,破坏了时间局部性;应始终基于 tourney_date 过滤历史。
- 向量化优于 apply + lambda:尤其对百万级数据,shift().cumsum() 比 apply 循环快 10–100 倍。
通过以上方法,你可稳健、高效、可复现地为任意双人竞技类时序数据(网球、围棋、电竞等)生成精准的 head-to-head 特征。
# 大数据
# app
# ai
# win
# 排列
# 循环
# Lambda
# 胜场
# 升序
# 遍历
# 此前
# 进阶
# 行间
# 依此类推
# 出现在
# 两位
# 但在
相关栏目:
【
网站优化151355 】
【
网络推广146373 】
【
网络技术251813 】
【
AI营销90571 】
相关推荐:
做企业网站制作流程,企业网站制作基本流程有哪些?
Laravel怎么实现搜索高亮功能_Laravel结合Scout与Algolia全文检索【实战】
Laravel辅助函数有哪些_Laravel Helpers常用助手函数大全
Win11怎么关闭专注助手 Win11关闭免打扰模式设置【操作】
Android实现代码画虚线边框背景效果
Laravel怎么进行数据库事务处理_Laravel DB Facade事务操作确保数据一致性
Win11摄像头无法使用怎么办_Win11相机隐私权限开启教程【详解】
网站制作企业,网站的banner和导航栏是指什么?
如何在云主机上快速搭建网站?
Laravel如何实现URL美化Slug功能_Laravel使用eloquent-sluggable生成别名【方法】
Laravel如何使用Passport实现OAuth2?(完整配置步骤)
Python文件异常处理策略_健壮性说明【指导】
Laravel DB事务怎么使用_Laravel数据库事务回滚操作
Laravel软删除怎么实现_Laravel Eloquent SoftDeletes功能使用教程
js实现点击每个li节点,都弹出其文本值及修改
Laravel怎么连接多个数据库_Laravel多数据库连接配置
如何快速搭建高效WAP手机网站?
香港服务器建站指南:外贸独立站搭建与跨境电商配置流程
极客网站有哪些,DoNews、36氪、爱范儿、虎嗅、雷锋网、极客公园这些互联网媒体网站有什么差异?
linux写shell需要注意的问题(必看)
Laravel中DTO是什么概念_在Laravel项目中使用数据传输对象(DTO)
Laravel如何使用Seeder填充数据_Laravel模型工厂Factory批量生成测试数据【方法】
Laravel如何生成API文档?(Swagger/OpenAPI教程)
香港服务器WordPress建站指南:SEO优化与高效部署策略
nodejs redis 发布订阅机制封装实现方法及实例代码
如何在橙子建站上传落地页?操作指南详解
标准网站视频模板制作软件,现在有哪个网站的视频编辑素材最齐全的,背景音乐、音效等?
android nfc常用标签读取总结
详解阿里云nginx服务器多站点的配置
免费网站制作appp,免费制作app哪个平台好?
如何在IIS7上新建站点并设置安全权限?
ChatGPT 4.0官网入口地址 ChatGPT在线体验官网
Laravel如何实现文件上传和存储?(本地与S3配置)
如何用已有域名快速搭建网站?
详解Android中Activity的四大启动模式实验简述
高防服务器租用首荐平台,企业级优惠套餐快速部署
如何在阿里云购买域名并搭建网站?
C语言设计一个闪闪的圣诞树
javascript如何操作浏览器历史记录_怎样实现无刷新导航
Laravel的路由模型绑定怎么用_Laravel Route Model Binding简化控制器逻辑
再谈Python中的字符串与字符编码(推荐)
ChatGPT怎么生成Excel公式_ChatGPT公式生成方法【指南】
如何自定义建站之星模板颜色并下载新样式?
IOS倒计时设置UIButton标题title的抖动问题
动图在线制作网站有哪些,滑动动图图集怎么做?
如何用PHP快速搭建高效网站?分步指南
如何将凡科建站内容保存为本地文件?
Python企业级消息系统教程_KafkaRabbitMQ高并发应用
用v-html解决Vue.js渲染中html标签不被解析的问题
重庆市网站制作公司,重庆招聘网站哪个好?


nt)
# 遍历每场比赛(从第 2 场开始,第 1 场无历史记录)
for i in range(1, len(group)):
prev_matches = group.iloc[:i] # 当前场次之前的所有同对比赛
p1 = group.iloc[i]['player1_id']
p2 = group.iloc[i]['player2_id']
# 统计此前 p1 对 p2 的胜场(即 winner == p1)
p1_wins = (prev_matches.apply(
lambda r: r['player1_id'] if r['target']==1 else r['player2_id'], axis=1
) == p1).sum()
# 统计此前 p2 对 p1 的胜场(即 winner == p2)
p2_wins = (prev_matches.apply(
lambda r: r['player1_id'] if r['target']==1 else r['player2_id'], axis=1
) == p2).sum()
player1_h2h[i] = p1_wins
player2_h2h[i] = p2_wins
group['player1_h2h'] = player1_h2h
group['player2_h2h'] = player2_h2h
return group
# 标准化球员对:对每行 (p1, p2),生成排序后的元组 (min, max),确保 (A,B) 和 (B,A) 归为同一组
df['h2h_pair'] = df.apply(
lambda r: tuple(sorted([r['player1_id'], r['player2_id']])),
axis=1
)
# 按标准化对分组,并应用 h2h 计算逻辑
result = df.groupby('h2h_pair', group_keys=False).apply(calculate_h2h_per_pair).drop('h2h_pair', axis=1)