python - 使用 Continent : Country to a DataFrame with a Country Column 映射字典

Question

我有一个列 df['Country'] 的数据框。我创建了一个字典，其中所有大陆作为键，值作为相应国家/地区的列表。continents = { "North America" : ["Canada","Mexico","United States"]}我想将它映射到我的数据框，它在 df['Country'] 对应的大陆中创建一个包含每个值的列。
我已经尝试了以下但它返回一列 NA 值df['Continents'] = df['Country'].map(continents)我需要创建一个函数来映射大陆还是可以使用 pandas .map() 方法？

Answer 1

您的字典格式错误，您需要相反的格式(并且没有列表)。
您可以使用它来获得正确的格式:

countries_map = {country: continent
                 for continent,countries in continents.items()
                 for country in countries}

输出:

{'Canada': 'North America',
 'Mexico': 'North America',
 'United States': 'North America'}

然后继续如下:

df['Continents'] = df['Country'].map(countries_map)