Tour de France villes etape data exploration / map builder

requirements.txt

matplotlib

stages_TDF.csv

tdf_stages_for_gis.csv

villes_etape_geocoder.py

import csv
import json
import urllib.parse
import urllib.request
import matplotlib.pyplot as plt

villes_etape = []
villes_lat = []
villes_lon = []

api_key = ""

with open("stages_tdf.csv",
          encoding="utf-8") as f:
    stages_reader = csv.reader(f)
    first_row = False
    for row in stages_reader:
        if first_row:
            villes_etape.append(row[3])
            if row[3] != row[4]:
                villes_etape.append(row[4])
        first_row = True

set_villes_etape = list(set(villes_etape))
count_set_villes_etape = [villes_etape.count(x) for x in set_villes_etape]

fig = plt.figure()
ax = fig.add_axes([0, 0, 1, 1])
ax.bar(set_villes_etape, count_set_villes_etape)
plt.show()

for v in set_villes_etape:
    with urllib.request.urlopen(("http://api.positionstack.com/" +
                                 "v1/forward?access_key={}" +
                                 "&query={}").format(api_key,
                                                     urllib.parse.quote(v))) as r:
        api_resp_dict = json.loads(r.read().decode())
        try:
            villes_lat.append(api_resp_dict['data'][0]['latitude'])
            villes_lon.append(api_resp_dict['data'][0]['longitude'])
        except IndexError:
            print(api_resp_dict)
            villes_lat.append(None)
            villes_lon.append(None)
        except TypeError:
            print(v, api_resp_dict)    

with open('tdf_stages_for_gis.csv', 'w', encoding="utf-8") as file:
    writer = csv.writer(file)
    [writer.writerow(x) for x in zip(set_villes_etape,
                                     count_set_villes_etape,
                                     villes_lat,
                                     villes_lon)]