在Matplotlib中自动重新缩放ylim和xlim

时间:2012-06-11 16:32:43

标签: python matplotlib

我正在使用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缩放限制,就好像它是第一次被绘制一样?

1 个答案:

答案 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()