python编程--从json文件中读取并展示世界人口地图

Open Knowledge Foundation（http://data.okfn.org/）网站里面有大量可以免费下载的资源。

import json

import pygal_maps_world.maps
from pygal.style import LightColorizedStyle as LCS, RotateStyle as RS
from pygal.maps.world import COUNTRIES


def get_country_code(country_name):
    """将指定国家的名字以2字母形式表达"""
    for code, name in COUNTRIES.items():
        if name == country_name:
            return code
    # 如果没有找到那个国家
    return None


# 读取文件
filename = 'population_data.json'
with open(filename) as f:
    pop_data = json.load(f)

cc_populations = {}  # 将有代码的国家以一个字典形式储存
for pop_dict in pop_data:
    if pop_dict['Year'] == '2010':
        country_name = pop_dict['Country Name']
        population = int(float(pop_dict['Value']))  # 将字符串转变成float，然后转成int
        code = get_country_code(country_name)  # 得到国家的代码
        # 如果代码存在：
        if code:
            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():  # country code, country population
    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 =  pygal_maps_world.maps.World()
wm_style = RS('#336699', base_style=LCS)
wm = pygal_maps_world.maps.World()
wm.force_uri_protocol = 'http'
wm.title = 'World Population in 2010'
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')