2017-10-22 785 views 评论

基于python绘制世界人口地图

主程序：

#&nbsp;-*-&nbsp;coding:&nbsp;utf-8&nbsp;-*-
&quot;&quot;&quot;
Created&nbsp;on&nbsp;Sun&nbsp;Oct&nbsp;22&nbsp;19:57:24&nbsp;2017

@author:&nbsp;dqhpl
&quot;&quot;&quot;
import&nbsp;json
import&nbsp;pygal.maps.world
from&nbsp;country_codes&nbsp;import&nbsp;get_country_code
from&nbsp;pygal.style&nbsp;import&nbsp;RotateStyle&nbsp;as&nbsp;RS,&nbsp;LightColorizedStyle&nbsp;as&nbsp;LCS&nbsp;#利用as创建简写，便于书写

#将数据加载到一个列表之中
filename&nbsp;=&nbsp;&#39;population_data.json&#39;
with&nbsp;open(filename)&nbsp;as&nbsp;f:
&nbsp;&nbsp;&nbsp;&nbsp;pop_data&nbsp;=&nbsp;json.load(f)
#创建一个包含人口数量的字典
cc_populations&nbsp;=&nbsp;{}
#打印每个国家2015年的人口数量
for&nbsp;pop_dict&nbsp;in&nbsp;pop_data:
&nbsp;&nbsp;&nbsp;&nbsp;if&nbsp;pop_dict[&#39;Year&#39;]&nbsp;==&#39;2010&#39;:
&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;country&nbsp;=&nbsp;pop_dict[&#39;Country&nbsp;Name&#39;]
&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;population=&nbsp;int(float(pop_dict[&#39;Value&#39;]))#由于直接转int会出错。所以先转float再转int
&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;#print(country_name&nbsp;+&nbsp;&quot;:&nbsp;&quot;&nbsp;+&nbsp;str(population))
&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;code=&nbsp;get_country_code(country)
&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;if&nbsp;code:
&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;cc_populations[code]&nbsp;=&nbsp;population

#根据人口数量将所有的国家分为三组
cc_pops_1,cc_pops_2,cc_pops_3&nbsp;=&nbsp;{},{},{}
for&nbsp;cc,&nbsp;pop&nbsp;in&nbsp;cc_populations.items():
&nbsp;&nbsp;&nbsp;&nbsp;if&nbsp;pop&nbsp;&lt;10000000:
&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;cc_pops_1[cc]&nbsp;=&nbsp;pop
&nbsp;&nbsp;&nbsp;&nbsp;elif&nbsp;pop&nbsp;&lt;1000000000:
&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;cc_pops_2[cc]&nbsp;=&nbsp;pop
&nbsp;&nbsp;&nbsp;&nbsp;else:
&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;cc_pops_3[cc]&nbsp;=&nbsp;pop
#看看每组分别有多少个国家
print(len(cc_pops_1),&nbsp;len(cc_pops_2),&nbsp;len(cc_pops_3))


wm_style&nbsp;=&nbsp;RS(&#39;#336699&#39;,&nbsp;base_style=LCS)#创建了一种基调，LCS指的是用亮色风格
wm&nbsp;=&nbsp;pygal.maps.world.World(style=&nbsp;wm_style)#pygal升级，所以需要注意&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;
wm.title&nbsp;=&nbsp;&#39;World&nbsp;Population&nbsp;in&nbsp;2010,&nbsp;by&nbsp;Country&#39;
wm.add(&#39;0-10m&#39;,&nbsp;cc_pops_1)
wm.add(&#39;10m-1bn&#39;,&nbsp;cc_pops_2)
wm.add(&#39;&gt;10bn&#39;,&nbsp;cc_pops_3)

wm.render_to_file(&#39;world_population.svg&#39;)

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

"""

Created on Sun Oct 22 19:57:24 2017

@author: dqhpl

"""

import json

import pygal.maps.world

from country_codes import get_country_code

from pygal.style import RotateStyle as RS, LightColorizedStyle as LCS #利用as创建简写，便于书写

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

filename = 'population_data.json'

with open(filename) as f:

    pop_data = json.load(f)

#创建一个包含人口数量的字典

cc_populations = {}

#打印每个国家2015年的人口数量

for pop_dict in pop_data:

    if pop_dict['Year'] =='2010':

        country = pop_dict['Country Name']

        population= int(float(pop_dict['Value']))#由于直接转int会出错。所以先转float再转int

        #print(country_name + ": " + str(population))

        code= get_country_code(country)

        if code:

            cc_populations[code] = population

#根据人口数量将所有的国家分为三组

cc_pops_1,cc_pops_2,cc_pops_3 = {},{},{}

for cc, pop in cc_populations.items():

    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_style = RS('#336699', base_style=LCS)#创建了一种基调，LCS指的是用亮色风格

wm = pygal.maps.world.World(style= wm_style)#pygal升级，所以需要注意                     

wm.title = 'World Population in 2010, by Country'

wm.add('0-10m', cc_pops_1)

wm.add('10m-1bn', cc_pops_2)

wm.add('>10bn', cc_pops_3)

wm.render_to_file('world_population.svg')

country_codes.py

#&nbsp;-*-&nbsp;coding:&nbsp;utf-8&nbsp;-*-
&quot;&quot;&quot;
Created&nbsp;on&nbsp;Sun&nbsp;Oct&nbsp;22&nbsp;20:09:53&nbsp;2017

@author:&nbsp;dqhpl
&quot;&quot;&quot;
#&nbsp;=============================================================================
#&nbsp;The&nbsp;i18n&nbsp;module&nbsp;was&nbsp;removed&nbsp;in&nbsp;pygal-2.0.0,&nbsp;
#however,&nbsp;it&nbsp;can&nbsp;now&nbsp;be&nbsp;found&nbsp;in&nbsp;the&nbsp;pygal_maps_world&nbsp;plugin.
#&nbsp;
#&nbsp;You&nbsp;can&nbsp;install&nbsp;that&nbsp;with&nbsp;pip&nbsp;install&nbsp;pygal_maps_world.&nbsp;
#Then&nbsp;you&nbsp;can&nbsp;access&nbsp;COUNTRIES&nbsp;as&nbsp;pygal.maps.world.COUNTRIES:
#&nbsp;
#&nbsp;from&nbsp;pygal.maps.world&nbsp;import&nbsp;COUNTRIES
#&nbsp;Whats&nbsp;left&nbsp;of&nbsp;the&nbsp;i18n&nbsp;module&nbsp;can&nbsp;be&nbsp;imported&nbsp;with:
#&nbsp;
#&nbsp;from&nbsp;pygal_maps_world&nbsp;import&nbsp;i18n
#&nbsp;=============================================================================
from&nbsp;pygal_maps_world&nbsp;import&nbsp;i18n
from&nbsp;pygal.maps.world&nbsp;import&nbsp;COUNTRIES

def&nbsp;get_country_code(country_name):
&nbsp;&nbsp;&nbsp;&nbsp;&quot;&quot;&quot;根据指定的国家，返回pygal使用的两个字母的国别码&quot;&quot;&quot;
&nbsp;&nbsp;&nbsp;&nbsp;for&nbsp;code,&nbsp;name&nbsp;in&nbsp;COUNTRIES.items():
&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;if&nbsp;name&nbsp;==&nbsp;country_name:
&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;return&nbsp;code
&nbsp;&nbsp;&nbsp;&nbsp;#如果没有找到指定的国家，就返回None
&nbsp;&nbsp;&nbsp;&nbsp;return&nbsp;None