# %load q04_find_top_10/build.py
# default imports
from greyatomlib.olympics_project_new.q03_better_event.build import q03_better_event,q02_country_operations, q01_rename_columns
path = './data/olympics.csv'
OlympicsDF=q01_rename_columns(path)
OlympicsDF=q02_country_operations(OlympicsDF)
OlympicsDF=q03_better_event(OlympicsDF)
def q04_find_top_10(df=OlympicsDF,v1='Total_Summer',v2='Total_Winter',v3='Total'):
top_10_summer=list(df.sort_values(v1,ascending=False)['Country_Name'][1:11])
top_10_winter=list(df.sort_values(v2,ascending=False)['Country_Name'][1:11])
top_10_overall=list(df.sort_values(v3,ascending=False)['Country_Name'][1:11])
common_overall=list(set(top_10_summer).intersection(set(top_10_winter).intersection(set(top_10_overall))))
return top_10_summer,top_10_winter,top_10_overall,common_overall
path = './data/olympics.csv'
import pandas as pd
def q04_find_top_10(DF1, Total_Summer, Total_Winter, Total):
OlympicsDF = DF1[0:-1]
SummerList = OlympicsDF.sort_values(['Total_Summer'],
ascending=False).head(10)
WinterList = OlympicsDF.sort_values(['Total_Winter'],
ascending=False).head(10)
Top = OlympicsDF.sort_values(['Total'], ascending=False).head(10)
CommonElements = pd.merge(pd.merge(SummerList,
WinterList,
on='Country_Name'),
Top,
on='Country_Name')
return list(SummerList['Country_Name']), list(
WinterList['Country_Name']), list(Top['Country_Name']), list(
CommonElements['Country_Name'])
DF = q01_rename_columns(path)
DF = q02_country_operations(DF)
DF = q03_better_event(DF)
q04_find_top_10(DF, DF['Total_Summer'], DF['Total_Winter'], DF['Total'])
[
'United States', 'Soviet Union', 'Great Britain', 'Germany', 'France',
'Italy', 'Sweden', 'China', 'East Germany', 'Russia'
]
