Я работаю с набором данных df_cities, содержащим следующие столбцы:
city_name, country_name, population, language
У меня есть следующая функция для сортировки таблицы по тому, находится ли страна в каком-либо списке:
special_countries = ['Morocco', 'Turkey', 'Japan' ]
def sort_function(df, country_name):
if country_name in special_countries:
return df.sort_values(by='population')
else:
return df
Далее я хочу использовать groupby для применения этой сортировки:
df_cities.groupby('country_name').apply(sort_function, kwargs={'country_name': x})
Это конечно не работает, так как я нигде не объявил x. У меня вопрос: как я могу предоставить ключ группы (в данном случае название страны) в качестве аргумента функции, которую я передаю в .apply ()?