python - 在 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()

第一个绘图命令从 [0,1] 生成一个绘图，我可以看到一切正常。最后，y 数据数组从 [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()