python - 在所有子图的顶部显示图例

Question

我有 1x3 的 matplotlib 子图。我试图在所有子图的顶部显示子图的图例。但是，我只显示最后一个。

df = pd.DataFrame({"a":[1,2,3],"b":[4,5,6],"c":[7,8,9]})
fig, axes = plt.subplots(nrows=1, ncols=3)
df.plot(ax=axes[0],legend=False)
plt.legend(bbox_to_anchor=(0., 1.02, 1., .102), loc=3,
           ncol=2, mode="expand", borderaxespad=0.)
df.plot(ax=axes[1],legend=False)
plt.legend(bbox_to_anchor=(0., 1.02, 1., .102), loc=3,
           ncol=2, mode="expand", borderaxespad=0.)
df.plot(ax=axes[2],legend=False)
plt.legend(bbox_to_anchor=(0., 1.02, 1., .102), loc=3,
           ncol=2, mode="expand", borderaxespad=0.)

如何在所有子图的顶部显示图例？

Answer 1

使用 ax.legend(...) 而不是 plt.legend。

一般来说，最好避免混合使用 pyplot 和 axes 方法。事实上，我建议或多或少只使用 plt.subplots() 和 plt.show() 并在其他地方使用轴/图形方法。它使在哪些轴上绘制以及对图形与轴的操作变得更加清晰。

举个例子:

import pandas as pd
import matplotlib.pyplot as plt

df = pd.DataFrame({"a":[1,2,3],"b":[4,5,6],"c":[7,8,9]})

fig, axes = plt.subplots(nrows=1, ncols=3, figsize=(10, 4))
for ax in axes:
    df.plot(ax=ax,legend=False)
    ax.legend(bbox_to_anchor=(0., 1.02, 1., .102), loc=3,
              ncol=2, mode="expand", borderaxespad=0.)

# Make some room at the top for the legend...
fig.subplots_adjust(top=0.8)

plt.show()