如何使用pycountry_convert在列出国家所在大陆的DataFrame中创建一个新列？

Question

我有一个DataFrame，如下所示

df = pd.DataFrame({'Country':['Italy', 'Spain', 'China']})

我想创建一个名为“欧洲大陆”的新专栏。

我还知道pycountry_convert，它输出一个大陆代码。例如：

country_code = pc.country_name_to_country_alpha2("Germany", cn_name_format="default")
print(country_code)
continent_name = pc.country_alpha2_to_continent_code(country_code)
print(continent_name)

如何使用此代码更新“欧洲大陆”的新列？

Answer 1

使用lambda函数：

f = lambda x: pc.country_name_to_country_alpha2(x, cn_name_format="default")
df['new'] = df['Country'].apply(f)

或清单理解：

df['new'] = [pc.country_name_to_country_alpha2(x, cn_name_format="default") 
                                                                    for x in df['Country']]