Как создать график соотношения классов с помощью Seaborn?

Пользователь

от nikko , в категории: Python , 9 месяцев назад

Как создать график соотношения классов с помощью Seaborn?

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

2 ответа

Пользователь

от camren , 9 месяцев назад

@nikko 

Для создания графика соотношения классов с помощью библиотеки Seaborn можно использовать функцию countplot(). Эта функция позволяет отобразить количество наблюдений в каждой категории переменной.


Вот пример кода, показывающий, как создать график соотношения классов с помощью Seaborn:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
import seaborn as sns
import matplotlib.pyplot as plt

# Загрузка данных
data = sns.load_dataset('iris')

# Создание графика
sns.countplot(x='species', data=data)

# Добавление заголовка
plt.title('Class Distribution')

# Отображение графика
plt.show()


В данном примере используется набор данных "iris", который включен в библиотеку Seaborn. Функция countplot() принимает аргументы x и data, где x - это переменная, для которой нужно отобразить соотношение классов, а data - это данные, содержащие эту переменную.


После создания графика можно добавить заголовок с помощью функции title(). Затем график отображается с помощью функции show().


Таким образом, данный код создаст график соотношения классов переменной "species" в наборе данных "iris".

Пользователь

от shirley.vonrueden , месяц назад

@nikko 

Отличный пример! Можно также настроить внешний вид графика с помощью дополнительных параметров функции countplot(), таких как palette для выбора цветовой схемы, hue для разделения данных на группы, order для задания порядка отображения категорий и других.


Вот небольшой пример, как можно настроить оформление графика countplot():

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
import seaborn as sns
import matplotlib.pyplot as plt

# Загрузка данных
data = sns.load_dataset('iris')

# Создание графика с настроенными параметрами
sns.countplot(x='species', data=data, palette='Set1', order=data['species'].value_counts().index)

# Добавление заголовка и подписей к осям
plt.title('Class Distribution of Species in Iris Dataset')
plt.xlabel('Species')
plt.ylabel('Count')

# Отображение графика
plt.show()


Этот код создаст график соотношения классов переменной "species" в наборе данных "iris" с использованием цветовой схемы 'Set1' из библиотеки Seaborn. График будет отсортирован по убыванию количества наблюдений каждого класса.