制作世界人口地图

world_population.py

import json
from country_code import get_country_code
import pygal
import pygal_maps_world.maps
from pygal.style import RotateStyle
from pygal.style import LightColorizedStyle
#from pygal.style import LightColorizedStyle as LCS, RotateStyle as RS
#将数据加载到一个列表中
filename = 'population_data.json'
with open(filename) as f:
    pop_data = json.load(f)
cc_populations = {}
#打印每个国家2010年的人口数量
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)
        #print(code)
        if code:
            #print(code + " " + str(population))
            cc_populations[code] = population
        '''else:
            print('Error - ' + country_name)'''

cc_pops_1 ={}
cc_pops_2 ={}
cc_pops_3 ={}
for cc,pop in cc_populations.items():
    if pop < 10000000:
        cc_pops_1[cc] = pop
    elif pop < 1000000000:
        cc_pops_2[cc] = pop
    else:
        cc_pops_3[cc] = pop
# 看看每组分别包含多少个国家
print(len(cc_pops_1), len(cc_pops_2), len(cc_pops_3))

wm_style = RotateStyle('#336699',base_style = LightColorizedStyle)
wm = pygal_maps_world.maps.World(style = wm_style)
wm.title = 'World Population in 2010, by Country'
wm.add('0-10m',cc_pops_1)
wm.add('10m-1bn',cc_pops_2)
wm.add('>1bn',cc_pops_3)
wm.render_to_file('world_population.svg')
       

country_code.py

from pygal.maps.world import COUNTRIES
#from pygal.i18n import COUNTRIES
"""pygal.i18n 已经不存在了，现在已经更改成了 pygal_maps_world ，需要单独通过pip下载,将#from pygal.i18n import COUNTRIES改为from pygal.maps.world import COUNTRIES"""

def get_country_code(country_name):
    """根据指定的国家，返回Pygal使用的两个字母的国别码"""
    for code,name in COUNTRIES.items():
        if name == country_name:
            return code
    #如果没有找到指定的国家，就返回None
    return None

print(get_country_code('Andorra'))
print(get_country_code('United Arab Emirates'))
print(get_country_code('Afghanistan'))