Python Pygal绘制世界人口地图

数据集可在 https://datahub.io/JohnSnowLabs/population-figures-by-country 下载

1#coding=utf-8
2import json
3from country_codes import get_country_code
4import pygal.maps.world
5from pygal.style import RotateStyle as RS, LightColorizedStyle as LCS
6
7# 将数据加载到列表中
8filename = 'population_data.json'
9with open(filename) as f:
10    pop_data = json.load(f)
11
12# 创建一个包含人口数量的字典
13cc_populations = {}
14for pop_dict in pop_data:
15    country_name = pop_dict['Country_Name']
16    population = pop_dict['Population_in_2016']
17    code = get_country_code(country_name)
18    if code:
19        cc_populations[code] = population
20
21# 根据人口数量将国家分成4组
22cc_pop_1, cc_pop_2, cc_pop_3, cc_pop_4 = {}, {}, {}, {}
23for cc, pop in cc_populations.items():
24    if pop < 10000000:
25        cc_pop_1[cc] = pop
26    elif pop < 100000000:
27        cc_pop_2[cc] = pop
28    elif pop < 1000000000:
29        cc_pop_3[cc] = pop
30    else:
31        cc_pop_4[cc] = pop
32
33# wm_style = RS('#336699', base_style=LCS)
34wm = pygal.maps.world.World()
35wm.title = 'World Population in 2016, by Country'
36wm.add(u'0-1千万', cc_pop_1)
37wm.add(u'1千万-1亿', cc_pop_2)
38wm.add(u'1亿-10亿', cc_pop_3)
39wm.add(u'>10亿', cc_pop_4)
40
41wm.render_to_file('world_population.svg')
42
43from pygal_maps_world.i18n import COUNTRIES
44
45def get_country_code(country_name):
46    for code, name in COUNTRIES.items():
47        if name == country_name:
48            return code
49    return None