diff --git a/q01_load_data/build.py b/q01_load_data/build.py index a29c139..08e2c4e 100644 --- a/q01_load_data/build.py +++ b/q01_load_data/build.py @@ -1,7 +1,19 @@ +# %load q01_load_data/build.py import pandas as pd import numpy as np from sklearn.model_selection import train_test_split +path = 'data/elecdemand.csv' - +def q01_load_data(data): + + df = pd.read_csv(data) # Load data and converting to DataFrame + + # Make the Datetime column as object and Drop the index. + df['Datetime'] = pd.to_datetime(df['Datetime'], format = '%Y-%m-%d') + #df.index = df['Datetime'] + return (df.shape, df) +q01_load_data(path) + + diff --git a/q01_load_data/tests/test_sol.pkl b/q01_load_data/tests/test_sol.pkl new file mode 100644 index 0000000..7912fb6 Binary files /dev/null and b/q01_load_data/tests/test_sol.pkl differ diff --git a/q01_load_data/tests/user_sol.pkl b/q01_load_data/tests/user_sol.pkl new file mode 100644 index 0000000..2ad49fc Binary files /dev/null and b/q01_load_data/tests/user_sol.pkl differ