Matplotlib中的表格图例

Question

我想在matplotlib中制作复杂的图例。我编写了以下代码

import matplotlib.pylab as plt
import numpy as np

N = 25
y = np.random.randn(N)
x = np.arange(N)

y2 = np.random.randn(25)

# serie A
p1a, = plt.plot(x, y,       "ro", ms=10, mfc="r", mew=2, mec="r")
p1b, = plt.plot(x[:5], y[:5] ,  "w+", ms=10, mec="w", mew=2) 
p1c, = plt.plot(x[5:10], y[5:10], "w*", ms=10, mec="w", mew=2) 

# serie B
p2a, = plt.plot(x, y2,       "bo", ms=10, mfc="b", mew=2, mec="b")
p2b, = plt.plot(x[15:20], y2[15:20] ,  "w+", ms=10, mec="w", mew=2) 
p2c, = plt.plot(x[10:15], y2[10:15], "w*", ms=10, mec="w", mew=2) 


plt.legend([p1a, p2a, (p1a, p1b), (p2a,p2b), (p1a, p1c), (p2a,p2c)], 
 ["No prop", "No prop", "Prop +", "Prop +", "Prop *", "Prop *"], ncol=3, numpoints=1)

plt.show()

它生成类似于这样的图形：

但我希望像这样绘制复杂的图例：

我还尝试使用table函数创建图例，但是无法将修补程序对象放置到单元格的正确位置。

- Serenity

---

我不确定，但我相信在这个问题的被接受答案中有一个完全做到这一点的例子。或者至少可以指引你正确的方向？ - Ajean

不，这个例子中每个标记都有自己的标签。 - Serenity

没错，你可以在那里放置空字符串。实际上，我正在寻找一个以前在这里见过的不同的例子（有人写了一个令人赞叹的传说），但是我找不到它。只是一种想法，因为我认为那个例子使用了空字符串。很抱歉我找不到它... - Ajean

是的，我也试过了。结果非常丑陋：http://i.imgur.com/JHbNt3z.png - Serenity

没有一种顶级可配置的方法来实现这个。虽然有工具可以做到这一点（处理程序创建工具、布局框hpackers/vpackers等），但作为替代图例实现，这将是一件有趣的事情（并且最好至少在我们的文档中出现）。 - tacaswell

我创建了一个 tablelegend 函数 在这个答案中，它可能对此有用。 - Zaus

Answer 1

这个解决方案是否足够符合您的要求？它受到了Ricardo答案的启发，但我只为每列使用了一个图例对象，并利用了title关键字来设置每列的标题。为了使标记位于每列的中心，我使用了负值的handletextpad将其向后推。没有对单独行进行图例设置。我还必须在标题字符串中插入一些空格，以使它们在屏幕上呈现时大小相等。

当我保存图形时，我也注意到需要对图例框的确切位置进行进一步调整，但由于我猜测您可能希望在代码中进行更多的调整，所以我将其留给您。您可能还需要自己调整handletextpad以使它们“完美”对齐。

import matplotlib.pylab as plt
import numpy as np
plt.close('all')

N = 25
y = np.random.randn(N)
x = np.arange(N)

y2 = np.random.randn(25)

# serie A
p1a, = plt.plot(x, y,       "ro", ms=10, mfc="r", mew=2, mec="r")
p1b, = plt.plot(x[:5], y[:5] ,  "w+", ms=10, mec="w", mew=2) 
p1c, = plt.plot(x[5:10], y[5:10], "w*", ms=10, mec="w", mew=2) 

# serie B
p2a, = plt.plot(x, y2,       "bo", ms=10, mfc="b", mew=2, mec="b")
p2b, = plt.plot(x[15:20], y2[15:20] ,  "w+", ms=10, mec="w", mew=2) 
p2c, = plt.plot(x[10:15], y2[10:15], "w*", ms=10, mec="w", mew=2) 

line_columns = [
                p1a, p2a,
                (p1a, p1b), (p2a, p2b),
                (p1a, p1c), (p2a, p2c)
                ]


leg1 = plt.legend(line_columns[0:2], ['', ''], ncol=1, numpoints=1, 
                  title='No prop', handletextpad=-0.4, 
                  bbox_to_anchor=[0.738, 1.])
leg2 = plt.legend(line_columns[2:4], ['', ''], ncol=1, numpoints=1, 
                  title=' Prop  + ', handletextpad=-0.4,
                  bbox_to_anchor=[0.87, 1.])
leg3 = plt.legend(line_columns[4:6], ['', ''], ncol=1, numpoints=1, 
                  title=' Prop  * ', handletextpad=-0.4, 
                  bbox_to_anchor=[0.99, 1.])

plt.gca().add_artist(leg1)
plt.gca().add_artist(leg2)
plt.gca().add_artist(leg3)

plt.gcf().show()

编辑

也许这个会更好用一些。你仍然需要微调一些东西，但bboxes的对齐问题已经解决了。

leg = plt.legend(line_columns, ['']*len(line_columns), 
             title='No Prop    Prop +    Prop *',  
             ncol=3, numpoints=1, handletextpad=-0.5)

- pathoren

---

很难正确放置这个图例，因为锚点不会自动计算。 - Serenity

是的，我也注意到了...请查看我的编辑答案。我用一个标签替换了所有标签。现在唯一需要调整的是title和handletextpad值的内部间距。这样好吗？ - pathoren

Answer 2

似乎没有标准方法，只有一些技巧可用。

值得注意的是，您应该检查最适合您的大小bbox因子。

迄今为止我能找到的最好的方法，也许可以带给你更好的解决方案：

N = 25
y = np.random.randn(N)
x = np.arange(N)

y2 = np.random.randn(25)

# Get current size
fig_size = list(plt.rcParams["figure.figsize"])

# Set figure width to 12 and height to 9
fig_size[0] = 12
fig_size[1] = 12
plt.rcParams["figure.figsize"] = fig_size

# serie A
p1a, = plt.plot(x, y,       "ro", ms=10, mfc="r", mew=2, mec="r")
p1b, = plt.plot(x[:5], y[:5] ,  "w+", ms=10, mec="w", mew=2) 
p1c, = plt.plot(x[5:10], y[5:10], "w*", ms=10, mec="w", mew=2) 

# serie B
p2a, = plt.plot(x, y2,       "bo", ms=10, mfc="b", mew=2, mec="b")
p2b, = plt.plot(x[15:20], y2[15:20] ,  "w+", ms=10, mec="w", mew=2) 
p2c, = plt.plot(x[10:15], y2[10:15], "w*", ms=10, mec="w", mew=2) 

v_factor = 1.
h_factor = 1.

leg1 = plt.legend([(p1a, p1a)], ["No prop"], bbox_to_anchor=[0.78*h_factor, 1.*v_factor])
leg2 = plt.legend([(p2a, p2a)], ["No prop"], bbox_to_anchor=[0.78*h_factor, .966*v_factor])

leg3 = plt.legend([(p2a,p2b)], ["Prop +"], bbox_to_anchor=[0.9*h_factor, 1*v_factor])
leg4 = plt.legend([(p1a, p1b)], ["Prop +"], bbox_to_anchor=[0.9*h_factor, .966*v_factor])

leg5 = plt.legend([(p1a, p1c)], ["Prop *"], bbox_to_anchor=[1.*h_factor, 1.*v_factor])
leg6 = plt.legend([(p2a,p2c)], ["Prop *"], bbox_to_anchor=[1.*h_factor, .966*v_factor])

plt.gca().add_artist(leg1)
plt.gca().add_artist(leg2)
plt.gca().add_artist(leg3)
plt.gca().add_artist(leg4)
plt.gca().add_artist(leg5)
plt.gca().add_artist(leg6)
plt.show()

- Ricardo Silveira

Answer 3

我改进了@pathoren的答案，使图例自动在一个循环中定位，根据图例的bbox坐标。这段代码允许显示复杂图例的所有所需边框线：

import matplotlib.pylab as plt
import numpy as np
plt.close('all')

# test data
N = 25
y = np.random.randn(N)
x = np.arange(N)
y2 = np.random.randn(25)

# serie A
p1a, = plt.plot(x, y, "ro", ms=10, mfc="r", mew=2, mec="r")
p1b, = plt.plot(x[:5], y[:5], "w+", ms=10, mec="w", mew=2) 
p1c, = plt.plot(x[5:10], y[5:10], "w*", ms=10, mec="w", mew=2) 
# serie B
p2a, = plt.plot(x, y2, "bo", ms=10, mfc="b", mew=2, mec="b")
p2b, = plt.plot(x[15:20], y2[15:20], "w+", ms=10, mec="w", mew=2) 
p2c, = plt.plot(x[10:15], y2[10:15], "w*", ms=10, mec="w", mew=2) 

# legend handlers
columns = [p1a, p2a, 
 (p1a, p1b), (p2a, p2b),
 (p1a, p1c), (p2a, p2c)]

ax = plt.gca()
fig = plt.gcf()
legs = []
# set the first legend in desired position
leg = plt.legend(columns[0:2], ['', ''], ncol=1, numpoints=1, 
 borderaxespad=0., title='No prop.', framealpha=.75,
 facecolor='w', edgecolor='k', loc=2, fancybox=None)
ax.add_artist(leg)
fig.canvas.draw()
plt.pause(1.e-3)

# get bbox postion of 1st legend to calculate
# postion of 2nd and 3rd legends according to loc
for i,si in enumerate(['+','*']):
    bbox = leg.get_window_extent().inverse_transformed(ax.transAxes)
    # next legends
    leg = plt.legend(columns[(i+1)*2:(i+1)*2+2], ['', ''], ncol=1, numpoints=1, 
     title='Prop. '+si, framealpha=.75, borderaxespad=.0,
     bbox_to_anchor=(bbox.x1-bbox.height*.08, bbox.y0, bbox.width, bbox.height),
     facecolor='w', edgecolor='k')
    ax.add_artist(leg)
    fig.canvas.draw()
    plt.pause(1.e-3)

plt.show()

- Serenity