我正在使用matplotlib在Python中绘制数据。我正在根据一些计算更新绘图的数据,并希望ylim和xlim自动重新缩放。相反,发生的是基于初始图的极限设置比例。 MWE是
import random
import matplotlib.pyplot as pyplot
pyplot.ion()
x = range(10)
y = lambda m: [m*random.random() for i in range(10)]
pLine, = pyplot.plot(x, y(1))
for i in range(10):
pLine.set_ydata(y(i+1))
pyplot.draw()
第一个绘图命令从[0,1]
生成一个图,我可以看到一切都很好。最后,y数据数组来自[0,10)
,其中大部分大于1
,但数字的y限制仍为[0,1]
。
我知道我可以使用pyplot.ylim(...)
手动更改限制,但我不知道要将它们更改为什么。在for
循环中,我是否可以告诉pyplot缩放限制,就好像它是第一次被绘制一样?
答案 0 :(得分:43)
您需要更新轴的dataLim,然后根据dataLim更新轴的viewLim。 approrpiate方法是axes.relim()
和ax.autoscale_view()
方法。
您的示例如下所示:
import random
import matplotlib.pyplot as pyplot
pyplot.ion()
x = range(10)
y = lambda m: [m*random.random() for i in range(10)]
pLine, = pyplot.plot(x, y(1))
for i in range(10):
pLine.set_ydata(y(i+1))
ax = pyplot.gca()
# recompute the ax.dataLim
ax.relim()
# update ax.viewLim using the new dataLim
ax.autoscale_view()
pyplot.draw()