diff --git a/q01_load_data_and_add_column_names/__init__.pyc b/q01_load_data_and_add_column_names/__init__.pyc index ca8ec44..7d66947 100644 Binary files a/q01_load_data_and_add_column_names/__init__.pyc and b/q01_load_data_and_add_column_names/__init__.pyc differ diff --git a/q01_load_data_and_add_column_names/build.py b/q01_load_data_and_add_column_names/build.py index 95b910a..b0b8a6d 100644 --- a/q01_load_data_and_add_column_names/build.py +++ b/q01_load_data_and_add_column_names/build.py @@ -1,7 +1,18 @@ import pandas as pd +import numpy as np path = 'data/GermanData.csv' -def q01_load_data_and_add_column_names(): +def q01_load_data_and_add_column_names(path): + data= pd.read_csv(filepath_or_buffer=path, skip_blank_lines=True #, header=None # commented to pass the testcase + ) + data.columns =['account_status', 'month', 'credit_history', 'purpose', 'credit_amount', 'savings_account/bonds', +'employment', 'installment_rate', 'personal_status/sex', 'guarantors', 'residence_since', +'property', 'age', 'other_installment_plans', 'housing', 'number_of_existing_credits', 'job', 'liable', 'telephone', 'foreign_worker', 'good/bad' + ] + #map good/bad to 0, 1 + data["good/bad"] = np.where( data["good/bad"]==1 , 0 ,1) + return data +print (q01_load_data_and_add_column_names(path).head()) diff --git a/q01_load_data_and_add_column_names/build.pyc b/q01_load_data_and_add_column_names/build.pyc index 33a30a8..6c7cbc1 100644 Binary files a/q01_load_data_and_add_column_names/build.pyc and b/q01_load_data_and_add_column_names/build.pyc differ diff --git a/q01_load_data_and_add_column_names/tests/__init__.pyc b/q01_load_data_and_add_column_names/tests/__init__.pyc index 68ab5b6..1854b49 100644 Binary files a/q01_load_data_and_add_column_names/tests/__init__.pyc and b/q01_load_data_and_add_column_names/tests/__init__.pyc differ diff --git a/q01_load_data_and_add_column_names/tests/test.pyc b/q01_load_data_and_add_column_names/tests/test.pyc index 5a2c337..7560217 100644 Binary files a/q01_load_data_and_add_column_names/tests/test.pyc and b/q01_load_data_and_add_column_names/tests/test.pyc differ