项目2：数据可视化（五）绘制世界人口地图

绘制世界人口地图

• 《Python编程：从入门到实践》这本书中使用pygal.i18n模块，该模块已被遗弃，改用pygal.maps.world模块，需要安装：
• pip install pygal_maps_world

import json
import pygal.maps.world
from pygal.maps.world import *
from pygal.style import RotateStyle as RS, LightColorizedStyle as LCS

# 将数据加载到一个列表中
filename = 'population_data.json'
with open(filename) as f:
    pop_data = json.load(f)

# 根据国家名称获得pygal使用的两位字母的国别码，若国家名未找到返回None
def get_country_code(country_name):
    for code, name in COUNTRIES.items():
        if name == country_name:
            return code
    return None

# 创建世界地图
"""修改基色，使三个国家分组使用同一种色调，并选择使用亮色调"""
"""'#336699'是十六进制的RGB色彩"""
wm_style = RS('#336699', base_style = LCS)
wm = pygal.maps.world.World(style = wm_style)
wm.title = "World population in 2016, by Country"

# 创建国别码与人口对应的字典
cc_populations = {}
for pop_dict in pop_data:
    if pop_dict['Year'] == 2016:
        country_name = pop_dict['Country Name']
        population = pop_dict['Value']
        code = get_country_code(country_name)
        if code:
            cc_populations[code] = population

# 将所有国家按人口多少分为3组
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

# 为绘制的世界地图增加项目
"""add接收两个实参，第一个是标签，第二个是{国别码：人口}的字典"""
wm.add('0-10m', cc_pops_1)
wm.add('10m-1bn', cc_pops_2)
wm.add('>1bn', cc_pops_3)
wm.render_to_file('WorldPopulationIn2016.svg')