我有一个数据框如下：
在上面的数据框中，'Month'是一个有序的Categorical列定义为：
cats = ['January', 'February', 'March', 'April','May','June', 'July', 'August','September', 'October', 'November', 'December']
month_gr['Month'] = pd.Categorical(month_gr['Month'], cats, ordered = True)
使用 Seaborn 条形图：
ax = sns.barplot(data = month_gr, x = 'Item Name', y = 'Total', hue = 'Month')
ax.set_xticklabels(ax.get_xticklabels(), rotation= 90, ha = 'right')
Outputs:
上面的图例显示了 12 个月的全部Categorical柱子。我只想显示图例 4 个月['June', 'July', 'August', 'September']，因为我的数据只包含这 4 个月。有没有办法动态控制图例，使其仅显示可用的Categories传递给data?
您可以创建“已用月份”列表，然后将该列表设置为hue_order。这也确保只有那些月份才会占用酒吧的空间。
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
month_col = ['June'] * 5 + ['July'] * 5 + ['August'] * 5 + ['September'] * 7
month_gr = pd.DataFrame({'Month': month_col,
'Item Name': [*'abcdebdefgbcefgabcdefg'],
'Total': np.random.randint(100, 1000, len(month_col))})
cats = ['January', 'February', 'March', 'April', 'May', 'June', 'July', 'August', 'September', 'October', 'November', 'December']
month_gr['Month'] = pd.Categorical(month_gr['Month'], cats, ordered=True)
used_months = [m for m in cats if m in month_gr['Month'].to_list()]
ax = sns.barplot(data=month_gr, x='Item Name', y='Total',
hue='Month', hue_order=used_months, palette=sns.color_palette("Set2"))
plt.show()
