Matplotlib设置填充形状之间的动画

cld4siwp  于 2022-11-24  发布在  其他
关注(0)|答案(6)|浏览(155)

我试图在matplotlib中制作一个fill_between形状的动画,但我不知道如何更新PolyCollection的数据。举个简单的例子:我有两条线,我总是在它们之间填充。当然,线是变化的,是动画的。
下面是一个虚拟示例:

import matplotlib.pyplot as plt

# Init plot:
f_dummy = plt.figure(num=None, figsize=(6, 6));
axes_dummy = f_dummy.add_subplot(111);

# Plotting:
line1, = axes_dummy.plot(X, line1_data, color = 'k', linestyle = '--', linewidth=2.0, animated=True);
line2, = axes_dummy.plot(X, line2_data, color = 'Grey', linestyle = '--', linewidth=2.0, animated=True);
fill_lines = axes_dummy.fill_between(X, line1_data, line2_data, color = '0.2', alpha = 0.5, animated=True);

f_dummy.show();
f_dummy.canvas.draw();
dummy_background = f_dummy.canvas.copy_from_bbox(axes_dummy.bbox);

# [...]    

# Update plot data:
def update_data():
   line1_data = # Do something with data
   line2_data = # Do something with data
   f_dummy.canvas.restore_region( dummy_background );
   line1.set_ydata(line1_data);
   line2.set_ydata(line2_data);
   
   # Update fill data too

   axes_dummy.draw_artist(line1);
   axes_dummy.draw_artist(line2);

   # Draw fill too
   
   f_dummy.canvas.blit( axes_dummy.bbox );

问题是如何在每次调用update_data()时基于line1_dataline2_data更新fill_betweenPoly数据,并在blit之前绘制它们(“# Update fill data too”和“# Draw fill too”)。我尝试了fill_lines.set_verts(),但没有成功,也找不到示例。

s4chpxco

s4chpxco1#

好的,正如有人指出的,我们在这里处理的是一个集合,所以我们必须删除和重绘。所以在update_data函数的某个地方,删除所有与之关联的集合:

axes_dummy.collections.clear()

并绘制新的“fill_between”多边形集合:

axes_dummy.fill_between(x, y-sigma, y+sigma, facecolor='yellow', alpha=0.5)

需要类似的技巧将未填充的等值线图覆盖在填充的等值线图上,因为未填充的等值线图也是一个集合(我想是线条的集合?)。

doinxwow

doinxwow2#

这不是我答案,但我发现它最有用:
http://matplotlib.1069221.n5.nabble.com/animation-of-a-fill-between-region-td42814.html
你好,Mauricio,Patch对象比线对象更难处理,因为与线对象不同的是,它是从用户提供的输入数据中删除的一个步骤。下面有一个与你想做的类似的示例:http://matplotlib.org/examples/animation/histogram.html
基本上,你需要在每一帧修改路径的顶点,看起来像这样:

from matplotlib import animation
import numpy as np
import matplotlib.pyplot as plt

fig, ax = plt.subplots()
ax.set_xlim([0,10000])

x = np.linspace(6000.,7000., 5)
y = np.ones_like(x)

collection = plt.fill_between(x, y)

def animate(i):
    path = collection.get_paths()[0]
    path.vertices[:, 1] *= 0.9

animation.FuncAnimation(fig, animate,
                        frames=25, interval=30)

看一下路径。顶点,看看它们是如何排列的。希望这对你有帮助,杰克

hgtggwj0

hgtggwj03#

如果你不想使用anitmation,或者要从你的图中删除所有的东西来更新只填充,你可以使用这样的方法:
调用fill_lines.remove(),然后再次调用axes_dummy.fill_between()来绘制新的。

rsaldnfx

rsaldnfx4#

初始化pyplot交互模式

import matplotlib.pyplot as plt

plt.ion()

在绘制填充时使用可选的label参数:

plt.fill_between(
    x, 
    y1, 
    y2, 
    color="yellow", 
    label="cone"
)

plt.pause(0.001) # refresh the animation

在我们的脚本后面,我们可以通过标签来删除特定的填充或填充列表,这样就可以逐个对象地创建动画。

axis = plt.gca()

fills = ["cone", "sideways", "market"]   

for collection in axis.collections:
    if str(collection.get_label()) in fills:
        collection.remove()
        del collection

plt.pause(0.001)

您可以对要删除的对象组使用相同的标签;或以其它方式根据需要用标签对标签进行编码
例如,如果我们的填充标记为:
“圆锥1”“圆锥2”“横向1”

if "cone" in str(collection.get_label()):

我会删除前缀为“cone”的两个词。
也可以采用相同的方式设置线的动画效果

for line in axis.lines:
bejyjqdl

bejyjqdl5#

另一个有效的习惯用法是保存一个绘制对象的列表;此方法似乎适用于任何类型的绘制对象。

# plot interactive mode on
plt.ion()

# create a dict to store "fills" 
# perhaps some other subclass of plots 
# "yellow lines" etc. 
plots = {"fills":[]}

# begin the animation
while 1: 

    # cycle through previously plotted objects
    # attempt to kill them; else remember they exist
    fills = []
    for fill in plots["fills"]:
        try:
            # remove and destroy reference
            fill.remove()
            del fill
        except:
            # and if not try again next time
            fills.append(fill)
            pass
    plots["fills"] = fills   

    # transformation of data for next frame   
    x, y1, y2 = your_function(x, y1, y2)

    # fill between plot is appended to stored fills list
    plots["fills"].append(
        plt.fill_between(
            x,
            y1,
            y2,
            color="red",
        )
    )

    # frame rate
    plt.pause(1)
h43kikqp

h43kikqp6#

与这里的大多数答案相反,每次更新fill_between返回的PolyCollection的数据时,不必删除和重绘它。相反,您可以修改底层Path对象的verticescodes属性。

import numpy as np
import matplotlib.pyplot as plt

#dummy data
x = np.arange(10)
y0 = x-1
y1 = x+1

fig = plt.figure()
ax = fig.add_subplot()
p = ax.fill_between(x,y0,y1)

现在你想用新的数据xnewy0newy1new来更新p

v_x = np.hstack([xnew[0],xnew,xnew[-1],xnew[::-1],xnew[0]])
v_y = np.hstack([y1new[0],y0new,y0new[-1],y1new[::-1],y1new[0]])
vertices = np.vstack([v_x,v_y]).T
codes = np.array([1]+(2*len(xnew)+1)*[2]+[79]).astype('uint8')

path = p.get_paths()[0]
path.vertices = vertices
path.codes = codes

说明:path.vertices包含由fill_between绘制的面片的顶点,包括附加的开始和结束位置,path.codes包含如何使用它们的说明(1=将指针移动到,2=将线绘制到,79=闭合多边形)。

相关问题