python - 在一张图表上绘制两个图后，没有发现带有标签的句柄会导致图例错误

Question

我正在逐月绘制一些功能并突出显示其中的一些功能。在我添加高亮之前，图例可以自动显示，但现在它返回 No handles with labels found to put in legend错误。
示例数据

df = pd.DataFrame(np.random.randn(10, 4), columns=list('ABCD'))
highlight = ['A', 'B']

fig, ax = plt.subplots(figsize=(15, 5))
plt.plot(df.loc[:, ~df.columns.isin(highlight)], c='gray', alpha=0.5) 
plt.plot(df.loc[:, df.columns.isin(highlight)]) 
months = pd.date_range('2019-04-01','2019-08-01', freq='MS').strftime("%Y-%m").tolist()
plt.legend(loc='center left', bbox_to_anchor=(1.0, 0.5))
z = ax.set_xticklabels(months, rotation=45)

我想在一张图表上绘制两个图会导致此问题，但不知道如何解决。我不想手动指定图例。

Answer 1

使用 pandas.DataFrame.plot 而不是 plt.plot

这可确保将列标题正确分配为正确行

的标签。

它返回一个 matplotlib.axes.Axes 或 numpy.ndarray其中。

此示例使用 pandas v1.2.3

import pandas as pd
import matplotlib.pyplot as plt

ax = df.loc[:, df.columns.isin(highlight)].plot(figsize=(15, 5))
df.loc[:, ~df.columns.isin(highlight)].plot(c='gray', alpha=0.5, ax=ax)

plt.legend(title='column labels', loc='center left', bbox_to_anchor=(1.0, 0.5))
plt.show()