diff --git a/__pycache__/__init__.cpython-36.pyc b/__pycache__/__init__.cpython-36.pyc new file mode 100644 index 0000000..09d735f Binary files /dev/null and b/__pycache__/__init__.cpython-36.pyc differ diff --git a/q01_rename_columns/__pycache__/__init__.cpython-36.pyc b/q01_rename_columns/__pycache__/__init__.cpython-36.pyc index 86045c5..03aaa6a 100644 Binary files a/q01_rename_columns/__pycache__/__init__.cpython-36.pyc and b/q01_rename_columns/__pycache__/__init__.cpython-36.pyc differ diff --git a/q01_rename_columns/__pycache__/build.cpython-36.pyc b/q01_rename_columns/__pycache__/build.cpython-36.pyc index 56daa0d..956bba7 100644 Binary files a/q01_rename_columns/__pycache__/build.cpython-36.pyc and b/q01_rename_columns/__pycache__/build.cpython-36.pyc differ diff --git a/q01_rename_columns/build.py b/q01_rename_columns/build.py index 5d2373e..9577fd0 100644 --- a/q01_rename_columns/build.py +++ b/q01_rename_columns/build.py @@ -1,5 +1,28 @@ -# default imports +import numpy as np import pandas as pd +import matplotlib.pyplot as plt +def q01_rename_columns(path): + df = pd.read_csv(path, skiprows=1) + df.head() + df.rename(columns={'Unnamed: 0': 'Country', + '01 !':'Gold_Summer' , + '02 !':'Silver_Summer', + '03 !':'Bronze_Summer', + 'Total':'Total_Summer', + '01 !.1': 'Gold_Winter', + '02 !.1': 'Silver_Winter' , + '03 !.1' : 'Bronze_Winter', + 'Total.1':'Total_Winter', + '01 !.2' :'Gold_Total', + '02 !.2': 'Silver_Total', + '03 !.2':'Bronze_Total', + 'Combined total' :'Total' + },inplace=True) + return df; + +print (q01_rename_columns('./data/olympics.csv')) + + diff --git a/q01_rename_columns/tests/__pycache__/__init__.cpython-36.pyc b/q01_rename_columns/tests/__pycache__/__init__.cpython-36.pyc index ee14684..444b9ce 100644 Binary files a/q01_rename_columns/tests/__pycache__/__init__.cpython-36.pyc and b/q01_rename_columns/tests/__pycache__/__init__.cpython-36.pyc differ diff --git a/q01_rename_columns/tests/__pycache__/test.cpython-36.pyc b/q01_rename_columns/tests/__pycache__/test.cpython-36.pyc index bd0596e..75747d0 100644 Binary files a/q01_rename_columns/tests/__pycache__/test.cpython-36.pyc and b/q01_rename_columns/tests/__pycache__/test.cpython-36.pyc differ