commit-live-students · NeehaK · May 5, 2018 · May 5, 2018 · May 9, 2018 · May 9, 2018
diff --git a/__pycache__/__init__.cpython-36.pyc b/__pycache__/__init__.cpython-36.pyc
diff --git a/q01_load_data/__pycache__/__init__.cpython-36.pyc b/q01_load_data/__pycache__/__init__.cpython-36.pyc
diff --git a/q01_load_data/__pycache__/build.cpython-36.pyc b/q01_load_data/__pycache__/build.cpython-36.pyc
diff --git a/q01_load_data/build.py b/q01_load_data/build.py
@@ -1,10 +1,14 @@
+#%load q01_load_data/build.py
 import pandas as pd
 import numpy as np
-from sklearn.model_selection import train_test_split
-
 
+path ='./data/olympics.csv'
 def q01_load_data(path):
-    "write your solution here"
-    # use .read_csv function to read the
-    # data and header=0 to skip the first row
+    data = pd.read_csv(path)
+    data_header = data.iloc[0]
+    data=data[1:]
+    data_header[0]='country name'
+    data.columns=data_header
+    return data
+
 
diff --git a/q01_load_data/tests/__pycache__/__init__.cpython-36.pyc b/q01_load_data/tests/__pycache__/__init__.cpython-36.pyc
diff --git a/q01_load_data/tests/__pycache__/test.cpython-36.pyc b/q01_load_data/tests/__pycache__/test.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,22 @@
+# %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
 
+path= './data/olympics.csv'
 def q02_rename_columns(path):
-    "write your solution here"
+    'write your solution here'
     df = q01_load_data(path)
-
+    data_header = list(df.columns.values)
+    for i in range(0,len(data_header),1):
+        if(str(data_header[i])=='01 !'):
+            data_header[i]='Gold'
+        if(str(data_header[i])=='02 !'):
+            data_header[i]='Silver'
+        if(str(data_header[i])=='03 !'):
+            data_header[i]='Bronze'
+    df=df[1:]
+    df.columns=data_header
+    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
diff --git a/q03_split_country/__pycache__/__init__.cpython-36.pyc b/q03_split_country/__pycache__/__init__.cpython-36.pyc
diff --git a/q03_split_country/__pycache__/build.cpython-36.pyc b/q03_split_country/__pycache__/build.cpython-36.pyc
diff --git a/q03_split_country/build.py b/q03_split_country/build.py
@@ -1,10 +1,19 @@
+# %load q03_split_country/build.py
 import pandas as pd
 import numpy as np
 from sklearn.model_selection import train_test_split
 from greyatomlib.olympics_project.q02_rename_columns.build import q02_rename_columns
 
-
+path='./data/olympics.csv'
 def q03_summer_gold_medals(path):
-    "write your solution here"
-    df = q02_rename_columns(path)
-
+    'write your solution here'
+    df2 = q02_rename_columns(path)
+    df2. rename(columns={'country name': 'country_name'}, inplace=True)
+    df2['country name'] = df2['country_name'].str.split('(', 1).str.get(0)
+    df2.set_index('country name', inplace=True)
+    df2 = df2.drop('country_name', 1)
+    df2 = df2.drop('Totals', 0)
+    return df2
+
+
+
diff --git a/q03_split_country/tests/__pycache__/__init__.cpython-36.pyc b/q03_split_country/tests/__pycache__/__init__.cpython-36.pyc
diff --git a/q03_split_country/tests/__pycache__/test.cpython-36.pyc b/q03_split_country/tests/__pycache__/test.cpython-36.pyc
diff --git a/q04_country_with_most_gold_medals/__pycache__/__init__.cpython-36.pyc b/q04_country_with_most_gold_medals/__pycache__/__init__.cpython-36.pyc
diff --git a/q04_country_with_most_gold_medals/__pycache__/build.cpython-36.pyc b/q04_country_with_most_gold_medals/__pycache__/build.cpython-36.pyc
diff --git a/q04_country_with_most_gold_medals/build.py b/q04_country_with_most_gold_medals/build.py
@@ -1,11 +1,32 @@
+# %load q04_country_with_most_gold_medals/build.py
 import pandas as pd
 import numpy as np
 from sklearn.model_selection import train_test_split
 from greyatomlib.olympics_project.q03_split_country.build import q03_summer_gold_medals
-
+path='./data/olympics.csv'
 
 def q04_country_with_most_gold_medals(path):
-    "write your solution here"
-    df = q03_summer_gold_medals(path)
+    'write your solution here'
+    data = q03_summer_gold_medals(path)
+    x,y=data.shape
+    list1=[]
+    for j in range(0,x,1):
+        sum1=0
+        for i in range(0,len(data.columns),1):
+            if str(data.columns[i])=='Gold':
+                sum1+=int(data.iloc[j,i])
+
+        list1.append(sum1)
+
+    max_val=max(list1)
+    max_index=list1.index(max_val)
+    data= data.reset_index()
+
+    max_country=data.iloc[max_index,0]
+
 
+
+    return max_country
+
+
 
diff --git a/q04_country_with_most_gold_medals/tests/__pycache__/__init__.cpython-36.pyc b/q04_country_with_most_gold_medals/tests/__pycache__/__init__.cpython-36.pyc
diff --git a/q04_country_with_most_gold_medals/tests/__pycache__/test.cpython-36.pyc b/q04_country_with_most_gold_medals/tests/__pycache__/test.cpython-36.pyc