Как построить Tureewap в Python?

Cureemap в Python – это визуализация данных, которые разделяют прямоугольник в подпэп. Размер каждого поддателя пропорционален данным, которые он представляет. Это немного похоже на пирог. Хотя, Treewaps могут представлять гораздо более сложные данные по сравнению с пирогарным диаграммой.

Это может помочь вам визуализировать, как одно значения составляют целое. Treeemap Диаграммы также позволяют визуализировать иерархические данные с использованием вложенных прямоугольников.

В этом руководстве мы узнаем, как списать Treewaps в Python, используя библиотеку Squarify в Python.

Давайте начнем с установки Squarify.

pip install squarify

Используя скважин, чтобы построить TREEWAP в Python

Как только мы установили Squarify, мы можем начать, импортируя его в нашу ноутбук. Давайте также импортируем matplotlib.

import matplotlib.pyplot as plt
import squarify

1. Построение базового TREEWAP

Чтобы построить очень Basic Treemap, нам просто нужны значения для каждого прямоугольника. После построения TREEWAP прямоугольники будут пропорциональны эти значения.

import matplotlib.pyplot as plt
import squarify 
sizes = [40, 30, 5, 25]
squarify.plot(sizes)
plt.show()

2. Добавьте этикетки в ваш TREEWAP

Вы можете добавлять этикетки в Treemap в Python, используя следующие строки кода:

import matplotlib.pyplot as plt
import squarify 
sizes=[40, 30, 5, 25]
label=["A", "B", "C", "D"]
squarify.plot(sizes=sizes, label=label, alpha=0.6 )
plt.show()

Если вы снова запустите один и тот же кусок кода, вы получите следующий вывод:

Вы можете видеть, что цветовая гамма нашего TREEWAP отличается каждый раз, когда мы запускаем его. Цвета для прямоугольников выбираются случайным образом Отказ Treemap также дает вам возможность упомянуть цвета вместе с размерами и метками. Мы узнаем, как изменить цвета TREEWAP Next.

3. Измените цвета в вашем TREEWAP

Чтобы изменить цвета в TRUEWAP в Python, сделайте список цветов, которые вы хотите иметь TREEWAP. Затем пройдите этот список, чтобы Squarify.Plot метод.

import matplotlib.pyplot as plt
import squarify 
sizes=[40, 30, 5, 25]
label=["A", "B", "C", "D"]
color=['red','blue','green','grey']
squarify.plot(sizes=sizes, label=label, color=color, alpha=0.6 )
plt.show()

4. Выключите осью раза

Чтобы записать TREEWAP без оси сюжета, используйте:

plt.axis('off')

Эта строка кода выключит ось участка. Полный код выглядит следующим образом:

import matplotlib.pyplot as plt
import squarify 
sizes=[40, 30, 5, 25]
label=["A", "B", "C", "D"]
color=['red','blue','green','grey']
squarify.plot(sizes=sizes, label=label, color=color, alpha=0.6 )
plt.axis('off')
plt.show()

Груз TREEWAP для набора данных

В этой части учебника мы узнаем, как построить TREEWAP для набора данных. Мы собираемся использовать титанический набор данных. Давайте начнем, импортируя набор данных. Чтобы упростить процесс импорта набора данных, мы собираемся использовать морской библиотека.

1. Импорт набора данных

Чтобы импортировать набор данных Titian от библиотеки Seborbor в вашу тетрадь Python, используйте:

import seaborn as sns
titanic = sns.load_dataset('titanic')
titanic.head()

Набор данных содержит информацию о пассажирах Титаника.

Мы хотим построить TREEWAP для людей, которые выжили в соответствии с классом, в котором они путешествовали.

Данные в исходном формате не готов к построению TREEWAP. Мы проведем некоторые манипуляции и попытаемся извлечь данные, которые мы можем использовать для заговора TREEWAP.

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

2. Подготовка данных для построения

Вы можете использовать функцию Groupby на DataSet, как показано ниже:

n = titanic.groupby('class')[['survived']].sum()

Это дает нам сумму общего выживших, сгруппированных в соответствии с классом.

Теперь нам нужно извлечь данные и этикетки в виде списков от этого.

a = titanic.groupby('class')[['survived']].sum().index.get_level_values(0).tolist()
print(a)

Выход:

['First', 'Second', 'Third']

Это дает нам этикетки в виде списка. Чтобы получить значения, соответствующие этим этикеткам, используйте:

d = titanic.groupby('class')[['survived']].sum().reset_index().survived.values.tolist()
print(d)

Выход:

[136, 87, 119]

Теперь у нас есть этикетки и данные в виде списков. Мы можем использовать их, чтобы построить TREEWAP.

3. Сюжет Truewap

Чтобы построить TREEWAP, используйте следующую строку кода:

squarify.plot(sizes=d, label=a, alpha=.8)
plt.axis('off')
plt.show()

Визуализируя Treemap, мы можем получить грубую идею о количестве выживших в первом, втором и третьего класса. Просто посмотрев на Treemap, мы можем уверенно сказать, что второй класс имеет наименьшее количество выживших.

Заполните код, чтобы построить TREEWAP в Python

Полный код из этого раздела приведен ниже:

import seaborn as sns
import squarify 
import matplotlib.pyplot as plt

titanic = sns.load_dataset('titanic')

a = titanic.groupby('class')[['survived']].sum().index.get_level_values(0).tolist()

d = titanic.groupby('class')[['survived']].sum().reset_index().survived.values.tolist()

squarify.plot(sizes=d,label=a, alpha=.8 )
plt.axis('off')
plt.show()

Заключение

В этом руководстве мы узнали, как построить Tureewap в Python, используя квадрат. Надеюсь, вы веселились с нами.