На этом шаге мы рассмотрим использование параметров для задания и изменения цветовой схемы.
Если стоит задача изучить зависимость мощности автомобиля от года выпуска отдельно для стран США и Японии, то можно воспользоваться параметром hue для выделения стран цветом:
import matplotlib.pyplot as plt import seaborn as sns import pandas as pd sns.set_style("darkgrid") df = sns.load_dataset("mpg") sns.lineplot(x='model_year', y='horsepower', hue='origin', data=df[df['origin'] != 'europe']) plt.show()
Рис.1. Использование параметра hue для разделения данных по странам
Для дополнительной настройки цветового разделения можно использовать следующие параметры функции lineplot(): palette, hue_order, hue_norm.
Изменим цвет и зададим порядок:
import matplotlib.pyplot as plt import seaborn as sns import pandas as pd sns.set_style("darkgrid") df = sns.load_dataset("mpg") df_usa_jp = df[df['origin'] != 'europe'] sns.lineplot(x='model_year', y='horsepower', hue='origin', palette={'usa': 'r', 'japan': 'b'}, hue_order=['japan', 'usa'], data=df_usa_jp) plt.show()
Рис.2. Пример использования параметров palette и hue_order
На следующем шаге мы рассмотрим настройку стиля.