figure changes size at end of loop if drawn dynamically

As per the title, consider the following snippet:

import time
import ipympl
import matplotlib.pyplot as plt

fig, ax = plt.subplots(1, 1, figsize=(5, 5))
data = []
plt.show()
for idx in range(4):
    data.append(idx ** 2)
    ax.clear()
    ax.plot(data)
    fig.canvas.draw()
    time.sleep(0.2)

and this is what I get running it:

I used plt.show() at the beginning of the loop because nothing gets drawn otherwise.
Is there a way to avoid this?

首先，在带有 ipympl 的 Jupyter Notebook 中运行提供的 Python 代码片段，并观察图形在循环期间调整大小的情况。然后跟踪示例所使用的动态绘制行为，最后验证图形在全部四次更新中都保持其预期大小。