创建matplotlib图例后如何修改它？

Question

我可以访问figure实例fig=pylab。gcf（）。我知道在这个图中有一个图例，我可以通过myLegend=fig.gca（）访问它。图例。现在我想更改图例的属性。其中一些我可以通过设置器访问，比如myLegend。将帧设置为on（真）。

创建图例时，它接受多个关键字参数：

类matplotlib。传奇传奇（父项、句柄、标签、loc=None、numpoints=None、markerscale=None、scatterpoints=None、scatteryoffsets=None、prop=None、fontsize=None、borderpad=None、labelspace=None、handlelight=None、handleextpad=None、borderaxespad=None、columnspacing=None、ncol=1、mode=None、fancybox=None、shadow=None、title=None、framealpha=None、bb）ox\u to\u anchor=None、bbox\u transform=None、frameon=None、handler\u map=None）

创建图例后，如何修改图例中的所有关键字参数？

其中一个问题是numpoints（图例中的标记数，默认为2）。下面是我想如何更改它的示例：

这显示了我想如何编程它

import pylab
pylab.plot(0,0,'ro', label = 'one point')
pylab.legend(loc = "lower left")
# no modifications above this line
setattr(pylab.gcf().gca().legend_, 'numpoints',1)
pylab.show()

这显示了我希望它看起来是什么样子

import pylab
pylab.plot(0,0,'ro', label = 'one point')
pylab.legend(numpoints = 1, loc = "lower left")
pylab.show()

我已经检查了源代码，有一个numpoint变量已更改，但大写字母未更新到屏幕。我错过了什么？

Answer 1

您在图例中看到的实际上是一个Line2D。创建该行后更改numpoints不会更新所述行，因此您必须获得Line2D对象的句柄，并手动删除其中一个点：

import pylab
pylab.plot(0,0,'ro', label = 'one point')
legend = pylab.legend(loc = "lower left")
markers = legend.get_children()[0].get_children()[1].get_children()[0].get_children()[0].get_children()[0].get_children()[1]
markers.set_data(map(pylab.mean, markers.get_data()))
pylab.show()

需要get_children（）。上面的代码片段应该足以让您大致了解情况，但在实际应用程序中，获取句柄的更好方法是遵循图例指南中有关图例处理程序的提示，并使用定制的HandlerLine2D，以某种方式存储该行。

Answer 2

您可以使用命令pylab。图例再次使用正确的关键字/参数。这将修改现有图例，而不是创建新图例。下面你会发现你的例子，稍微修改了一下。

import pylab
pylab.plot(0,0,'ro', label = 'one point')
pylab.legend(loc = "lower left")
# Change the number of markers shown in the legend
pylab.legend(numpoints = 1, loc = "lower left")

pylab.show()

希望能有帮助。

Answer 3

我写了一个函数modify_legend，它在创建图例后修改图例。它基本上从已经创建的图例中读取所有参数，用您提供的键值参数更新它，并再次用所有可能的参数调用传奇（...）。

然后，您的问题将通过以下方式解决：

import pylab
pylab.plot(0,0,'ro', label = 'one point')
pylab.legend(loc = "lower left")

modify_legend(numpoints = 1)

pylab.show()

以下是修改图例的代码：

def modify_legend(**kwargs):
    import matplotlib as mpl

    l = mpl.pyplot.gca().legend_

    defaults = dict(
        loc = l._loc,
        numpoints = l.numpoints,
        markerscale = l.markerscale,
        scatterpoints = l.scatterpoints,
        scatteryoffsets = l._scatteryoffsets,
        prop = l.prop,
        # fontsize = None,
        borderpad = l.borderpad,
        labelspacing = l.labelspacing,
        handlelength = l.handlelength,
        handleheight = l.handleheight,
        handletextpad = l.handletextpad,
        borderaxespad = l.borderaxespad,
        columnspacing = l.columnspacing,
        ncol = l._ncol,
        mode = l._mode,
        fancybox = type(l.legendPatch.get_boxstyle())==mpl.patches.BoxStyle.Round,
        shadow = l.shadow,
        title = l.get_title().get_text() if l._legend_title_box.get_visible() else None,
        framealpha = l.get_frame().get_alpha(),
        bbox_to_anchor = l.get_bbox_to_anchor()._bbox,
        bbox_transform = l.get_bbox_to_anchor()._transform,
        frameon = l._drawFrame,
        handler_map = l._custom_handler_map,
    )

    if "fontsize" in kwargs and "prop" not in kwargs:
        defaults["prop"].set_size(kwargs["fontsize"])

    mpl.pyplot.legend(**dict(defaults.items() + kwargs.items()))

代码注释：

• 一些参数可以很容易地从图例对象中读取，其他参数（如标题，fancybox）需要一些“艺术性”。您可以检查matplotlib。传奇传说__初始化以查看它是如何完成的以及为什么完成的
• 当最初使用prop创建图例时，fontsize参数上的附加条件用于覆盖字体大小，因为prop通常覆盖fontsize
• 我没有测试所有的案例，因为我没有太多的时间（特别是bbox\u-to\u锚定和bbox\u转换-参数），所以请尝试并改进代码：）