Легенда Matplotlib Axes показывает только один ярлык в Барх

Question

У меня есть 15 барговых сюжетов, которые выглядят так:

Не могу заставить легенду работать, поэтому я посмотрю[2,3,4] в виде отдельных меток на графике и в легенде.

У меня проблемы с выполнением этой работы для подграфов.Мой код:

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
def plot_bars_by_data(data, title):
    fig, axs = plt.subplots(8,2, figsize=(20,40))
    fig.suptitle(title, fontsize=20)
    fig.subplots_adjust(top=0.95)
    plt.rcParams.update({'font.size': 13})
    axs[7,1].remove()

    column_index = 0
    for ax_line in axs:
        for ax in ax_line:
            if column_index < len(data.columns): 
                column_name = data.columns[column_index]
                current_column_values = data[column_name].value_counts().sort_index()
                ax.barh([str(i) for i in current_column_values.index], current_column_values.values)
                ax.legend([str(i) for i in current_column_values.index])
                ax.set_title(column_name)
                column_index +=1

    plt.show()

# random data
df_test = pd.DataFrame([np.random.randint(2,5,size=15) for i in range(15)], columns=list('abcdefghijlmnop'))
plot_bars_by_data(df_test, "testing")

Я просто получаю столбики 8x2, которые выглядят как на графике выше.Как я могу это исправить?Я использую Python 3.6 и ноутбук Jupyter Python.

Answer 1

Используйте следующие строки в вашем коде.Я не могу поместить весь вывод здесь как большую фигуру с большим количеством подзаговоров и, следовательно, показывающую конкретный подзаговор.Оказывается, сначала вы должны создать дескриптор для своего подзаговора, а затем передать значения легенды и дескриптор для получения желаемых легенд.

colors = ['r', 'g', 'b']
axx = ax.barh([str(i) for i in current_column_values.index], current_column_values.values, color=colors)
ax.legend(axx, [str(i) for i in current_column_values.index])

Пример вывода