#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