You are on page 1of 1

#importar bibliotecas

import inline
import matplotlib
import pandas as pd
import numpy as np
import csv
from sklearn.linear_model import LogisticRegression
from sklearn.tree import DecisionTreeClassifier
from sklearn.naive_bayes import GaussianNB
from sklearn.svm import SVC
from sklearn.metrics import accuracy_score , f1_score, precision_score,
recall_score
from sklearn.feature_selection import SelectKBest
from sklearn.model_selection import GridSearchCV
from sklearn.preprocessing import StandardScaler, MinMaxScaler
from IPython.display import display
import matplotlib.pyplot as plt
import seaborn as sns
from sklearn.preprocessing import scale

#lendo o arquivo BRAS.csv


#O arquivo utilizado foi baixado do site www.football-data.co.uk, que agrega
informações de diversos campeonatos
#ao redor do mundo.

data=pd.read_csv('BRA.csv',delimiter=',')
data['IDX'] = 'NAN'
i = 0
print(len(data))
list =[]
jogo = []
while i < len(data):
jogo = data.loc[i]
IDX = round(float(jogo['MaxA']) - float(jogo['MaxH']),2)
jogo['IDX'] = IDX
list.append(jogo)
i = i +1
print(list)
f = open('BRAIDX.csv', 'w', newline='', encoding='utf-8')
w = csv.writer(f)
w.writerow(["Country", "League", "Season", "Date", "Time", "Home", "Away", "HG",
"AG", "Res", "PH", "PD", "PA", "MaxH", "MaxD", "MaxA", "AvgH", "AvgD", "AvgA",
"IDX"])
b = 0
while b < len(list):
w.writerow(list[b])
b = b+1

You might also like