Создание графика 'Столбец в круге'

Введение

В этом уроке вы научитесь создавать график "Бар из пирога" с использованием библиотеки Matplotlib для Python. График "Бар из пирога" - это комбинация круговой диаграммы и накопленной столбчатой диаграммы, где первая секция пирога раскрывается в столбчатую диаграмму с дальней разбивкой ее характеристик. Этот график полезен, когда вы хотите показать распределение целого набора данных, выделяя при этом конкретные категории.

Советы по работе с ВМ

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

Иногда вам может потребоваться подождать несколько секунд, пока Jupyter Notebook не загрузится полностью. Валидация операций не может быть автоматизирована из-за ограничений в Jupyter Notebook.

Если вы сталкиваетесь с проблемами во время обучения, не стесняйтесь обращаться к Labby. Оставьте отзыв после занятия, и мы оперативно решим проблему для вас.

Импортируем необходимые библиотеки

Прежде чем начать создавать график, нам нужно импортировать необходимые библиотеки. В данном случае мы будем использовать matplotlib.pyplot и numpy.

import matplotlib.pyplot as plt
import numpy as np

Определяем данные для графика

Далее необходимо определить данные, которые мы будем использовать для создания графика. В данном случае мы будем использовать следующие данные:

## pie chart parameters
overall_ratios = [.27,.56,.17]
labels = ['Approve', 'Disapprove', 'Undecided']
explode = [0.1, 0, 0]

## bar chart parameters
age_ratios = [.33,.54,.07,.06]
age_labels = ['Under 35', '35-49', '50-65', 'Over 65']

Создаем круговую диаграмму

Теперь мы можем создать круговую диаграмму. Начнем с определения объектов фигуры и оси:

## make figure and assign axis objects
fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(9, 5))
fig.subplots_adjust(wspace=0)

Затем задаем параметры для круговой диаграммы и строим ее:

## rotate so that first wedge is split by the x-axis
angle = -180 * overall_ratios[0]
wedges, *_ = ax1.pie(overall_ratios, autopct='%1.1f%%', startangle=angle,
                     labels=labels, explode=explode)

Создаем столбчатую диаграмму

Далее мы создаем накопленную столбчатую диаграмму. Начнем с определения параметров для диаграммы:

## bar chart parameters
bottom = 1
width =.2

## Adding from the top matches the legend.
for j, (height, label) in enumerate(reversed([*zip(age_ratios, age_labels)])):
    bottom -= height
    bc = ax2.bar(0, height, width, bottom=bottom, color='C0', label=label,
                 alpha=0.1 + 0.25 * j)
    ax2.bar_label(bc, labels=[f"{height:.0%}"], label_type='center')

Соединяем круговую и столбчатую диаграммы

Наконец, мы соединяем круговую и столбчатую диаграммы с использованием ConnectionPatch:

## use ConnectionPatch to draw lines between the two plots
theta1, theta2 = wedges[0].theta1, wedges[0].theta2
center, r = wedges[0].center, wedges[0].r
bar_height = sum(age_ratios)

## draw top connecting line
x = r * np.cos(np.pi / 180 * theta2) + center[0]
y = r * np.sin(np.pi / 180 * theta2) + center[1]
con = ConnectionPatch(xyA=(-width / 2, bar_height), coordsA=ax2.transData,
                      xyB=(x, y), coordsB=ax1.transData)
con.set_color([0, 0, 0])
con.set_linewidth(4)
ax2.add_artist(con)

## draw bottom connecting line
x = r * np.cos(np.pi / 180 * theta1) + center[0]
y = r * np.sin(np.pi / 180 * theta1) + center[1]
con = ConnectionPatch(xyA=(-width / 2, 0), coordsA=ax2.transData,
                      xyB=(x, y), coordsB=ax1.transData)
con.set_color([0, 0, 0])
ax2.add_artist(con)
con.set_linewidth(4)

Отображаем график

Наконец, мы отображаем график:

plt.show()

Резюме

В этом руководстве вы узнали, как создать график "Столбец в круге" с использованием библиотеки Matplotlib для Python. График "Столбец в круге" полезен, когда вы хотите показать распределение целого набора данных, выделяя при этом определенные категории. Вы также узнали, как использовать ConnectionPatch для соединения круговой и столбчатой диаграмм.