为DataFrame的两个子图制作单个图例

Question

我创建了一个在不同子图上具有两个轴的图。当前，一个覆盖另一个。问题在于使图例在堆栈中包含两个标签。我该怎么办？

d = data.groupby('atemp_rounded').sum().reset_index()
fig = plt.figure()
ax1 = fig.add_subplot(111) # don't know what 111 stands for...
ax2 = ax1.twinx()
d.plot(ax=ax1, y='casual')
d.plot(ax=ax2, y='registered', color='g')
plt.show()

Answer 1

您可以关闭各个图的图例，而创建一个图形图例。要将其放置在轴边界内，需要在轴坐标中指定位置。

import matplotlib.pyplot as plt
import pandas as pd

df = pd.DataFrame({"A" : [3,2,1], "B" : [2,2,1]})

fig = plt.figure()
ax1 = fig.add_subplot(111) # don't know what 111 stands for...
ax2 = ax1.twinx()
df.plot(ax=ax1, y='A', legend=False)
df.plot(ax=ax2, y='B', color='g', legend=False)
fig.legend(loc="upper right", bbox_to_anchor=(0,0,1,1), bbox_transform=ax1.transAxes)
plt.show()