Створення лійкової діаграми за допомогою plotly

Всіх вітаю! У даній статті ми поговоримо про те, як створювати лійкову діаграму за допомогою бібліотеки plotly у Python.

Встановлення бібліотеки plotly

Windows:

pip install plotly

MacOs:

pip3 install plotly

Linux:

pip install plotly

Створення діаграми

Тепер безпосередньо розглянемо процес створення лійкової діаграми. Для цього варто спочатку імпортувати бібліотеку:

from plotly import graph_objects as go

Тепер впроваджуємо список з кольорами для діаграми та словник із даними:

colors = ["#A594F9", "#CDC1FF", "#E5D9F2", "#F5EFFF"]
options = {"А": 200, "Б": 150, "В": 50, "Г": 30}

Тепер надаємо всі ці дані для створення самої діаграми:

figure = go.Figure(go.Funnelarea(
    labels=list(options.keys()),
    values=list(options.values()),
    marker=dict(colors=colors),
    textinfo="label+value"
    ))

Зауважте, що тут дані для labels та values записані у вигляді списків. Також важливо те, що да допомогою параметра textinfo ми відображаємо на діаграмі не лише варіанти, а й кількість респондентів, які обрали ці варіанти (label — варіанти, а value — кількість респондентів).

Далі переходимо до загальних налаштувань діаграми:

figure.update_layout(
    font=dict(size=13),
    margin=dict(t=130),
    title=dict(
        text="<b>Лійкова діаграма</b>",
        font=dict(size=20),
        subtitle=dict(text="Для прикладу було взято уявні варіанти відповідей та кількості респондентів",
                      font=dict(size=14))
    ),

    legend=dict(
        title="Варіанти відповідей"
    ),
    template="plotly_dark",
)

Якщо розбирати цю частину детальніше, то спочатку ми задаємо загальний розмір шрифту діаграми:

font=dict(size=13)

Тепер вказуємо відступ діаграми зверху:

margin=dict(t=130)

Додаємо заголовок та підзаголвок:

title=dict(
        text="<b>Лійкова діаграма</b>",
        font=dict(size=20),
        subtitle=dict(text="Для прикладу було взято уявні варіанти відповідей та кількості респондентів", font=dict(size=14))

)

Додаємо легенду:

legend=dict(
    title="Варіанти відповідей"
)

Задаємо темний фон для діаграми:

template="plotly_dark"

Відображаємо діаграму:

figure.show()

Тепер запускаємо даний код:

Висновок

У рамках даної статті було наочно продемонстровано використання бібліотеки plotly для створення гарних та зручних лійкових діаграм.
В результаті ми отримали цілком робочий код, який може бути корисним для подальшого застосування. Дякую за увагу!