将不完整/未填充的行中的matplotlib图例条目居中？

Question

假设我正在制作一个包含五个项目的情节，并且只有空间来创建一个包含3列的图例（比这更宽的列更多），例如。

import matplotlib.pyplot as plt
f, a = plt.subplots()
for i in range(5):
    a.plot(np.arange(10),np.random.rand(10),label='Item #%d'%i)
a.legend(ncol=3)

底行中的尾随两个条目是左对齐，在右侧留下一个非常美观的空白空间。当你必须标记非常多的行时，这尤其成问题。

有没有办法居中未填充的行中的条目？

Answer 1

matplotlib图例是基于列的。您不能拥有跨多个列的图例条目。也就是说，当然可以在具有奇数行的图例中“居中”奇数个条目，并且在具有偶数行的图例中类似地“居中”偶数个条目。这可以通过在外部位置使用空艺术家来完成。

import matplotlib.pyplot as plt
import numpy as np

ncols = 3  # or 4
nlines = 7 # or 10

x = np.linspace(0,19)
f = lambda x,p: np.sin(x*p)
h = [plt.plot(x,f(x,i/10.), label="Line {}".format(i))[0] for i in range(nlines)]
h.insert(nlines//ncols, plt.plot([],[],color=(0,0,0,0), label=" ")[0])
plt.legend(handles=h, ncol=ncols, framealpha=1)

plt.show()

如果列数为奇数且图例条目数为偶数，反之亦然，则无法进行上述操作。一个选项可能是使用两个不同的图例并将它们放在另一个下面，使得它看起来像最后一行的条目居中。

import matplotlib.pyplot as plt
import numpy as np

ncols = 3
nlines = 8

x = np.linspace(0,19)
f = lambda x,p: np.sin(x*p)
h = [plt.plot(x,f(x,i/10.), label="Line {}".format(i))[0] for i in range(nlines)]

kw = dict(framealpha=1, borderaxespad=0, bbox_to_anchor=(0.5,0.2), edgecolor="w")
leg1 = plt.legend(handles=h[:nlines//ncols*ncols], ncol=ncols, loc="lower center", **kw)
plt.gca().add_artist(leg1)
leg2 = plt.legend(handles=h[nlines//ncols*ncols:], ncol=nlines-nlines//ncols*ncols, 
                   loc="upper center", **kw)

plt.show()

这里的缺点是图例没有边框，两个图例都需要单独定位。为了克服这一点，我们需要深入挖掘并使用传说中的一些私人属性（注意，这些可能会在不另行通知的情况下从版本更改为版本）。
然后，可以在创建后删除第二个图例并将其_legend_handle_box放入第一个图例中。

import matplotlib.pyplot as plt
import numpy as np

ncols = 3
nlines = 8

x = np.linspace(0,19)
f = lambda x,p: np.sin(x*p)
h = [plt.plot(x,f(x,i/10.), label="Line {}".format(i))[0] for i in range(nlines)]

kw = dict(framealpha=1, bbox_to_anchor=(0.5,0.2))
leg1 = plt.legend(handles=h[:nlines//ncols*ncols], ncol=ncols, loc="lower center", **kw)
plt.gca().add_artist(leg1)
leg2 = plt.legend(handles=h[nlines//ncols*ncols:], ncol=nlines-nlines//ncols*ncols)

leg2.remove()
leg1._legend_box._children.append(leg2._legend_handle_box)
leg1._legend_box.stale = True

plt.show()