compute_per_capita_gdp.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
"""
Created on Thu Jun 29 11:26:56 2017

@author: saliksyed
"""
import csv

gdp_data = csv.reader(open("gdp.dat"))

gdp_by_country_code = {}
for row in gdp_data:
    country = row[0]
    country_code = row[1]
    gdp_values = []
    for value in row[5:]:
        # We need to do this because there is some missing data which is represented by ""
        if value == "":
            final_parsed_value = None
        else:
            final_parsed_value = float(value)

        gdp_values.append(final_parsed_value)
    
    country_data = {}
    
    country_data["country"] = country
    country_data["country_code"] = country_code
    country_data["gdp_values"] = gdp_values
    gdp_by_country_code[country_code] = country_data
    


population_data = csv.reader(open("population.dat","r"))

population_by_country_code = {}
for row in population_data:
    country = row[0]
    country_code = row[1]
    population_data = []
    for value in row[5:]:
        if value == "":
            population_data.append(None)
        else:
            population_data.append(float(value))
    country_data = {}   
    country_data["country"] = country
    country_data["country_code"] = country_code
    country_data["population_values"] = population_data
    population_by_country_code[country_code] = country_data
    


per_capita_gdp_by_country_code = {}
for country_code in population_by_country_code.keys():
    if country_code in population_by_country_code and country_code in gdp_by_country_code:
        gdp = gdp_by_country_code[country_code]["gdp_values"]
        population = population_by_country_code[country_code]["population_values"]
        name = gdp_by_country_code[country_code]["country"]
        gdp_per_capita = []
        for i in xrange(0, len(gdp)):
            if gdp[i] and population[i]:
                gdp_per_capita.append(gdp[i]/population[i])
            else:
                gdp_per_capita.append(None)
        country_data = {}
        country_data["country"] = country
        country_data["country_code"] = country_code
        country_data["gdp_per_capita"] = gdp_per_capita
        per_capita_gdp_by_country_code[country_code] = country_data

print per_capita_gdp_by_country_code