Treemap with Python

Cedric Vidonne

Lei Chen

Treemap with Python

As a variation of a tree diagram, a treemap is meant to show hierarchical structure using the size of the rectangle to represent quantity. Each category is assigned a rectangle, with subcategories displayed inside the large rectangle, in proportionate size against each other.

More about: Treemap


# import libraries
import matplotlib.pyplot as plt
import squarify
import pandas as pd['unhcrpyplotstyle','treemap'])

#load and reshape the data
df = pd.read_csv('')

#compute data for plotting
sizes = df['staff_number']
label = df['region']

#calculate percentage
df['percent'] = (df['staff_number'] / 
                  df['staff_number'].sum()) * 100
list = df['percent']
new_list = [f'{i:.1f}% \n' for i in list]

#plot the chart
fig, ax = plt.subplots()
treemap = squarify.plot(sizes=sizes, label=new_list+label, color='#0072BC', ec='#ffffff', text_kwargs={"color":"#ffffff"})
noax = plt.axis('off')
#set chart title
plt.title('UNHCR global workforce by region | 2021')

#set chart source and copyright
plt.annotate('Source: UNHCR', (0,0), (0, -10), xycoords='axes fraction', textcoords='offset points', va='top', color = '#666666', fontsize=9)
plt.annotate('©UNHCR, The UN Refugee Agency', (0,0), (0, -20), xycoords='axes fraction', textcoords='offset points', va='top', color = '#666666', fontsize=9)

#adjust chart margin and layout

#show chart

A treemap showing UNHCR global workforce by region | 2021

Related chart with Python