Matplotlib：外部图例，分布在多个子图上

Question

我有一个带有2个子图的图，所有图都共享相应的图，即相同颜色的相同标签。我希望在图的顶部有一个图例，延伸到两个子图上。与以下代码类似：

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-1,1,1000)
y1 = x
y2 = 0.1*x**2
y3 = x**3
y4 = x**4
y5 = x**5

fig, grid = plt.subplots(1,2,sharex="col", sharey="row")
fig.subplots_adjust(wspace=0, hspace=0)
grid[0].plot(x,y1, label='1')
grid[0].plot(x,y2, label='2')
grid[0].plot(x,y3, label='3')
grid[0].plot(x,y4, label='4')
grid[0].plot(x,y5, label='5')
grid[1].plot(x,y1, label='1')

grid[0].legend(bbox_to_anchor=(0., 1.02, 1., .102), loc=3, ncol=5, borderaxespad=0.)
plt.show()

这个例子看起来很不错。唯一的问题是图例从第一个图的开头延伸到第二个图的中间。但是，我希望这个图例可以延伸到两个图上。我怎么能这样做？

注意：原始问题使用2x3网格（2行，3列）。

Answer 1

原则上，How to put the legend out of the plot特别是this post的所有内容都适用。这里一个简单的解决方案是使用图形的子图参数来确定图例的图坐标中的边界框并使用mode="expand"。

bb = (fig.subplotpars.left, fig.subplotpars.top+0.02, 
      fig.subplotpars.right-fig.subplotpars.left,.1)

grid[0].legend(bbox_to_anchor=bb, mode="expand", loc="lower left",
               ncol=5, borderaxespad=0., bbox_transform=fig.transFigure)

Answer 2

您可以尝试使用figlegend之类的内容。

fig, grid = plt.subplots(1,2,sharex="col", sharey="row")
fig.subplots_adjust(wspace=0, hspace=0)
a = grid[0].plot(x,y1)
b = grid[0].plot(x,y2)
c = grid[0].plot(x,y3)
d = grid[0].plot(x,y4)
e = grid[0].plot(x,y5)
f = grid[1].plot(x,y1)

fig.legend((a[0], b[0], c[0], d[0], e[0]), ('label 1', 'label 2', 'label 3', 'label 4', 'label 5'), 'upper center', ncol = 5)
plt.show()

变量a, b, c, d, e, f是具有Line2D类型的单个对象的列表。或者，您可以继续以您完成的方式分配标签，然后像这样调用该属性

fig.legend((a[0], b[0], c[0], d[0], e[0]), (a[0]._label, b[0]._label, c[0]._label, d[0]._label, e[0]._label), 'upper center', ncol = 5)