How to make a choropleth

country-websites.csv

plot.py

import pandas as pd
import plotly.express as px
import pycountry

def plot_chloropleth(csv_path):
	data = pd.read_csv(csv_path, names=['country', 'url'])

	data['country_iso'] = 'XXX'
	data['length'] = 0
	data['short_domain'] = 'XXX'

	for index, row in data.iterrows():

		try:
			# Search for three-letter country code
			country_search = pycountry.countries.search_fuzzy(row['country'])
			data.at[index,'country_iso'] = country_search[0].alpha_3
		except:
			pass

		temp_url = row['url']
		if 'http://' in temp_url:
			temp_url = temp_url.replace('http://', '')
		if 'https://' in temp_url:
			temp_url = temp_url.replace('https://', '')
		if 'www.' in temp_url:
			temp_url = temp_url.replace('www.', '')
		if '/' in temp_url:
			temp_url = temp_url.replace('/', '')

		data.at[index,'length'] = len(temp_url)
		data.at[index,'short_domain'] = temp_url

	fig = px.choropleth(data, locations="country_iso", # use our three-letter ISO codes as the locations
					color="length", # color the map by length
					hover_name="country",
					hover_data=['length', 'short_domain'], # add the character length and short_url as hover information
					labels={'length': 'Characters'}, # change the display name for length to Characters
					color_continuous_scale=px.colors.sequential.Brwnyl) # We'll use a brown-yellow colour palette

	fig.update_layout(
    		title_text='Number of characters in government website domain name',
    		legend_title_text='Characters'
	)

	fig.show()

plot_chloropleth('country-websites.csv')