基于国家频率计数的颜色图

时间:2019-12-12 03:18:04

标签: python plotly choropleth

我在“国家/地区”列中设置了用户数据,并希望绘制用户在各个国家/地区的分布图。我将数据集转换为字典,其中的键是国家/地区名称和值-国家/地区的频率计数。字典看起来像这样:

'usa': 139421,
'canada': 21601, 
'united kingdom': 18314,
'germany': 17024,
'spain': 13096,
 [...]

要在世界地图上绘制分布图,我使用了以下代码:

#Convert to dictionary
counts = users['Country'].value_counts().to_dict()

#Country names
def getList(dict): 
    return [*dict]

countrs = getList(counts)

#Frequency counts
freqs = list(counts.values())

#Plotting
data = dict(
        type = 'choropleth',
        colorscale = 'Viridis',
        reversescale = True,
        locations = countrs,
        locationmode = "country names",
        z = freqs,
        text = users['Country'],
        colorbar = {'title' : 'Number of Users'},
      ) 

layout = dict(title = 'Number of Users per Country',
                geo = dict(showframe = False)
             )

choromap = go.Figure(data = [data],layout = layout)
iplot(choromap,validate=False)

这是我得到的结果: enter image description here

着色错误;它表明所有国家/地区都在0-20K的范围内,这是错误的。有没有办法来解决这个问题?谢谢

1 个答案:

答案 0 :(得分:1)

如果无法访问完整的数据集,这确实很难回答。我建议从以下示例开始:

情节1:

enter image description here

在这里,您可以简单地将lifeExp替换为您的数据,并且只要数据格式正确,一切就可以了。在以下代码段中,我为每个国家/地区创建了随机整数,以表示您的counts变量。

代码:

import plotly.express as px
import numpy as np

np.random.seed(12)
gapminder = px.data.gapminder().query("year==2007")
gapminder['counts'] = np.random.uniform(low=100000, high=200000, size=len(gapminder)).tolist()

fig = px.choropleth(gapminder, locations="iso_alpha",
                    color="counts", 
                    hover_name="country", # column to add to hover information
                    color_continuous_scale=px.colors.sequential.Plasma)

fig.show()

图2:

enter image description here

让我知道这对您有何帮助。

编辑:关于您数据的建议:

如果您有一本包含国家/地区名称和国家/地区的字典,则可以轻松地为其构建一个数据框并执行左连接以获取此信息:

图2:

enter image description here

只需确保您的 词典值是列表 ,并且确保国家名称的拼写和格式正确。

代码2:

import plotly.express as px
import numpy as np
import pandas as pd

np.random.seed(12)
gapminder = px.data.gapminder().query("year==2007")
#gapminder['counts'] = np.nan

d = {'United States': [139421],
    'Canada': [21601], 
    'United Kingdom': [18314],
    'Germany': [17024],
    'Spain': [13096]}

yourdata = pd.DataFrame(d).T.reset_index()
yourdata.columns=['country', 'count']

df=pd.merge(gapminder, yourdata, how='left', on='country')

fig = px.choropleth(df, locations="iso_alpha",
                    color="count", 
                    hover_name="country", # column to add to hover information
                    color_continuous_scale=px.colors.sequential.Plasma)

fig.show()