commit-live-students · shawna2z · May 5, 2018
diff --git a/__pycache__/__init__.cpython-36.pyc b/__pycache__/__init__.cpython-36.pyc
diff --git a/q02_rename_columns/__pycache__/__init__.cpython-36.pyc b/q02_rename_columns/__pycache__/__init__.cpython-36.pyc
diff --git a/q02_rename_columns/__pycache__/build.cpython-36.pyc b/q02_rename_columns/__pycache__/build.cpython-36.pyc
diff --git a/q02_rename_columns/build.py b/q02_rename_columns/build.py
@@ -1,9 +1,13 @@
+# %load q02_rename_columns/build.py
 import pandas as pd
 import numpy as np
 from sklearn.model_selection import train_test_split
 from greyatomlib.olympics_project.q01_load_data.build import q01_load_data
 
-def q02_rename_columns(path):
-    "write your solution here"
+def q02_rename_columns(path='./data/olympics.csv'):
+    'write your solution here'
     df = q01_load_data(path)
-
+    df.rename(columns={'01 !':'Gold','02 !':'Silver','03 !':'Bronze'},inplace=True)
+    return df
+
+
diff --git a/q02_rename_columns/tests/__pycache__/__init__.cpython-36.pyc b/q02_rename_columns/tests/__pycache__/__init__.cpython-36.pyc
diff --git a/q02_rename_columns/tests/__pycache__/test.cpython-36.pyc b/q02_rename_columns/tests/__pycache__/test.cpython-36.pyc