Python: график Pandas DataFrame с назначенной вручную легендой

Question

Я пытаюсь вручную назначить легенду для графика, основанного на Pandas DataFrame. Я думал, что использование ключевого слова label функции pd.plot должно быть способом go. Однако я изо всех сил ...

Вот мой игрушечный пример:

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd

# Create dummy dataframe
df = pd.DataFrame(np.random.randint(0, 100, (20, 2)),
                  index=pd.date_range('20190101', periods=20),
                  columns=list('AB'))

# Create plot
fig, ax1 = plt.subplots(1, 1)
ax1.plot(df, label=['line1', 'line2'])
ax1.legend()

Answer 1

Полагаю, вы имеете в виду построение графиков непосредственно от pd.DataFrame. Функция pd.DataFrame.plot() возвращает объект matplotlib.axes.Axes, который можно использовать для управления легендой.

import numpy as np
import pandas as pd

# Create dummy dataframe
df = pd.DataFrame(np.random.randint(0, 100, (20, 2)),
                  index=pd.date_range('20190101', periods=20),
                  columns=list('AB'))

# Create plot directly from pandas.DataFrame
ax = df.plot()
ax.legend(['line1', 'line2'])

дает