同一个figu中的连续情节动画

Question

以matplotlib gallery为例，我想绘制一个离散的Lévy路径（应该可以扩展到2D或3D设置）。所以，我从一个简单的（时间，布朗路径）图开始，想通过把不同的路径“粘在一起”来模拟跳跃。下面是一个最小的工作示例：

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


np.random.seed(19680801)

def generateRandomLines(dt, T, N):
    t = np.linspace(0,T,N)

    dX = np.sqrt(dt) * np.random.randn(1, N)
    X = np.cumsum(dX, axis=1)

    lineData = np.vstack((t,X))

    return lineData

# Returns Line2D objects
def updateLines(num, dataLines, lines):
    for u, v in zip(lines, dataLines):
        u.set_data(v[:2, :num])
    return lines


N = 500 # Number of points
T = .5
dt = T/(N-1)


fig, ax = plt.subplots()

data = [
        [generateRandomLines(dt, T, N),'violet'],
        [.5+generateRandomLines(dt, T, N),'violet'],
        [1+generateRandomLines(dt, T, N),'violet'],
        [1.5+generateRandomLines(dt, T, N),'violet'],
        ]

ax = plt.axes(xlim=(-.5, 2.5), ylim=(-2.5, 2.5))

ax.set_xlabel('t')
ax.set_ylabel('B(t)')
ax.set_title('2D Brownian Paths')

## Create a list of line2D objects
dataLines = [dat[0] for dat in data]
lines = [ax.plot(dat[0][0, 0:1], dat[0][1, 0:1], color=dat[1])[0] for dat in data]

## Create the animation object
anim = animation.FuncAnimation(fig, updateLines, N+1, fargs=(dataLines, lines),
                               interval=20, repeat=False, blit=False)

plt.tight_layout()
plt.show()

目前，在data中生成的四条路径同时被绘制/设置动画。但我希望动画对象在同一个图形中一个接一个地显示出来。你知道吗

显然，我并不真正理解如何处理函数中的line和dataLines对象

def updateLines(num, dataLines, lines):
    for u, v in zip(lines, dataLines):
        u.set_data(v[:2, :num])
    return lines

与question ("Consecutive matplotlib animation in same figure")类似，我尝试对lines对象进行迭代，这会产生zip错误（“参数#1必须支持迭代”）。你知道吗

先谢谢你。你知道吗

Answer 1

用以下方式重新定义updateLines解决了我的问题

# Returns Line2D objects
def updateLines(num, dataLines, lines):
    if num < N:
        lines[0].set_data(dataLines[0][:2, :num])
    elif num < N * 2:
        lines[1].set_data(dataLines[1][:2, :num-N])
    elif num < N * 3:
        lines[2].set_data(dataLines[2][:2, :num-2*N])
    else:
        lines[3].set_data(dataLines[3][:2, :num-3*N])
    return lines

适当增加点数到N * 4 + 1

## Create the animation object
anim = animation.FuncAnimation(fig, updateLines, N * 4 + 1, fargs=(dataLines, lines),
                               interval=20, repeat=False, blit=False)