python中的动态绘图

Question

我正在尝试在 Python 中创建一个绘图，其中正在绘制的数据会随着我的模拟进行而更新。在 MATLAB 中，我可以使用以下代码来做到这一点：

t = linspace(0, 1, 100);
figure
for i = 1:100
x = cos(2*pi*i*t);
plot(x)
drawnow
end

我正在尝试在animation 模块中使用matplotlib 的FuncAnimation 函数在类中执行此操作。它调用了一个函数plot_voltage，它在我的模拟中的每个时间步之后重新计算电压。我的设置如下：

import matplotlib.pyplot as plt
import matplotlib.animation as animation

def __init__(self):
    ani = animation.FuncAnimation(plt.figure(2), self.plot_voltage)
    plt.draw()

def plot_voltage(self, *args):
    voltages = np.zeros(100)
    voltages[:] = np.nan

    # some code to calculate voltage

    ax1 = plt.figure(2).gca()
    ax1.clear()
    ax1.plot(np.arange(0, len(voltages), 1), voltages, 'ko-')`

当我的模拟运行时，数字会显示，但只是冻结。但是，代码运行没有错误。有人可以告诉我我缺少什么吗？

Answer 1

这是使用FuncAnimation将matlab代码翻译成matplotlib：

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

t = np.linspace(0, 1, 100)
fig = plt.figure()
line, = plt.plot([],[])

def update(i):
    x = np.cos(2*np.pi*i*t)
    line.set_data(t,x)

ani = animation.FuncAnimation(fig, update, 
                frames=np.linspace(1,100,100), interval=100)
plt.xlim(0,1)
plt.ylim(-1,1)
plt.show()