Удалить определенные категории из Seaborn Boxplot

Question

У меня есть этот код:

plt.figure(figsize=(12,7))
sns.boxplot(data=resolved_time_length,
x='Time_to_resolve_days',y='Ticket_category',
       linewidth=2.5)#, hue='Priority_level')#,color='pink')

plt.xlim(-1,30)

plt.title('Average Time to Resolve an IoT Ticket in Days',fontsize=20)
plt.xlabel('Resolve Time in Days',fontsize=14)
plt.ylabel('Ticket Category', fontsize=14)
plt.show()

и выдает этот график:

Все, что я хочу сделать, это не показывать ни одной категории на оси Y, которая имеет менее 5 наблюдений (в данном случае это категории Functionality, Internal-Add и Internal-Data). Я попытался, установив ось Y, чтобы:

y=Time_to_resolve_days['Ticket_category']<5 

но это не работает, поэтому мне нужна ваша помощь, пожалуйста!

Answer 1

Редактировать: Проверено, работает следующее:

plt.figure(figsize=(12,7))


#create a filtered dataframe
data=resolved_time_length.drop(columns=resolved_time_length.apply(pd.Series.nunique, axis=0).index[resolved_time_length.apply(pd.Series.nunique, axis=0)<5])


#plot filtered dataframe
sns.boxplot(data=data,
x='Time_to_resolve_days',y='Ticket_category',
       linewidth=2.5)#, hue='Priority_level')#,color='pink')


plt.xlim(-1,30)

plt.title('Average Time to Resolve an IoT Ticket in Days',fontsize=20)
plt.xlabel('Resolve Time in Days',fontsize=14)
plt.ylabel('Ticket Category', fontsize=14)
plt.show()

Answer 2

Вы можете фильтровать Pandas DataFrame, используя pandas.DataFrame.loc¶ :

resolved_time_length.loc[resolved_time_length['Time_to_resolve_days'] < 5]

Затем используйте отфильтрованный DataFrame в качестве входных данных:

sns.boxplot(data=resolved_time_length.loc[resolved_time_length['Time_to_resolve_days'] < 5], 
x='Time_to_resolve_days',
y='Ticket_category', linewidth=2.5)#, 
hue='Priority_level')#,color='pink')

Больше информации здесь: https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.loc.html