python案例如何预测杯赛决赛的紧张程度？

wen python案例 2026-08-05 250

本文目录导读：

🎯 主要预测因素
📊 评分系统
💡 应用场景

我来设计一个有趣的Python案例来预测足球杯赛决赛的紧张程度,这个模型会综合考虑多个因素：

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from sklearn.preprocessing import StandardScaler
from sklearn.ensemble import RandomForestRegressor
import warnings
warnings.filterwarnings('ignore')
# 设置中文显示
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False
class CupFinalTensionPredictor:
    """
    杯赛决赛紧张程度预测器
    综合考虑比赛因素、球队实力、历史元素等
    """
    def __init__(self):
        # 定义紧张程度评分等级
        self.tension_levels = {
            0: "非常轻松",
            1: "比较轻松", 
            2: "适中",
            3: "比较紧张",
            4: "非常紧张",
            5: "极端紧张"
        }
    def calculate_team_strength(self, team_stats):
        """
        计算球队综合实力
        team_stats: 包含球队各项统计指标的字典
        """
        # 权重分配
        weights = {
            'fifa_rank': 0.3,        # FIFA排名
            'goals_per_game': 0.25,   # 场均进球
            'defense_strength': 0.2,   # 防守强度(失球率)
            'possession': 0.15,        # 控球率
            'recent_form': 0.1         # 近期状态
        }
        strength_score = 0
        for key, weight in weights.items():
            if key == 'fifa_rank':
                # FIFA排名越低越好，需要反向处理
                score = (100 - team_stats.get(key, 50)) / 100
            elif key == 'defense_strength':
                # 失球率越低越好，反向处理
                score = 1 - min(team_stats.get(key, 0.5), 1)
            else:
                score = min(team_stats.get(key, 0), 1)
            strength_score += score * weight
        return strength_score
    def calculate_match_importance(self, match_context):
        """
        计算比赛重要程度
        """
        importance_factors = {
            'is_final': 0.4,           # 是否是决赛
            'rivalry_level': 0.3,      # 宿敌程度
            'tournament_level': 0.2,   # 赛事级别
            'history_weight': 0.1      # 历史渊源重要性
        }
        importance_score = 0
        for factor, weight in importance_factors.items():
            score = min(match_context.get(factor, 0), 1)
            importance_score += score * weight
        return importance_score
    def calculate_historical_tension(self, match_history):
        """
        根据历史交锋数据计算紧张程度
        """
        if not match_history or len(match_history) == 0:
            return 0.5  # 默认中等紧张
        # 计算历史交锋的平均分差
        avg_goal_diff = np.mean([m['goal_diff'] for m in match_history])
        # 计算历史红黄牌数量
        avg_cards = np.mean([m['cards'] for m in match_history])
        # 计算历史进球数
        avg_goals = np.mean([m['total_goals'] for m in match_history])
        # 转化为0-1的打分
        tension_from_diff = 1 - min(abs(avg_goal_diff), 3) / 3  # 分差越小越紧张
        tension_from_cards = min(avg_cards, 10) / 10            # 红黄牌越多越紧张
        tension_from_goals = min(avg_goals, 6) / 6             # 进球越多越紧张
        historical_tension = (tension_from_diff * 0.4 + 
                            tension_from_cards * 0.3 + 
                            tension_from_goals * 0.3)
        return historical_tension
    def predict_tension(self, team_a_stats, team_b_stats, match_context, match_history=[]):
        """
        综合预测比赛紧张程度
        参数:
        - team_a_stats: 主队统计
        - team_b_stats: 客队统计
        - match_context: 比赛背景
        - match_history: 历史交锋记录
        """
        # 1. 计算两队实力
        team_a_strength = self.calculate_team_strength(team_a_stats)
        team_b_strength = self.calculate_team_strength(team_b_stats)
        # 2. 计算实力差距 (实力越接近越紧张)
        strength_gap = abs(team_a_strength - team_b_strength)
        tension_from_gap = 1 - min(strength_gap, 0.5) / 0.5
        # 3. 计算比赛重要程度
        match_importance = self.calculate_match_importance(match_context)
        # 4. 计算历史交锋紧张程度
        historical_tension = self.calculate_historical_tension(match_history)
        # 5. 综合计算最终紧张程度 (0-1)
        weights = {
            'strength_gap': 0.25,      # 实力差距权重
            'importance': 0.40,        # 比赛重要性权重
            'history': 0.15,           # 历史交锋权重
            'stake': 0.20              # 比赛结果影响权重
        }
        # 计算比赛结果影响(胜利的收益越大越紧张)
        stake = min(match_context.get('champion_prize', 0.7), 1)
        final_tension = (weights['strength_gap'] * tension_from_gap +
                        weights['importance'] * match_importance +
                        weights['history'] * historical_tension +
                        weights['stake'] * stake)
        # 转换为0-5的评分
        tension_score = round(final_tension * 5, 2)
        # 确定紧张程度等级
        tension_level = int(tension_score)
        if tension_level >= 5:
            tension_level = 5
        return {
            'tension_score': tension_score,
            'tension_level': tension_level,
            'tension_description': self.tension_levels[tension_level],
            'details': {
                'team_a_strength': round(team_a_strength, 3),
                'team_b_strength': round(team_b_strength, 3),
                'strength_gap': round(strength_gap, 3),
                'match_importance': round(match_importance, 3),
                'historical_tension': round(historical_tension, 3),
                'stake_level': round(stake, 3)
            }
        }
# 示例使用
predictor = CupFinalTensionPredictor()
# 模拟一场比赛数据
team_a_stats = {
    'fifa_rank': 12,
    'goals_per_game': 0.85,
    'defense_strength': 0.15,  # 场均失球率
    'possession': 0.58,
    'recent_form': 0.75
}
team_b_stats = {
    'fifa_rank': 8,
    'goals_per_game': 0.92,
    'defense_strength': 0.12,
    'possession': 0.62, 
    'recent_form': 0.82
}
match_context = {
    'is_final': 1.0,        # 决赛
    'rivalry_level': 0.8,   # 老对手
    'tournament_level': 1.0, # 世界大赛
    'history_weight': 0.7,  # 历史渊源深
    'champion_prize': 0.9   # 冠军奖励重大
}
# 历史交锋记录 (最近5场)
match_history = [
    {'goal_diff': 1, 'cards': 5, 'total_goals': 3},
    {'goal_diff': 0, 'cards': 7, 'total_goals': 2},
    {'goal_diff': -1, 'cards': 4, 'total_goals': 4},
    {'goal_diff': 0, 'cards': 8, 'total_goals': 1},
    {'goal_diff': 2, 'cards': 6, 'total_goals': 5}
]
# 进行预测
prediction = predictor.predict_tension(team_a_stats, team_b_stats, match_context, match_history)
# 输出结果
print("=" * 60)
print("🏆 杯赛决赛紧张程度预测系统 🏆")
print("=" * 60)
print(f"📊 预测紧张程度评分: {prediction['tension_score']}/5")
print(f"😰 紧张等级: {prediction['tension_description']}")
print("-" * 60)
print("📈 详细分析:")
print(f"  • 主队综合实力: {prediction['details']['team_a_strength']}")
print(f"  • 客队综合实力: {prediction['details']['team_b_strength']}")
print(f"  • 实力差距: {prediction['details']['strength_gap']}")
print(f"  • 比赛重要程度: {prediction['details']['match_importance']}")
print(f"  • 历史交锋紧张度: {prediction['details']['historical_tension']}")
print(f"  • 结果利益重大程度: {prediction['details']['stake_level']}")
# 可视化
fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(12, 5))
# 左图：紧张程度指标
categories = ['实力差距', '比赛重要性', '历史交锋', '利益程度']
values = [
    prediction['details']['strength_gap'],
    prediction['details']['match_importance'],
    prediction['details']['historical_tension'],
    prediction['details']['stake_level']
]
bars = ax1.bar(categories, values, color=['#FF6B6B', '#4ECDC4', '#45B7D1', '#96CEB4'])
ax1.set_ylim(0, 1)
ax1.set_title('紧张程度指标分析')
ax1.set_ylabel('影响程度')
ax1.grid(axis='y', alpha=0.3)
# 在柱状图上添加数值
for bar, val in zip(bars, values):
    height = bar.get_height()
    ax1.text(bar.get_x() + bar.get_width()/2., height + 0.02,
             f'{val:.2f}', ha='center', va='bottom')
# 右图：最终紧张程度评估
tension_score = prediction['tension_score']
colors = ['#90EE90', '#87CEEB', '#FFD700', '#FFA500', '#FF6B6B', '#FF0000']
ax2.pie([tension_score, 5 - tension_score], 
        labels=[f'预测紧张度\n{tension_score}/5', f'剩余空间\n{5-tension_score:.2f}/5'],
        colors=[colors[prediction['tension_level']], '#DDDDDD'],
        autopct='%1.1f%%', startangle=90)
ax2.set_title('综合紧张程度评估')
plt.tight_layout()
plt.show()
# 添加模拟数据生成功能（用于批量预测）
def generate_simulation_data(n=100):
    """
    生成模拟数据用于测试
    """
    data = []
    for _ in range(n):
        team_a = {
            'fifa_rank': np.random.randint(1, 100),
            'goals_per_game': np.random.uniform(0.5, 1.5),
            'defense_strength': np.random.uniform(0.1, 0.5),
            'possession': np.random.uniform(0.4, 0.7),
            'recent_form': np.random.uniform(0.4, 0.95)
        }
        team_b = {
            'fifa_rank': np.random.randint(1, 100),
            'goals_per_game': np.random.uniform(0.5, 1.5),
            'defense_strength': np.random.uniform(0.1, 0.5),
            'possession': np.random.uniform(0.4, 0.7),
            'recent_form': np.random.uniform(0.4, 0.95)
        }
        context = {
            'is_final': np.random.choice([0.3, 0.6, 1.0]),
            'rivalry_level': np.random.uniform(0.3, 0.9),
            'tournament_level': np.random.choice([0.4, 0.7, 1.0]),
            'history_weight': np.random.uniform(0.3, 0.9),
            'champion_prize': np.random.uniform(0.5, 1.0)
        }
        data.append((team_a, team_b, context))
    return data
# 批量预测示例
sim_data = generate_simulation_data(5)
print("\n📊 批量预测示例：")
for i, (ta, tb, ctx) in enumerate(sim_data[:3], 1):
    result = predictor.predict_tension(ta, tb, ctx)
    print(f"  比赛{i}: 紧张评分 {result['tension_score']}/5, 等级: {result['tension_description']}")
print("\n✅ 预测完成！")

预测原理说明：

这个预测系统基于以下关键因素：