Python Pygal繪製世界人口地圖

右介發表於2018-04-10
Python地圖

 

資料集可在 https://datahub.io/JohnSnowLabs/population-figures-by-country 下載

#coding=utf-8
import json
from country_codes import get_country_code
import pygal.maps.world
from pygal.style import RotateStyle as RS, LightColorizedStyle as LCS

# 將資料載入到列表中
filename = 'population_data.json'
with open(filename) as f:
    pop_data = json.load(f)

# 建立一個包含人口數量的字典
cc_populations = {}
for pop_dict in pop_data:
    country_name = pop_dict['Country_Name']
    population = pop_dict['Population_in_2016']
    code = get_country_code(country_name)
    if code:
        cc_populations[code] = population

# 根據人口數量將國家分成4組
cc_pop_1, cc_pop_2, cc_pop_3, cc_pop_4 = {}, {}, {}, {}
for cc, pop in cc_populations.items():
    if pop < 10000000:
        cc_pop_1[cc] = pop
    elif pop < 100000000:
        cc_pop_2[cc] = pop
    elif pop < 1000000000:
        cc_pop_3[cc] = pop
    else:
        cc_pop_4[cc] = pop

# wm_style = RS('#336699', base_style=LCS)
wm = pygal.maps.world.World()
wm.title = 'World Population in 2016, by Country'
wm.add(u'0-1千萬', cc_pop_1)
wm.add(u'1千萬-1億', cc_pop_2)
wm.add(u'1億-10億', cc_pop_3)
wm.add(u'>10億', cc_pop_4)

wm.render_to_file('world_population.svg')

 

from pygal_maps_world.i18n import COUNTRIES

def get_country_code(country_name):
    for code, name in COUNTRIES.items():
        if name == country_name:
            return code
    return None

 

相關文章