matplotlib 更新“axvspan”的位置

aiqt4smr  于 2023-03-13  发布在  其他
关注(0)|答案(1)|浏览(185)

我想“模拟”两个对象在图中的位置。我想用axvspan来做这个。我想用滑块移动一个axvspan。我想做这样的事情:

import matplotlib.pyplot as plt
from matplotlib.widgets import Slider

fig, ax = plt.subplots()
plt.subplots_adjust(bottom=0.25)

t = 2
g0 = 10

filter_loc, = plt.axvspan((50-g0-t), (50-g0), color = "blue")
sensor_loc = plt.axvspan(50,80, color="red")
plt.axis([0, 1, -10, 10])

axgap = plt.axes([0.25, 0.1, 0.65, 0.03])
sgap = Slider(axgap, 'Gap', 0.01, 30.0, valinit=g0)

def update(val):
    gap = sgap.val
    filter_loc.set_xdata(50-gap-t,50-gap)    
    fig.canvas.draw_idle()

sgap.on_changed(update)
plt.show()

当尝试它在几个方面,我总是得到错误:

filter_loc, = plt.axvspan((50-g0-t), (50-g0))
TypeError: 'Polygon' object is not iterable

所以在我看来axvspan并不喜欢使用滑块来更新值。有什么方法我仍然可以通过使用axvspan来实现这一点吗?

0x6upsns

0x6upsns1#

filter_loc后面有一个额外的逗号,除此之外,filter_loc没有set_xdata属性,你可以通过调用set_xy()来修改它,例如下面这样做:

def set_xvalues(polygon, x0, x1):
    _ndarray = polygon.get_xy()
    _ndarray[:, 0] = [x0, x0, x1, x1, x0]
    polygon.set_xy(_ndarray)

def update(val):
    gap = sgap.val
    set_xvalues(filter_loc, 50-gap-t, 50-gap)
    fig.canvas.draw_idle()

顺便说一句,当前plt.axis()值阻止显示蓝色方块。

相关问题