DataVisualization1/popdata.py at master · svdmaar/DataVisualization1 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
import json, pygal
from country_code import get_country_code
#import country_code
from pygal.style import RotateStyle, LightColorizedStyle

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"]))
        #print(country_name + ": " + str(population))
        code = get_country_code(country_name)
        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 = RotateStyle("#336699", base_style=LightColorizedStyle)
#wm_style = RotateStyle("#336699")
#wm_style = LightColorizedStyle
wm = pygal.maps.world.World(style = wm_style)
wm.title = "World Population in 2010, by country"
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")