无法让最后几个国家在pygal世界地图上显示其人口

时间:2019-04-01 14:08:45

标签: python pygal

我正在研究python崩溃过程,并在pygal世界地图上绘制种群。由于国家/地区名称不是标准的,因此必须专门检索某些国家/地区代码。我开始尝试通过玻利维亚和刚果获得这种非标准的国家/地区代码,但是在pygal地图上,两者仍然空白。附件是两个相关的模块,将不胜感激。

获取国家/地区代码的代码:

from pygal.maps.world import COUNTRIES

def get_country_code(country_name):
    """return the pygal 2-digit country code for
    given country"""
    for code, name in COUNTRIES.items():
        if name == country_name:
            return code
    if country_name == 'Bolivia, Plurinational State of':
        return 'bo'
    elif country_name == 'Congo, the Democratic Republic of the':
        return 'cd'

        #if the country wasnt found, return none
    return None

,然后将其导出到pygal地图的程序

import json

from pygal.maps.world import World

from pygal.style import RotateStyle

from country_codes import get_country_code

#load the data into a list
filename = 'population_data.json'
with open(filename) as f:
    pop_data = json.load(f)

#build a dictionary of population data
cc_population = {}


#print the 2010 population for each country
for pop_dict in pop_data:
    if pop_dict['Year'] == '2010':
        country_name = pop_dict['Country Name']
        population = int(float(pop_dict['Value']))
        code = get_country_code(country_name)
        if code:
            cc_population[code] = population

#Group the countries into 3 population levels
cc_pops_1, cc_pops_2, cc_pops_3 = {}, {}, {}
for cc, pop in cc_population.items():
    if pop < 10000000:
        cc_pops_1[cc] = pop
    elif pop < 1000000000:
        cc_pops_2[cc] = pop
    else:
        cc_pops_3[cc] = pop

wm_style = RotateStyle('#994033')
wm = World(style=wm_style)
wm.title = 'World population in 2010, by country'
wm.add('0-10 mil', cc_pops_1)
wm.add('10m-1bn', cc_pops_2)
wm.add('>1bn', cc_pops_3)

wm.render_to_file('world_population.svg')

1 个答案:

答案 0 :(得分:1)

您似乎正在检查Pygal世界地图模块中定义的国家/地区的名称,但是应该检查json数据文件中使用的名称。

例如,假设json文件使用名称“玻利维亚”,则需要将该特定比较更改为

if country_name == 'Bolivia':
    return 'bo'

您可以通过在函数的最后一个print之前添加return语句来标识需要以这种方式处理的其他任何国家。当您运行该程序时,控制台上会列出所有缺少的国家,以及您需要检查的特定文本。

#if the country wasnt found, return none
print(country_name)
return None