Pandas: как создавать и настраивать легенды сюжетов

Редакция Кодкампа -

Вы можете использовать следующий базовый синтаксис, чтобы добавить легенду к графику в pandas:

plt.legend(['A', 'B', 'C', 'D'], loc='center left', title='Legend Title')

В следующем примере показано, как использовать этот синтаксис на практике.

Пример: создание и настройка легенды графика в Pandas

Предположим, у нас есть следующие Pandas DataFrame:

import pandas as pd

#create DataFrame
df = pd.DataFrame({'A ':7, 'B':12, 'C ':15, 'D ':17}, index=['Values'])

Мы можем использовать следующий синтаксис для создания гистограммы для визуализации значений в DataFrame и добавления легенды с настраиваемыми метками:

import matplotlib.pyplot as plt

#create bar chart
df.plot (kind='bar')

#add legend to bar chart
plt.legend(['A Label', 'B Label', 'C Label', 'D Label'])

Мы также можем использовать аргумент loc и аргумент title , чтобы изменить местоположение и заголовок легенды:

import matplotlib.pyplot as plt

#create bar chart
df.plot (kind='bar')

#add custom legend to bar chart
plt.legend(['A Label', 'B Label', 'C Label', 'D Label'],
 loc='upper left', title='Labels')

Наконец, мы можем использовать аргумент size для изменения размера шрифта в легенде:

import matplotlib.pyplot as plt

#create bar chart
df.plot (kind='bar')

#add custom legend to bar chart
plt.legend(['A Label', 'B Label', 'C Label', 'D Label'], prop={' size ': 20 })

Обратите внимание, что размер шрифта в легенде теперь намного больше.

Дополнительные ресурсы

В следующих руководствах объясняется, как выполнять другие распространенные операции в pandas:

Как создать круговую диаграмму из Pandas DataFrame
Как сделать диаграмму рассеяния из Pandas DataFrame
Как создать гистограмму из Pandas DataFrame