import pandas as pd | |
data_set_options = [ | |
'Airline', | |
'Milk production' | |
] | |
def import_sample_data(sample_data_selected, data_set_options): | |
if sample_data_selected == data_set_options[0]: | |
data = pd.read_csv('data/processed/airline.csv', | |
parse_dates=['Month'], index_col='Month') | |
if sample_data_selected == data_set_options[1]: | |
data = pd.read_csv('data/processed/milk_production.csv', | |
parse_dates=['date'], index_col='date') | |
graph_data = data.reset_index() | |
graph_data.columns.values[0] = 'Date' | |
return data, graph_data | |