NBA/src/python/improtData.py

# coding=utf-8
'''
导入数据，13个csv文件，并查看相关数据格式。
Created on 2017年9月12日
@vsersion:python 3.6
@author: liuyuqi
'''

# 导入需要的第三方库

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
from nt import chdir

projectPath = "D:/app/NBA"


def preImport():
    print("start!")
    chdir(projectPath)
    print("finish!")

# 导入数据，预览数据
def importData():
    team_season = pd.read_csv('../input/NBAdata/team_season.csv')
    advanced_basic = pd.read_csv('../input/NBAdata/advanced_basic.csv')
    advanced_shooting = pd.read_csv('../input/NBAdata/advanced_shooting.csv')
    avg = pd.read_csv('../input/NBAdata/avg.csv')
    coach_playoff = pd.read_csv('../input/NBAdata/coach_playoff.csv')
    coach_season = pd.read_csv('../input/NBAdata/coach_season.csv')
    player_playoff = pd.read_csv('../input/NBAdata/player_playoff.csv')
    player_salary = pd.read_csv('../input/NBAdata/player_salary.csv')
    player_season = pd.read_csv('../input/NBAdata/player_season.csv')
    single = pd.read_csv('../input/NBAdata/single.csv')
    team_playoff = pd.read_csv('../input/NBAdata/team_playoff.csv')
    team_season = pd.read_csv('../input/NBAdata/team_season.csv')
    tot = pd.read_csv('../input/NBAdata/tot.csv')
    
    team_season.head()
    team_playoff.columns