На этом шаге мы рассмотрим параметры, используемые для настройки цветов.
Выделим цветом страну производителя:
import matplotlib.pyplot as plt import seaborn as sns mpg = sns.load_dataset("mpg") sns.scatterplot(x='mpg', y='displacement', hue='origin', data=mpg) plt.show()
Рис.1. Использование параметра hue для разделения данных по странам
За дополнительную настройку цветовой схемы отвечают параметры palette, hue_order и hue_norm, их назначение и порядок использования аналогичен одноимённым параметрам функции lineplot() (см. 91 шаг). Для демонстрации работы с ними приведём несколько примеров. Изменим цветовую палитру:
import matplotlib.pyplot as plt import seaborn as sns mpg = sns.load_dataset("mpg") sns.scatterplot(x='mpg', y='displacement', hue='origin', palette='plasma', data=mpg) plt.show()
Рис.2. Пример использование цветовой палитры plasma
Зададим свой набор цветов:
import matplotlib.pyplot as plt import seaborn as sns mpg = sns.load_dataset("mpg") cmap = {'usa': 'y', 'japan': 'g', 'europe': 'r'} sns.scatterplot(x='mpg', y='displacement', hue='origin', palette=cmap, data=mpg) plt.show()
Рис.3. Пример работы с предварительно подготовленной цветовой схемой
Изменим порядок применения цветов:
import matplotlib.pyplot as plt import seaborn as sns mpg = sns.load_dataset("mpg") order = ['japan', 'europe', 'usa'] sns.scatterplot(x='mpg', y='displacement', hue='origin', hue_order=order, data=mpg) plt.show()
Рис.4. Изменение порядка применения цветов
Обратите внимание на легенду, в отличие от предыдущих примеров перечисление стран теперь идёт в порядке, указанном в переменной order.
На следующем шаге мы рассмотрим настройку стиля маркеров.