-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathget_data.py
34 lines (25 loc) · 1.11 KB
/
get_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
34
import numpy as np
import pandas as pd
import os
data_path = os.path.abspath('.') + '/DataFiles/'
def main():
# Import datasets
teams = pd.read_csv('Teams.csv')
seeds = pd.read_csv('NCAATourneySeeds.csv')
reg_season = pd.read_csv('RegularSeasonCompactResults.csv')
cities = pd.read_csv('Cities.csv')
conferences = pd.read_csv('Conferences.csv')
conf_tourney = pd.read_csv('ConferenceTourneyGames.csv')
game_cities = pd.read_csv('GameCities.csv')
tourney_results = pd.read_csv('NCAATourneyCompactResults.csv')
detailed_tourney_results = pd.read_csv('NCAATourneyDetailedResults.csv')
tourney_all = pd.read_csv('NCAATourneySeedRoundSlots.csv')
tourney_seeds = pd.read_csv('NCAATourneySeeds.csv')
tourney_slots = pd.read_csv('NCAATourneySlots.csv')
detailed_reg_season = pd.read_csv('RegularSeasonDetailedResults.csv')
seasons = pd.read_csv('Seasons.csv')
second_tourney = pd.read_csv('SecondaryTourneyCompactResults.csv')
detailed_second_tourney = pd.read_csv('SecondaryTourneyTeams.csv')
team_coaches = pd.read_csv('TeamCoaches.csv')
team_conferences = pd.read_csv('TeamConferences.csv')
print(teams.head())