第一个子图上的 Matplotlib 图例 [重复]

Question

我只想用不同数字的子图（根据结果从一到五个）在我的图中绘制一个图例。图例应始终绘制在第一个 plot1 上方。有没有办法明确告诉 matplotlib 使用第一个图来绘制图例，而不是最后一个2？

我尝试手动计算位置，以防有多个绘图，但它不是很精确，也不是一个好的解决方案。

编辑：解释一下。我知道如何为许多子图只绘制一个图例（如图 2 所示），但我想知道如何将图例修复到绘制的第一个图而不是最后一个图。我试图根据情节编号计算位置，但它并不精确，有时看起来很丑（传说进入情节）。我想知道我是否可以告诉 matplotlib 将图例始终修复到 subplot1。

import matplotlib.pyplot as plt
import matplotlib.lines as mlines
plt.figure(1, figsize=(12, 10))
nrows = 5
ncols = 1
plot_number = 1
for i in range(3):
    plot = plt.subplot(nrows, ncols, plot_number)
    plot_number += 1

p3 = mlines.Line2D([], [], color='r')
p4 = mlines.Line2D([], [], color='b')
plt.legend([p3, p4], ['one', 'two'],loc=4, ncol=4,     mode="", borderaxespad=.5, frameon=True,
        bbox_to_anchor=(0., 1.02, 1., .102))
plt.show()

Answer 1

如果你在这里使用面向对象的接口，一切都会变得更简单。然后，您可以在特定子图的轴上调用 legend。我们可以使用plt.subplots 稍微简化您的代码：

import matplotlib.pyplot as plt
import matplotlib.lines as mlines

nrows = 5
ncols = 1
fig, axes = plt.subplots(nrows,ncols,figsize=(12,10))

p3 = mlines.Line2D([], [], color='r')
p4 = mlines.Line2D([], [], color='b')

# axes[0] is the first subplot
axes[0].legend([p3, p4], ['one', 'two'],loc=4, ncol=4,     mode="", borderaxespad=.5, frameon=True,
        bbox_to_anchor=(0., 1.02, 1., .102))
plt.show()