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

Question

我正在使用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()

第一个plot命令从中生成一个图[0,1]，我可以看到一切正常。最后，y-data数组的起始位置[0,10)大于1，但该图的y限制仍然存在[0,1]。

我知道我可以使用手动更改限制pyplot.ylim(...)，但是我不知道将其更改为什么。在for循环中，我可以告诉pyplot像第一次绘制图形一样缩放限制吗？

Answer 1

您将需要更新轴的dataLim，然后基于dataLim更新轴的viewLim。合适的方法是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()