Question

Matplotlib的line2D个对象，例如通过调用plot返回的对象，有一个方便的方法set_data，让我快速更新由一行绘制的值，而不会影响其余的情节或线条的格式。

#sample plot
from matplotlib.pyplot import plot
p = plot(arange(10),arange(10))[0]

#now update the data
p.set_data(arange(10),arange(10)+2)
draw()

是否有相对简单（几行代码）的方式对errorbar绘图做同样的事情？我希望能够使用各种文本，箭头，线条等设置复杂的绘图，然后通过几组不同的数据快速循环绘图的错误栏部分。

errorbar返回的对象似乎相当复杂，到目前为止，我删除和重绘的尝试都失败了。

Answer 1

感谢the help from @tacaswell。下面是python函数，可以更新xerr和yerr，以及基线绘图的x_data和y_data：

def adjustErrbarxy(self, errobj, x, y, x_error, y_error):
    ln, (errx_top, errx_bot, erry_top, erry_bot), (barsx, barsy) = errobj
    x_base = x
    y_base = y

    xerr_top = x_base + x_error
    xerr_bot = x_base - x_error
    yerr_top = y_base + y_error
    yerr_bot = y_base - y_error

    errx_top.set_xdata(xerr_top)
    errx_bot.set_xdata(xerr_bot)
    errx_top.set_ydata(y_base)
    errx_bot.set_ydata(y_base)

    erry_top.set_xdata(x_base)
    erry_bot.set_xdata(x_base)
    erry_top.set_ydata(yerr_top)
    erry_bot.set_ydata(yerr_bot)

    new_segments_x = [np.array([[xt, y], [xb,y]]) for xt, xb, y in zip(xerr_top, xerr_bot, y_base)]
    new_segments_y = [np.array([[x, yt], [x,yb]]) for x, yt, yb in zip(x_base, yerr_top, yerr_bot)]
    barsx.set_segments(new_segments_x)
    barsy.set_segments(new_segments_y)

第一个输入参数（self用于python类）是已创建的errorbar绘图处理程序，也是需要更新其属性的对象; x和y是更新的numpy数组，应显示x和y轴的平均值;最后两个参数x_error和y_error是为errorbar和x数组计算的y范围。如果只需要更新错误栏，则x_base和y_base应分别写为ln.get_xdata()和ln.get_ydata()。

到目前为止，errorbar中matplotlib更新的解决方案确实非常重要，希望在未来的版本中更容易。

Answer 2

我遵循了 Tong 提供的链接，并了解到他的代码中存在问题-不需要输入 self 变量。 mitpre 还提供了一种更通用的解决方案-我使用了它，效果很好。下面是用于快速参考的代码：

def update_errorbar(errobj, x, y, xerr=None, yerr=None):
    ln, caps, bars = errobj


    if len(bars) == 2:
        assert xerr is not None and yerr is not None, "Your errorbar object has 2 dimension of error bars defined. You must provide xerr and yerr."
        barsx, barsy = bars  # bars always exist (?)
        try:  # caps are optional
            errx_top, errx_bot, erry_top, erry_bot = caps
        except ValueError:  # in case there is no caps
            pass

    elif len(bars) == 1:
        assert (xerr is     None and yerr is not None) or\
               (xerr is not None and yerr is     None),  \
               "Your errorbar object has 1 dimension of error bars defined. You must provide xerr or yerr."

        if xerr is not None:
            barsx, = bars  # bars always exist (?)
            try:
                errx_top, errx_bot = caps
            except ValueError:  # in case there is no caps
                pass
        else:
            barsy, = bars  # bars always exist (?)
            try:
                erry_top, erry_bot = caps
            except ValueError:  # in case there is no caps
                pass

    ln.set_data(x,y)

    try:
        errx_top.set_xdata(x + xerr)
        errx_bot.set_xdata(x - xerr)
        errx_top.set_ydata(y)
        errx_bot.set_ydata(y)
    except NameError:
        pass
    try:
        barsx.set_segments([np.array([[xt, y], [xb, y]]) for xt, xb, y in zip(x + xerr, x - xerr, y)])
    except NameError:
        pass

    try:
        erry_top.set_xdata(x)
        erry_bot.set_xdata(x)
        erry_top.set_ydata(y + yerr)
        erry_bot.set_ydata(y - yerr)
    except NameError:
        pass
    try:
        barsy.set_segments([np.array([[x, yt], [x, yb]]) for x, yt, yb in zip(x, y + yerr, y - yerr)])
    except NameError:
        pass

Matplotlib set_data用于错误栏图

2 个答案: