Matplotlib: средний уровень

Графики из DataFrame pandas

Превратите DataFrame pandas в график с помощью df.plot и настройте его средствами Matplotlib.

Введение

В pandas встроен Matplotlib: df.plot рисует график прямо из DataFrame и использует названия столбцов для легенды. Он возвращает Axes Matplotlib, так что всё изученное по-прежнему работает для заголовков, подписей и оформления.

x="month" кладёт месяцы на ось x, а y=[...] выбирает столбцы; kind выбирает тип графика ("line", "bar", "barh", "area"...). df.plot возвращает ax - обычный Axes Matplotlib, поэтому ax.set_ylabel и ax.set_title работают как раньше. rot=0 оставляет названия месяцев горизонтальными, а stacked=True сложил бы столбцы.

Интерактивный визуализатор

Меняйте kind, выбирайте столбцы и складывайте их. pandas рисует через Matplotlib и возвращает Axes, который можно дальше настраивать.

Запускаемый пример Matplotlib

Matplotlib
Запустить код →
main.py
import matplotlib.pyplot as plt
import pandas as pd

df = pd.DataFrame({
    "month": ["Jan", "Feb", "Mar", "Apr", "May", "Jun"],
    "coffee": [120, 110, 130, 150, 170, 160],
    "tea": [80, 85, 90, 70, 60, 65],
})

ax = df.plot(x="month", y=["coffee", "tea"], kind="bar", rot=0, figsize=(7, 4))
ax.set_ylabel("Cups sold")
ax.set_title("Drinks sold per month")
plt.show()
Получить кнопку запуска для этого примера График, который рисует этот код
Графики из DataFrame pandas