diff --git a/q01_load_data_and_add_column_names/__init__.pyc b/q01_load_data_and_add_column_names/__init__.pyc index ca8ec44..60b0c5b 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..31ec9b1 100644 --- a/q01_load_data_and_add_column_names/build.py +++ b/q01_load_data_and_add_column_names/build.py @@ -3,5 +3,13 @@ path = 'data/GermanData.csv' -def q01_load_data_and_add_column_names(): - +def q01_load_data_and_add_column_names(path): + column_names = ['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'] + + df = pd.read_csv(path,header=None,names=column_names) + df['good/bad'] = df['good/bad'].map({1:0,2:1}) + df.drop(999,axis=0,inplace=True) + return df diff --git a/q01_load_data_and_add_column_names/build.pyc b/q01_load_data_and_add_column_names/build.pyc index 33a30a8..8d66da2 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..05d035b 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..9830bb3 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