3

我正在尝试为类似于此示例的 matplotlib 错误栏图编写图例选择器。我希望能够单击图例中的错误栏/数据点来切换它们在轴中的可见性。问题是返回的图例对象plt.legend()不包含创建图例时使用的艺术家的任何数据。如果我例如。做：

import numpy as np
import matplotlib.pyplot as plt

fig, ax = plt.subplots()
x = np.linspace(0,10,100)
y = np.sin(x) + np.random.rand(100)
yerr = np.random.rand(100)

erbpl1 = ax.errorbar(x, y, yerr=yerr, fmt='o', label='A')
erbpl2 = ax.errorbar(x, 0.02*y, yerr=yerr, fmt='o', label='B')

leg = ax.legend()

从这里似乎不可能通过使用该leg对象来访问传说中的艺术家。通常，可以使用更简单的图例来做到这一点，例如：

plt.plot(x, y, label='whatever')
leg = plt.legend()
proxy_lines = leg.get_lines()

为您提供图例中使用的 Line2D 对象。但是，使用误差线图会leg.get_lines()返回一个空列表。这种方法很有意义，因为plt.errorbar返回一个matplotlib.container.ErrorbarContainer对象（其中包含数据点、误差线端盖、误差线线）。我希望图例有一个类似的数据容器，但我看不到这一点。我能管理的最接近的是leg.legendHandles指向误差线，而不是数据点或端盖。如果您可以选择图例，则可以使用 dict 将它们映射到原始图，并使用以下函数打开/关闭错误栏。

def toggle_errorbars(erb_pl):
    points, caps, bars = erb_pl
    vis = bars[0].get_visible()
    for line in caps:
        line.set_visible(not vis)
    for bar in bars:
        bar.set_visible(not vis)
    return vis

def onpick(event):
    # on the pick event, find the orig line corresponding to the
    # legend proxy line, and toggle the visibility
    legline = event.artist
    origline = lined[legline]

    vis = toggle_errorbars(origline)
    ## Change the alpha on the line in the legend so we can see what lines
    ## have been toggled
    if vis:
        legline.set_alpha(.2)
    else:
        legline.set_alpha(1.)
    fig.canvas.draw()

我的问题是，是否有一种解决方法可以让我在错误栏/其他复杂图例上进行事件选择？

4

1 回答 1

4

这使得标记可以选择：

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.legend_handler

from matplotlib.container import ErrorbarContainer

class re_order_errorbarHandler(matplotlib.legend_handler.HandlerErrorbar):
    """
    Sub-class the standard error-bar handler 
    """
    def create_artists(self, *args, **kwargs):
        #  call the parent class function
        a_list = matplotlib.legend_handler.HandlerErrorbar.create_artists(self, *args, **kwargs)
        # re-order the artist list, only the first artist is added to the
        # legend artist list, this is the one that corresponds to the markers
        a_list = a_list[-1:] + a_list[:-1]
        return a_list

my_handler_map = {ErrorbarContainer: re_order_errorbarHandler(numpoints=2)}

fig, ax = plt.subplots()
x = np.linspace(0,10,100)
y = np.sin(x) + np.random.rand(100)
yerr = np.random.rand(100)

erbpl1 = ax.errorbar(x, y, yerr=yerr, fmt='o', label='A')
erbpl2 = ax.errorbar(x, 0.02*y, yerr=yerr, fmt='o', label='B')

leg = ax.legend(handler_map=my_handler_map)

lines = [erbpl1, erbpl2]
lined = dict()
# not strictly sure about ordering, but 
for legline, origline in zip(leg.legendHandles, lines):
    legline.set_picker(5)  # 5 pts tolerance
    lined[legline] = origline


def onpick(event):
    # on the pick event, find the orig line corresponding to the
    # legend proxy line, and toggle the visibility
    legline = event.artist
    origline = lined[legline]
    for a in origline.get_children():
        vis = not a.get_visible()
        a.set_visible(vis)
    # Change the alpha on the line in the legend so we can see what lines
    # have been toggled
    if vis:
        legline.set_alpha(1.0)
    else:
        legline.set_alpha(0.2)
    fig.canvas.draw()

fig.canvas.mpl_connect('pick_event', onpick)

这里发生的是标准处理程序ErrorbarContainers使用 4 个艺术家来生成图例条目（LineCollection用于条形图、LineCollection大写字母、Line2D连接线和Line2D标记）。生成艺术家的代码仅返回添加到图例的艺术家列表中的第一批艺术家（请参阅 参考资料matplotlib.legend_handler.HandlerBase.__call__）。误差线列表中的第一个艺术家恰好是作为垂直线的线条集合，最终以leg.legendHandles. 挑选不起作用的原因似乎是他们被其他艺术家隐藏了（我认为）。

解决方案是创建一个本地子类，HandlerErrorbar该子类重新排序艺术家列表，以便保存的艺术家成为标记leg.legendHandles的Line2D对象。

我可能会打开一个 PR 以使其成为默认行为。

于 2013-10-19T21:56:18.017 回答