Вы можете использовать следующий базовый синтаксис, чтобы добавить легенду к графику в 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