我想使用MatPlotLib绘制图形,该图形随时间变化。在每个时间步长处,都会将另一个数据点添加到绘图中。但是,应该只显示一个图形,其外观会随着时间而变化。
在我的测试示例中,该图是一个简单的线性图(y = x)。这是我尝试过的:
for i in range(100):
x = range(i)
y = range(i)
plt.plot(x, y)
plt.ion()
plt.show()
time.sleep(1)
但是,这里发生的是创建了多个窗口,因此到循环结束时,我有100个窗口。另外,我注意到对于最近的窗口,它只是一个白色窗口,并且该图仅出现在下一步中。
因此,我的两个问题是:
1)如何更改代码,以便仅显示一个窗口,其内容会随时间变化?
2)如何更改代码,以便在最近的时间步中,图实际上显示在窗口上,而不是仅显示白色窗口?
谢谢!
最佳答案
(1)
您可以在开始处设置plt.ion()
并将所有图形绘制到同一窗口。在循环中,使用plt.draw()
显示图形,使用plt.pause(t)
暂停。请注意,t
可能很小,但是该命令必须在此处才能使动画在大多数后端上运行。
您可能需要在使用plt.gca().cla()
绘制新内容之前清除轴。
import matplotlib.pyplot as plt
plt.ion()
for i in range(100):
x = range(i)
y = range(i)
# plt.gca().cla() # optionally clear axes
plt.plot(x, y)
plt.title(str(i))
plt.draw()
plt.pause(0.1)
plt.show(block=True) # block=True lets the window stay open at the end of the animation.
除了这种非常简单的方法之外,还可以使用http://matplotlib.org/examples/animation/index.html中提供的任何动画示例。
(2)
为了在新窗口中获取每个图,请使用
plt.figure()
并删除plt.ion()
。也只显示最后的窗口:import matplotlib.pyplot as plt
for i in range(100):
x = range(i)
y = range(i)
plt.figure()
plt.plot(x, y)
plt.title(str(i))
plt.show()
请注意,您可能会发现在这两种情况下,第一个图都是空的,仅仅是因为对于
i=0
,range(i) == []
是一个没有任何点的空列表。即使对于i=1
,也只绘制了一个点,但是当然,没有线可以将单个点与其自身相连。