-
Notifications
You must be signed in to change notification settings - Fork 0
/
process_data.py
33 lines (26 loc) · 1.09 KB
/
process_data.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
import pandas as pd
# data from https://www.pro-football-reference.com/years/2022/games.htm
df17 = pd.read_csv('2017.csv')
df17['isTie'] = df17['PtsW']==df17['PtsL']
df17['Week'] = '2017-'+df17['Week']
df18 = pd.read_csv('2018.csv')
df18['isTie'] = df18['PtsW']==df18['PtsL']
df18['Week'] = '2018-'+df18['Week']
df19 = pd.read_csv('2019.csv')
df19['isTie'] = df19['PtsW']==df19['PtsL']
df19['Week'] = '2019-'+df19['Week']
df20 = pd.read_csv('2020.csv')
df20['isTie'] = df20['PtsW']==df20['PtsL']
df20['Week'] = '2020-'+df20['Week']
df21 = pd.read_csv('2021.csv')
df21['isTie'] = df21['PtsW']==df21['PtsL']
df21['Week'] = '2021-'+df21['Week']
df22 = pd.read_csv('2022.csv')
df22['isTie'] = df22['PtsW']==df22['PtsL']
df22['Week'] = '2022-'+df22['Week'].astype(str)
df = pd.concat([df17, df18, df19, df20, df21, df22]).reset_index()
df = df[['Week', 'Date', 'Winner/tie', 'Loser/tie', 'isTie', 'PtsW', 'PtsL', 'type']]
df[df['type']=='boxscore'].to_csv('results.csv')
#get next weeks games
next_week=df[df['type']=='preview'].iloc[0]['Week']
df[df['Week'] == next_week].reset_index().to_csv('next_games.csv')