如何为每个代表相同标签的标记设置标题

Question

我在下图中有一个图例的第一版：

使用以下代码：

# Plot and save : kmax = 0.3
p11, = plt.plot([0], marker='None',
               linestyle='None', label='$k_{max} = 0.3$')
p1, = plt.plot(FoM_vs_Density_array_1[:,0],FoM_vs_Density_array_1[:,1], '-b', label = '$GC_{sp}$')
p2, = plt.plot(FoM_vs_Density_array_1[:,0],FoM_vs_Density_array_1[:,2], '-r', label = '$GC_{ph}$')
p3, = plt.plot(FoM_vs_Density_array_1[:,0],FoM_vs_Density_array_1[:,3], '-y', label = '$WL$')
p4, = plt.plot(FoM_vs_Density_array_1[:,0],FoM_vs_Density_array_1[:,4], '-g', label = '$GC_{ph} + WL + XC$')
p5, = plt.plot(FoM_vs_Density_array_1[:,0],FoM_vs_Density_array_1[:,5], '-m', label = \

                                                               '$GC_{sp} + (GC_{ph} + WL + XC)$')
# Plot and save : kmax = 1.0
p12, = plt.plot([0], marker='None',
               linestyle='None', label='$k_{max} = 1.0$')
p6, = plt.plot(FoM_vs_Density_array_2[:,0],FoM_vs_Density_array_2[:,1], '--b', label = '$GC_{sp}$')
p7, = plt.plot(FoM_vs_Density_array_2[:,0],FoM_vs_Density_array_2[:,2], '--r', label = '$GC_{ph}$')
p8, = plt.plot(FoM_vs_Density_array_2[:,0],FoM_vs_Density_array_2[:,3], '--y', label = '$WL$')
p9, = plt.plot(FoM_vs_Density_array_2[:,0],FoM_vs_Density_array_2[:,4], '--g', label = '$GC_{ph} + WL + XC$')
p10, =plt.plot(FoM_vs_Density_array_2[:,0],FoM_vs_Density_array_2[:,5], '--m', label = \

                                                               '$GC_{sp} + (GC_{ph} + WL + XC)$')
plt.legend(fontsize=14, loc='best', ncol=2, handleheight=1.4, labelspacing=0.05)

如您所见，我为每个标记和列都添加了一个标题（k_max = 0.3 和 k_max = 1.0）。

现在，为了避免这种冗余，我正在尝试合并所有重复的标签，同时保留每个标记的标题，方法是：

from matplotlib.legend_handler import HandlerTuple

# Plot and save : kmax = 0.3
p11, = plt.plot([0], marker='None', linestyle='None')
p1, = plt.plot(FoM_vs_Density_array_1[:,0],FoM_vs_Density_array_1[:,1], '-b')
p2, = plt.plot(FoM_vs_Density_array_1[:,0],FoM_vs_Density_array_1[:,2], '-r')
p3, = plt.plot(FoM_vs_Density_array_1[:,0],FoM_vs_Density_array_1[:,3], '-y')
p4, = plt.plot(FoM_vs_Density_array_1[:,0],FoM_vs_Density_array_1[:,4], '-g')
p5, = plt.plot(FoM_vs_Density_array_1[:,0],FoM_vs_Density_array_1[:,5], '-m')

# Plot and save : kmax = 1.0
p12, = plt.plot([0], marker='None', linestyle='None')
p6, = plt.plot(FoM_vs_Density_array_2[:,0],FoM_vs_Density_array_2[:,1], '--b') 
p7, = plt.plot(FoM_vs_Density_array_2[:,0],FoM_vs_Density_array_2[:,2], '--r') 
p8, = plt.plot(FoM_vs_Density_array_2[:,0],FoM_vs_Density_array_2[:,3], '--y') 
p9, = plt.plot(FoM_vs_Density_array_2[:,0],FoM_vs_Density_array_2[:,4], '--g') 
p10, = plt.plot(FoM_vs_Density_array_2[:,0],FoM_vs_Density_array_2[:,5], '--m') 

l = plt.legend([(p1,p6), (p2,p7), (p3,p8), (p4,p9), (p5,p10)], ['$GC_{sp}$', \
               '$GC_{ph}$', '$WL$', '$GC_{ph} + WL + XC$', '$GC_{sp} + (GC_{ph} + WL + XC)$'], \
               fontsize=14, loc='best', handlelength=2.5, handleheight=1.4, labelspacing=0.05, \
               handler_map={tuple: HandlerTuple(ndivide=None)})

这样，我得到了以下图像：

然后，发生了两个问题:

1) 与上图相比，两个标记之间的空间太小：如何在标记之间插入更大的空间以及为标记本身增加更多的长度（例如，对于破折号标记，如何像我在帖子开头第一张图例中的4条虚线标记一样拥有4条虚线）。

2) 如何将标题"k_max = 0.3"和"k_max = 1.0"放置在每个标记列的上方？这样，我可以在绘图中快速识别我考虑的情况（就像我在上面的第一个图中所做的那样，但是通过重复所有标签的显示来实现了冗余）。

- user1773603

Answer 1

为了解决您的问题，您可以尝试以下方法： 1.1 增加标记之间的间距，您可以在HandlerTuple()中提供额外的参数pad（来自此处）。它将调整不同标记部分之间的间距。代码如下：

l = plt.legend(..., handler_map={tuple: HandlerTuple(ndivide=None, pad=2)})

1.2 如果您想增加标记的宽度，可以像这样增加handlelength参数的值：

l = plt.legend(..., handlelength=6.5, ...)

这些值的结果将如下所示： 为了在标记列上方指定描述，您可以修改p11, = plt.plot([0], marker='None', linestyle='None')和p12, = plt.plot([0], marker='None', linestyle='None')行，如下所示：

# your code:
# p11, = plt.plot([0], marker='None', linestyle='None')
# p12, = plt.plot([0], marker='None', linestyle='None')
p11, = plt.plot([], marker='$0.3$', linestyle='None', color='k', markersize=18, mec='None')
p12, = plt.plot([], marker='$1.0$', linestyle='None', color='k', markersize=18, mec='None')

用空列表[]替换[0]将导致没有任何内容被绘制。 mec='None'将删除标记的边缘颜色。如果没有它，标记看起来像是粗体字。

要在图例中显示这些“线条”，请将以下内容添加到图例中：

l = plt.legend([(p11,p12), ....], ['$k_{max}$', ...], ...)

有了这些调整，图例应该看起来像这样：

完整代码应该看起来像这样：

# Plot and save : kmax = 0.3
p11, = plt.plot([], marker=r'$0.3$', linestyle='None', color='k', markersize=18, mec='None')
p1, = plt.plot(FoM_vs_Density_array_1[:,0],FoM_vs_Density_array_1[:,1], '-b')
p2, = plt.plot(FoM_vs_Density_array_1[:,0],FoM_vs_Density_array_1[:,2], '-r')
p3, = plt.plot(FoM_vs_Density_array_1[:,0],FoM_vs_Density_array_1[:,3], '-y')
p4, = plt.plot(FoM_vs_Density_array_1[:,0],FoM_vs_Density_array_1[:,4], '-g')
p5, = plt.plot(FoM_vs_Density_array_1[:,0],FoM_vs_Density_array_1[:,5], '-m')

# Plot and save : kmax = 1.0
p12, = plt.plot([], marker='$1.0$', linestyle='None', color='k', markersize=18, mec='None')
p6, = plt.plot(FoM_vs_Density_array_2[:,0],FoM_vs_Density_array_2[:,1], '--b')
p7, = plt.plot(FoM_vs_Density_array_2[:,0],FoM_vs_Density_array_2[:,2], '--r')
p8, = plt.plot(FoM_vs_Density_array_2[:,0],FoM_vs_Density_array_2[:,3], '--y')
p9, = plt.plot(FoM_vs_Density_array_2[:,0],FoM_vs_Density_array_2[:,4], '--g')
p10, = plt.plot(FoM_vs_Density_array_2[:,0],FoM_vs_Density_array_2[:,5], '--m')

l = plt.legend([(p11,p12),(p1,p6), (p2,p7), (p3,p8), (p4,p9), (p5,p10)], ['$k_{max}$','$GC_{sp}$', \
               '$GC_{ph}$', '$WL$', '$GC_{ph} + WL + XC$', '$GC_{sp} + (GC_{ph} + WL + XC)$'], \
               fontsize=14, loc='best', handlelength=6.5, handleheight=1.4, labelspacing=0.05, \
               handler_map={tuple: HandlerTuple(ndivide=None, pad=2)})

要像您在第一张图片中展示的那样格式化列名，请使用以下代码：

# change marker and markersize
p11, = plt.plot([], marker=r'$k_{max}=0.3$', linestyle='None', color='k', markersize=45, mec='None')
p12, = plt.plot([], marker='$k_{max}=1.0$', linestyle='None', color='k', markersize=45, mec='None')

# change plt.legend, use 'borderpad' parameter to increase the legend box,
# otherwise part of the column name will be written out of bounds
#                                                                         | change here
l = plt.legend([(p11,p12),(p1,p6), (p2,p7), (p3,p8), (p4,p9), (p5,p10)], ['','$GC_{sp}$', \
               '$GC_{ph}$', '$WL$', '$GC_{ph} + WL + XC$', '$GC_{sp} + (GC_{ph} + WL + XC)$'], \
               fontsize=14, loc='best', handlelength=6.5, handleheight=1.4, labelspacing=0.05, \
               borderpad=0.6, # change here
               handler_map={tuple: HandlerTuple(ndivide=None, pad=2)})

这将导致：

- Phil Leh

---

谢谢！它完美地运行了，我会尽快授予您奖励。 - user1773603