0

0

计算 DataFrame 中球员之间的历史交手记录(Head-to-Head)

花韻仙語

花韻仙語

发布时间:2025-12-30 14:29:17

|

197人浏览过

|

来源于php中文网

原创

计算 DataFrame 中球员之间的历史交手记录(Head-to-Head)

本文介绍如何基于比赛时间顺序,为每场网球对战动态计算双方球员的历史胜负次数,确保无论谁作为 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=int)

    # 遍历每场比赛(从第 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)

✅ 输出验证(与期望一致)

输入示例:

Proface Avatarize
Proface Avatarize

一个利用AI技术提供高质量专业头像和头像的工具

下载
data = {
    'tourney_date': ['2012-01-16', '2012-01-27', '2012-03-14', '2015-01-20', '2020-10-07', '2020-10-15', '2020-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 特征。

相关专题

更多
lambda表达式
lambda表达式

Lambda表达式是一种匿名函数的简洁表示方式,它可以在需要函数作为参数的地方使用,并提供了一种更简洁、更灵活的编码方式,其语法为“lambda 参数列表: 表达式”,参数列表是函数的参数,可以包含一个或多个参数,用逗号分隔,表达式是函数的执行体,用于定义函数的具体操作。本专题为大家提供lambda表达式相关的文章、下载、课程内容,供大家免费下载体验。

202

2023.09.15

python lambda函数
python lambda函数

本专题整合了python lambda函数用法详解,阅读专题下面的文章了解更多详细内容。

187

2025.11.08

excel制作动态图表教程
excel制作动态图表教程

本专题整合了excel制作动态图表相关教程,阅读专题下面的文章了解更多详细教程。

24

2025.12.29

freeok看剧入口合集
freeok看剧入口合集

本专题整合了freeok看剧入口网址,阅读下面的文章了解更多网址。

74

2025.12.29

俄罗斯搜索引擎Yandex最新官方入口网址
俄罗斯搜索引擎Yandex最新官方入口网址

Yandex官方入口网址是https://yandex.com;用户可通过网页端直连或移动端浏览器直接访问,无需登录即可使用搜索、图片、新闻、地图等全部基础功能,并支持多语种检索与静态资源精准筛选。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

207

2025.12.29

python中def的用法大全
python中def的用法大全

def关键字用于在Python中定义函数。其基本语法包括函数名、参数列表、文档字符串和返回值。使用def可以定义无参数、单参数、多参数、默认参数和可变参数的函数。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

16

2025.12.29

python改成中文版教程大全
python改成中文版教程大全

Python界面可通过以下方法改为中文版:修改系统语言环境:更改系统语言为“中文(简体)”。使用 IDE 修改:在 PyCharm 等 IDE 中更改语言设置为“中文”。使用 IDLE 修改:在 IDLE 中修改语言为“Chinese”。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

18

2025.12.29

C++的Top K问题怎么解决
C++的Top K问题怎么解决

TopK问题可通过优先队列、partial_sort和nth_element解决:优先队列维护大小为K的堆,适合流式数据;partial_sort对前K个元素排序,适用于需有序结果且K较小的场景;nth_element基于快速选择,平均时间复杂度O(n),效率最高但不保证前K内部有序。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

12

2025.12.29

php8.4实现接口限流的教程
php8.4实现接口限流的教程

PHP8.4本身不内置限流功能,需借助Redis(令牌桶)或Swoole(漏桶)实现;文件锁因I/O瓶颈、无跨机共享、秒级精度等缺陷不适用高并发场景。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

136

2025.12.29

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
CSS3 教程
CSS3 教程

共18课时 | 4.1万人学习

PostgreSQL 教程
PostgreSQL 教程

共48课时 | 6.2万人学习

Django 教程
Django 教程

共28课时 | 2.6万人学习

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

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号