Question

我想动画一些数据，我一直在关注另一个堆栈问题here中的示例以启用暂停。但是，我正在做一些不同的事情。在该示例中，它们使用正弦函数，该函数可以在参数中采用非整数值。我正在做的是在y轴上绘制一组数据并将其与相应的x值匹配（在这种情况下是时间）。

期望输出

我只想让输出绘制数据并更新刻度标记（因此下面的ax.set_xlim(0, i/sf)），同时能够暂停或播放动画。

关闭但不正确的解决方案

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

length = 8e6
data = np.random.rand(length)
sf = 100 #Sampling frequency in MHz
x = np.arange(length)/sf

pause = False

def init():
    line.set_data([], [])
    return line,

def animate(i):
    if not pause:
        y = data[0:i]
        xax = x[0:i]
        line.set_data(xax, y)
        ax.set_xlim(0, i/sf)
        time_text.set_text(time_template%(i/sf))
        return line, time_text

def onPress(event):
    if event.key==' ':
        global pause
        pause ^= True 

fig = plt.figure(figsize=(15,15)) 
ax = fig.add_subplot(111)
ax.set_ylim(min(data),max(data))
line, = ax.plot([], [], lw=2)

time_template = 'Time = %.1f $\mu$s'    # prints running simulation time
time_text = ax.text(0.05, 0.9, '', transform=ax.transAxes)
anim = am.FuncAnimation(fig, animate, interval=40, init_func=init)
fig.canvas.mpl_connect('key_press_event', onPress)

plt.show()

这个解决方案的问题在于，当我暂停然后取消暂停动画时，绘图会将数据绘制到该时刻的数据，如果我没有暂停它的话。

您应该能够复制并粘贴此代码，以便在您自己的计算机上重现。

还试过

我尝试了与我链接的示例类似的结构。它也不起作用。

当我运行程序时，问题似乎没有发生。我认为它正在绘制一些东西，因为当我试图在情节中移动时，我可以看到情节。

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

length = 8e6
data = np.random.rand(length)
sf = 100 #Sampling frequency in MHz
x = np.arange(length)/sf 

pause = False

def init():
    line.set_data([], [])
    return line,

def theData():
    i=0
    while i<50:
         if not pause:
            y = data[0:i]
            t = x[0:i]
            i=i+1
            yield t, y, i

def animate(theData):
    t = theData[0]
    y = theData[1]
    i = theData[2]
    line.set_data(t, y)
    ax.set_xlim(0, i/sf)
    time_text.set_text(time_template%(t))
    return line, time_text

def onPress(event):
    if event.key==' ':
        global pause
        pause ^= True 

fig = plt.figure(figsize=(15,15)) 
ax = fig.add_subplot(111)
ax.set_ylim(min(data),max(data))
line, = ax.plot([], [], lw=2)

time_template = 'Time = %.1f $\mu$s'    # prints running simulation time
time_text = ax.text(0.05, 0.9, '', transform=ax.transAxes)
anim = am.FuncAnimation(fig, animate, theData, interval=40, init_func=init)
fig.canvas.mpl_connect('key_press_event', onPress)

plt.show()

Answer 1

传递给animate的参数是一个帧编号。但是，无论动画是否暂停，此数字都会递增。

因此，引入您自己的全局变量frame，该变量记录真实的帧编号，并且仅在动画未暂停时递增：

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

length = 8*10**6
data = np.random.rand(length)
sf = 100 #Sampling frequency in MHz
x = np.arange(length)/sf

pause = False
frame = 0

def init():
    line.set_data([], [])
    return line,

def animate(i):
    global frame
    if not pause:
        frame += 1
        y = data[0:frame]
        xax = x[0:frame]
        line.set_data(xax, y)
        ax.set_xlim(0, frame/sf)
        time_text.set_text(time_template%(frame/sf))
        return line, time_text

def onPress(event):
    if event.key==' ':
        global pause
        pause ^= True 

fig = plt.figure(figsize=(15,15)) 
ax = fig.add_subplot(111)
ax.set_ylim(min(data),max(data))
line, = ax.plot([], [], lw=2)

time_template = 'Time = %.1f $\mu$s'    # prints running simulation time
time_text = ax.text(0.05, 0.9, '', transform=ax.transAxes)
anim = am.FuncAnimation(fig, animate, interval=40, init_func=init)
fig.canvas.mpl_connect('key_press_event', onPress)

plt.show()

基于示例将动画与代码暂停时遇到问题

期望输出

关闭但不正确的解决方案

还试过

1 个答案: