Python 绘制2016世界GDP地图

2016世界GDP地图

从https://datahub.io/core/gdp#data下载得到json文件。

# country_code.py 获取国家二字代码

# 从pygal.maps.world模块导入{国家代码：国家名字}的列表
from pygal.maps.world import COUNTRIES 

import json

# 创建找国家代码的函数
def get_country_code(country):
    for code, name in COUNTRIES.items():
        if name == country:
            return code

    return None

if __name__ == '__main__':
    filename = 'world_GDP.json'
    with open(filename) as f:
        data = json.load(f)

    for dic in data:
        for key in dic:
            if key == 'Year':
                if dic[key] == 2016:
                    country = dic['Country Name']
                    print(country + ': ', get_country_code(country))

# world_GDP.py 世界2016年GDP

# 导入json模块
import json

# 从country_code.py文件导入get_country_code函数
from country_code import get_country_code

# 将pygal_maps_world.maps模块导入并命名为maps
import pygal_maps_world.maps as maps

# 导入数据
filename = 'world_GDP.json'
with open(filename) as f:
    data = json.load(f)

# 世界GDP字典
cc_gdp = {}

# 解析数据
for dic in data:
    # 遍历所有在data中的字典

    for key in dic:
        # 遍历字典的键

        if key == 'Year':
            # 如果键是'Year'

            if dic[key] == 2016:
                # 如果字典的Year键的值为2016

                country = dic['Country Name']
                value = dic['Value']
                code = get_country_code(country)
                if code:
                    cc_gdp[code] = value
                else:
                    pass

# 创建世界地图并设置标题以及标签数据
wm = maps.World()
wm.title = 'World GDP in 2016, by country'
wm.add('', cc_gdp)

# 渲染地图，保存为
wm.render_to_file('world_gdp.svg')

# 保存文件可以使用浏览器查看