我正在研究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')
答案 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