Измените условное обозначение графика pandas

Меня всегда беспокоит, когда я делаю штрих с pandas, и я хочу изменить имена меток в легенде. Рассмотрим, например, вывод этого кода:

import pandas as pd
from matplotlib.pyplot import *

df = pd.DataFrame({'A':26, 'B':20}, index=['N'])
df.plot(kind='bar')

Теперь, если я хочу изменить имя в легенде, я обычно стараюсь:

legend(['AAA', 'BBB'])

Но в итоге я получаю следующее:

Фактически, первая пунктирная линия, по-видимому, соответствует дополнительному патчу.

Итак, я задаюсь вопросом, есть ли здесь простой трюк, чтобы изменить ярлыки, или мне нужно построить каждый из столбцов самостоятельно с помощью matplotlib и сами установить метки. Спасибо.

Ответы

Ответ 1

Изменение меток для Pandas df.plot():

import pandas as pd
from matplotlib.pyplot import *

fig, ax = subplots()
df = pd.DataFrame({'A':26, 'B':20}, index=['N'])
df.plot(kind='bar', ax=ax)
ax.legend(["AAA", "BBB"]);

Edit:

Одна строка меньше:

df = pd.DataFrame({'A':26, 'B':20}, index=['N'])
ax = df.plot(kind='bar')
ax.legend(["AAA", "BBB"]);