| 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 | |