python练习-NBA比赛结果预测

python预测nba比赛结果

# -*- coding: utf-8 -*-
“““
Created on Sun Jun  4 16:19:16 2017

@author: dc
“““

import pandas as pd 
import math
import numpy as np
from sklearn import linear_model
import csv

init_elo = 1600
team_elos = {}
#trainteamstat = {}

folder = ‘data‘

def PruneData（M_stat O_stat T_stat）:
    #这个函数要完成的任务在于将原始读入的诸多队伍的数据经过修剪，使其变为一个以team为索引
    #排列的特征数据
    
    #丢弃与球队实力无关的统计量
    pruneM = M_stat.drop（[‘Rk‘ ‘Arena‘]axis = 1）
    pruneO = O_stat.drop（[‘Rk‘‘G‘‘MP‘]axis = 1）
    pruneT = T_stat.drop（[‘Rk‘‘G‘‘MP‘]axis = 1）
    
    #将多个数据通过相同的index：team合并为一个数据
    mergeMO = pd.merge（pruneM pruneO how = ‘left‘ on = ‘Team‘）
    newstat = pd.merge（mergeMO pruneT  how = ‘left‘ on = ‘Team‘）
    
    #将team作为index的数据返回
    return newstat.set_index（‘Team‘ drop = True append = False）

def GetElo（team）:
    #
    try:
        return team_elos[team]
    except:
        team_elos[team] = init_elo
    return team_elos[team]


def CalcElo（winteam l

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2017-06-10 12:20  python-项目一NBA\
     目录           0  2017-06-10 12:27  python-项目一NBA\data\
     文件        4795  2017-03-09 17:50  python-项目一NBA\data\15-16Miscellaneous_Stat.csv
     文件        4078  2017-03-09 17:49  python-项目一NBA\data\15-16Opponent_Per_Game_Stat.csv
     文件        4057  2017-03-09 17:48  python-项目一NBA\data\15-16Team_Per_Game_Stat.csv
     文件       43180  2017-03-10 01:04  python-项目一NBA\data\16-17Schedule.csv
     文件       49089  2017-03-09 00:41  python-项目一NBA\data\2015-2016_result.csv
     文件       12348  2017-06-05 09:23  python-项目一NBA\data\Train_data.csv
     文件        5581  2017-06-10 12:26  python-项目一NBA\p1_NBA_prediction.py