Не понимаю причину этого исключения TypeError - PullRequest
Новая
       49

Не понимаю причину этого исключения TypeError

0 голосов
/ 21 февраля 2020

Импорт правильных модулей для этой программы, а затем преобразование 'gdp.json' в объект, который python может использовать с json.load(f). 

import json
import pygal
from pygal.style import LightColorizedStyle as LCS,RotateStyle as RC
from pygal.maps.world import World
from country_codes import get_country_code
#load data into a list
filename = 'gdp.json'
with open(filename) as f:
    gdp_data = json.load(f)

Создание словаря gdp_data: 

cc_gdps = {}
for gdp_dict  in gdp_data:
    if gdp_dict['Year'] == 2014:
        country_name = gdp_data["Country Name"]
        gdp = int(float(gdp_data['Value']))
        code = get_country_code(country_name)
    if code:
        cc_gdps[code] = gdp
#Group the countries into 3 gdp level
cc_gdps_1,cc_gdps_2,cc_gdps_3 = {},{},{}
for cc,gdp in cc_gdps.items():
    if gdp < 5000000000:
        cc_gdps_1[cc]=round(gdp/1000000000)
    elif gdp < 5000000000:
        cc_gdps_2[cc] = round(gdp/1000000000)
    else:
        cc_gdps_3[cc] = round(gdp/1000000000)
#see how many countries are in each level
print(len(cc_gdps_1),len(cc_gdps_2),len(cc_gdps_3))
wm_style = RC('#336699',base_style=LCS)
wm = World(style = wm_style)
wm.title = 'Global GDP in 2014, by country.'
wm.add('0-5bln',cc_gdps_1)
wm.add('5bln-50bln',cc_gdps_2)
wm.add('>50bln', cc_gdps_3)
wm.render_to_file('global_gdp.svg')

Вот метод get_country_code(country_name): 

from pygal.maps.world import COUNTRIES
def get_country_code(country_name):
    """Return the pygal 2-digit country code for given country."""
    for code,name in COUNTRIES.items():
        if name == country_name:
            return code
    #if the country wasnt found,return none.
    return None

Не уверен, что это может быть ...

Ответы [ 2 ]

0 голосов
/ 22 февраля 2020

Это вызывает проблему: 

country_name = gdp_data["Country Name"]
#                            ^^^

gdp_data относится к типу список , следовательно, он принимает только целые числа или фрагменты, строка в качестве индекса не имеет смысла для списка .

0 голосов
/ 21 февраля 2020

Я думаю, что есть проблема в вашей функции get_country_code. Измените это так. 

def get_country_code(country_name):
    """Return the pygal 2-digit country code for given country."""
    for code,name in COUNTRIES.items():
        if name == country_name:
            return code
        else:
            #if the country wasnt found,return none.
            return None
...