当前位置:首页 > 篮球快讯篮球快讯
Python 进行 NBA 比赛数据分析
发布时间:2024-06-02 01:07:15【篮球快讯】人次阅读
摘要 import pandas as pdimport mathimport csvimport randomimport numpy as npfrom sklearn import linear_modelfrom sklearn.model_selection import cross_val_scoreba
import pandas as pd
import math
import csv
import random
import numpy as np
from sklearn import linear_model
from sklearn.model_selection import cross_val_score
base_elo = 1600
team_elos = {}
team_stats = {}
X = []
y = []
#初始化数据,从T,O,M表格中读取数据,取出一些无关数据并将这三个表格通过team树形列进行连接:
#根据每个队伍的Miscellaneous Opponent,Team统计数据csv文件进行初始化
def initialize_data(Mstat,Ostat,Tstat):
new_Mstat = Mstat.drop(['Rk','Arena'],axis=1)
new_Ostat = Ostat.drop(['Rk',"G",'MP'],axis=1)
new_Tstat = Tstat.drop(['Rk',"G",'MP'],axis=1)
team_stats1 = pd.merge(new_Mstat,new_Ostat,how='left',on='Team')
team_stats1 = pd.merge(team_stats1,new_Tstat,how='left',on='Team')
return team_stats1.set_index('Team',inplace=False,drop=True)
def get_elo(team):
try:
return team_elos[team]
except:
team_elos[team] = base_elo
return team_elos[team]
def calc_elo(win_team,lose_team):
winner_rank = get_elo(win_team)
loser_rank = get_elo(lose_team)
#根据Logistic Distribution计算 PK 双方(A和B)对各自的胜率期望值计算公式
rank_diff = winner_rank - loser_rank
exp = (rank_diff *-1)/400
odds = 1/(1+math.pow(10,exp))
#根据rank界别修改k值
if winner_rank < 2100:
k = 32
elif winner_rank >=2100 and winner_rank <2400:
k = 24
else:
k=16
#更新rank数值
new_winner_rank = round(winner_rank+(k*(1-odds)))
new_loser_rank = round(loser_rank+(k*(0-odds)))
return new_winner_rank,new_loser_rank
#基于统计好的数据,给每只队伍的eloscore计算结果,建立对应15-16年数据集,我们认为主场作战的队伍更有优势,因此会给主场队伍加上100分
def build_dataSet(all_data):
print("Building data set..")
X = []
skip = 0
for index,row in all_data.iterrows():
Wteam = row['WTeam']
Lteam = row['LTeam']
#获取最初的elo或者每个队伍最初的elo值
team1_elo = get_elo(Wteam)
team2_elo = get_elo(Lteam)
#给主场比赛队伍加上100的elo值
if row['WLoc'] == 'H':
team1_elo += 100
else:
team2_elo += 100
#把elo当成评价每个队伍的第一个特征值
team1_features = [team1_elo]
team2_features = [team2_elo]
# 添加我们从basketball reference.com获得的每个队伍的统计信息
for key,value in team_stats.loc[Wteam].iteritems():
team1_features.append(value)
for key,value in team_stats.loc[Lteam].iteritems():
team2_features.append(value)
# 将两支队伍的特征值随机的分配在每场比赛数据的左右两侧
# 并将对应的0/1赋给y值
if random.random() > 0.5:
X.append(team1_features+team2_features)
y.append(0)
else:
X.append(team2_features+team1_features)
y.append(1)
if skip ==0:
print('X',X)
skip = 1
new_winner_rank,new_loser_rank = calc_elo(Wteam,Lteam)
team_elos[Wteam] = new_winner_rank
team_elos[Lteam] = new_loser_rank
return np.nan_to_num(X),y
#最终利用训练好的模型在 16~17 年的常规赛数据中进行预测
def predict_winner(team_1, team_2, model):
features = []
# team 1,客场队伍
features.append(get_elo(team_1))
for key, value in team_stats.loc[team_1].iteritems():
features.append(value)
标签:
很赞哦! ()
相关内容
2023第25届CUBAL大学生篮球联赛即将迎来总决赛,和九号一起见证冠军诞生
2024-09-21 在这个充满热血与梦想的夏日,我们共同期待一场年度重磅级篮球盛宴华大男篮再夺CUBAL福建基层赛冠军 周鲁男:以赛练兵 备战东南赛区
2024-09-21 晋江新闻网12月23日讯 20日晚,以91∶55击败福建师范大学男2022-2028年全球与中国篮球鞋行业发展研究分析与发展趋势预测报告
2024-09-21 篮球鞋是一种专门用于篮球运动的鞋类,因其能够提供良好的支撑【祝贺-校友组织荣誉】第二届北京MBA篮球超级联赛顺利闭幕,国发女篮荣获季军!
2024-09-20 2024年3月30日,第二届北京MBA篮球超级联赛在北京拉开帷幕,男篮共22