我正在使用包含以下列的数据集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。我的问题是：如何提供组的键（在本例中为country_name）作为我在.apply（）中传递的函数的参数？