Можно использовать:
Я заметил, что в pypi также есть библиотека treemap. Я ее не пробовал.
Вот пара вариантов: