python编程16章_[Python编程：从入门到实践] 第十六章：下载数据 习题答案

16-6 国内生产总值：Open Knowledge Foundation 提供了一个数据集，其中包含全球各国的国内生成总值(GDP)，可在http://data.okfn.org/data/core/gdp/找到这个数据集。请下载这个数据集的JSON版本，并绘制一个图标，将全球各国最近一年的GDP呈现出来。

"""我是用csv版本编的程"""

#!/usr/bin/env python3

# -*- coding: utf-8 -*-

from country_codes import get_country_code

from pygal_maps_world.maps import World

from pygal.style import RotateStyle as RS, LightColorizedStyle as LCS

import csv

# 将数据加载到一个列表中

filename = 'gdp.csv'

with open(filename) as f:

reader = csv.reader(f)

head_row = next(reader)

di_gdp = {}

for row in reader:

try:

country_name = row[0]

year = row[2]

value = int(float(row[3]))

except ValueError:

print('missing date')

else:

# 筛选年份为2015

if year == '2015':

code = get_country_code(country_name)

if code:

di_gdp[code] = value

# 根据人口数量进行分类

di_gdp1, di_gdp10, di_gdp100 = {}, {}, {}

for co, gdp in di_gdp.items():

if gdp < 1e+11:

di_gdp1[co] = gdp

elif gdp < 1e+12:

di_gdp10[co] = gdp

else:

di_gdp100[co] = gdp

# 呈现数据

wm_style = RS('#336699', base_style=LCS)

wm = World(style=wm_style)

wm.title = 'World GDP in 2015, by Country'

wm.add('0-10b', di_gdp1)

wm.add('10b-1tn', di_gdp10)

wm.add('>1tn', di_gdp100)

wm.render_to_file('world_gdp.svg')

16-8 测试模块 country_codes：我们编写模块country_codes时，使用了print语句来核实get_country_code()能否按预期的那样工作。请利用你在第11章学到的知识，为这个函数编写合适的测试。

"""Founction get_country_code"""

#!/usr/bin/env python3

# -*- coding: utf-8 -*-

from pygal.maps.world import COUNTRIES

def get_country_code(country_name):

"""查询国家名字对应的两位代码"""

for code, name in COUNTRIES.items():

if name == country_name:

return code

elif country_name == 'Bolivia':

return 'bo'

elif country_name == 'Congo, Dem. Rep.':

return 'cd'

elif country_name == 'Congo, Rep.':

return 'cg'

elif country_name == 'Dominica':

return 'do'

elif country_name == 'Egypt, Arab Rep.':

return 'eg'

elif country_name == 'Gambia, The':

return 'gm'

elif country_name == 'Hong Kong SAR, China':

return 'hk'

elif country_name == 'Iran, Islamic Rep.':

return 'ir'

elif country_name == 'Korea, Dem. Rep.':

return 'kp'

elif country_name == 'Korea, Rep.':

return 'kr'

elif country_name == 'Kyrgyz Republic':

return 'kg'

elif country_name == 'Lao PDR':

return 'la'

elif country_name == 'Libya':

return 'ly'

elif country_name == 'Macao SAR, China':

return 'mo'

elif country_name == 'Macedonia, FYR':

return 'mk'

elif country_name == 'Moldova':

return 'md'

elif country_name == 'Slovak Republic':

return 'sk'

elif country_name == 'Tanzania':

return 'tz'

elif country_name == 'Venezuela, RB':

return 've'

elif country_name == 'Vietnam':

return 'vn'

elif country_name == 'Yemen, Rep.':

return 'ye'

# 未找到则返回None

return None

# for code, name in COUNTRIES.items():

#     print(code + ': ' + name)

# print(get_country_code('China'))

"""Test country_code"""

import unittest

from country_codes import get_country_code

class TestCountryCodes(unittest.TestCase):

"""针对get_country_codes的测试类"""

def test_country_codes_inner(self):

country_name = 'Yemen, Rep.'

code = get_country_code(country_name)

self.assertEqual(code, 'ye')

def test_country_codes_outer(self):

country_name = 'Australia'

code = get_country_code(country_name)

self.assertEqual(code, 'au')

unittest.main()