Ошибка при составлении карты мира для отдельных стран с использованием фолиума - PullRequest
Новая
       26

Ошибка при составлении карты мира для отдельных стран с использованием фолиума

0 голосов
/ 13 октября 2019

Я пытаюсь построить картографию отдельных стран с использованием фолиума питона, как указано ниже: 

#!/usr/bin/python3 
import pandas as pd
import os
import folium

country_count = pd.read_csv('countries.csv')

country_geo = os.path.join('world-countries.json')

m = folium.Map(location=[0, 0], zoom_start=2)
m.choropleth(
 geo_data=country_geo,
 name='choropleth',
 data=country_count,
 columns=['Country_0'],
 key_on='feature.properties.name',
 fill_color='YlGn',
 fill_opacity=0.7,
 line_opacity=0.2,
 legend_name='Test'
)
folium.LayerControl().add_to(m)

m.save('#292_folium_chloropleth_country.html')

, где country.csv приведено ниже: 

Country                 
United States of America 
United Kingdom              
France                      
Canada                      
Japan                       
Italy                       
Germany                     
Russia                      
India                       
Spain                       
Australia                   
Hong Kong                   
South Korea

, когда я пытаюсь скомпилировать этот код, я получаю эту ошибку, которая показана ниже: 

Traceback (most recent call last):
  File "choropleth1.py", line 20, in <module>
    legend_name='Test'
  File "/home/ammar/.local/lib/python3.7/site-packages/folium/folium.py", line 418, in choropleth
    self.add_child(Choropleth(*args, **kwargs))
  File "/home/ammar/.local/lib/python3.7/site-packages/folium/features.py", line 1063, in __init__
    color_data = data.set_index(columns[0])[columns[1]].to_dict()
  File "/home/ammar/.local/lib/python3.7/site-packages/pandas/core/frame.py", line 4396, in set_index
    raise KeyError("None of {} are in the columns".format(missing))
KeyError: "None of ['Country'] are in the columns"

Есть ли какое-либо решение, которое я могу сделать choropleth карта для конкретных стран с использованием их названий или трехбуквенных кодов, таких как США для Соединенных Штатов Америки?

...