На этом шаге мы рассмотрим настройку легенды.
Рассмотрим несколько полезных инструментов для настройки элементов поля линейного графика, а именно: легенды и подписей осей. Более подробно про эти и другие инструменты можете прочитать, начиная с 73 шага.
За отображение легенды отвечает параметр legend, он может принимать следующие значения:
Рассмотрим пример: в наборе данных mpg признак cylinders принимает значения из множества {3, 4, 5, 6, 8}:
df = sns.load_dataset("mpg") print(set(df["cylinders"]))
{3, 4, 5, 6, 8}
Отобразим график с цветовым разделением по этому признаку с сжатым вариантом представления легенды:
import matplotlib.pyplot as plt import seaborn as sns df = sns.load_dataset("mpg") print(set(df["cylinders"])) sns.lineplot(x='model_year', y='horsepower', hue='cylinders', data=df, legend='brief') plt.show()
Рис.1. Представление сокращённого набора значений признака в легенде
Как вы можете видеть, в легенде отображён диапазон значений от 2 до 8. Теперь построим тот же график, но параметру legend присвоим значение 'full':
import matplotlib.pyplot as plt import seaborn as sns df = sns.load_dataset("mpg") print(set(df["cylinders"])) sns.lineplot(x='model_year', y='horsepower', hue='cylinders', data=df, legend='full') plt.show()
Рис.2. Представление полного набора значений признака в легенде
Теперь в легенде перечислены все значения признака cylinders. Уберём легенду с поля графика, для этого параметру legend присвоим значение False:
import matplotlib.pyplot as plt import seaborn as sns df = sns.load_dataset("mpg") print(set(df["cylinders"])) sns.lineplot(x='model_year', y='horsepower', hue='cylinders', data=df, legend=False) plt.show()
Рис.3. Поле графика без легенды
На следующем шаге мы рассмотрим подписи осей.