text_plain_output_1.png
import pandas as pd

DATA_PATH = '../input'
def csv_path(dataset='train', data_path=DATA_PATH):
    """
    """
    return '{}/{}.csv'.format(data_path, dataset)
def read_data(dataset='train', data_path=DATA_PATH):
    """
    """
    index_col = None
    index_type = ['train', 'test']
    if dataset in index_type:
        index_col = 'id'
    data_path = csv_path(dataset, data_path=data_path)
    return pd.read_csv(data_path, index_col=index_col)

train = read_data('train')
test = read_data('test')