Раскрасьте значения x и покажите в легенде, а не в виде галочек, в matplotlib (или seaborn)

Question

У меня есть большой df, который я сгруппировал для построения графика в гистограмме.Я сделал этот макет DF, чтобы показать, что я имею в виду.(И мне было очень весело создавать его ...)

my = pd.DataFrame(
{'names': ['Andrea', 'Donna', 'Kelly', 'Brenda', 'Allison', 'Jo', 'Amanda', 'Jane', 'Kerry', 'Abby', 'Elizabeth', 'Haleh'],
'episodes': [ 147, 292, 292, 111, 160, 111, 199, 172, 250, 189, 160,184 ],
'tv-show' : ['Beverly Hills, 90210', 'Beverly Hills, 90210', 'Beverly Hills, 90210', 'Beverly Hills, 90210',
'Melrose place', 'Melrose place', 'Melrose place', 'Melrose place',
'ER', 'ER', 'ER', 'ER']})
my

А потом я сгруппировал и нанес на график: my.groupby('tv-show').sum().plot(kind='bar', stacked = True)

ЧтоМне бы хотелось, чтобы это был сюжет, где названия телешоу в легенде, а не под осью X, и где шоу имеют разные цвета (конечно).

Answer 1

Непосредственно с помощью панд:

ax = my.groupby('tv-show').sum().transpose().plot.bar()
ax.set_xticks([])

Answer 2

Попробуйте это с sns:

new_df = my.groupby('tv-show').sum().reset_index()

sns.barplot(x='tv-show', y='episodes', 
            hue='tv-show', data=new_df)

Выход:

Answer 3

Другое альтернативное решение с использованием matplotlib может выглядеть примерно так:

import matplotlib.patches as mpatches

fig, ax = plt.subplots(figsize=(8, 6))

# Your dataframe "my" here

ax_ = my.groupby('tv-show').sum().plot(kind='bar', stacked=True, legend=False, ax=ax) 

colors = ['r', 'g', 'b']
handles = []
for col, lab, patch in zip(colors, np.unique(my['tv-show']), ax_.axes.patches):
    patch.set_color(col)
    handles.append(mpatches.Patch(color=col, label=lab))

ax_.legend(handles=handles)    
ax_.set_xticklabels([])