Добавить легенду на барплот с градиентом, используя морского волка

Question

Я построил фрейм данных, используя seaborn, где цвет основан на значениях в столбце "loan_status".

Данные:

    Count   loan_status sub_grade
sub_grade           
A1  9729    0.971323    A1
A2  9567    0.951814    A2
A3  10576   0.941944    A3
A4  15789   0.929761    A4 

Печать:

plt.figure(figsize=(12,4))
g = sns.barplot(x="sub_grade", y= "Count", order= sub_grades, data=df, palette=cm.Blues(df['loan_status']))

Я хотел бы добавить легенду для цветового градиента этого графика.

Как мне это сделать?

Answer 1

Это то, что вы ищете?

вы можете создать цветовую панель на лету, как это

g = sns.barplot(x='sub_grade', y = 'Count', data = df, palette = cm.Blues(df['load_status']))

plt.colorbar(cm.ScalarMappable(cmap=cm.blues))