在jupyter Notebook上绘制动态曲线

1、前言

在上一篇文章中介绍了在如Pycharm、VsCode、IDEA等,非交互式编程软件中绘制动态曲线的实现原理,那么问题来了:在如jupyter notebook这类交互式编程工具下又如何实现呢？

2、实现原理

在实现原理上,实则上是“换汤不换药”,本质上还是依靠动画的刷新原理。只不过在如Pycharm上的软件上,我们绘制的曲线图是以窗口形式打开的,因此我们刷新的是窗口;而在如Jupyter Notebook的软件上,由于我们的曲线图是显示在整个输出界面上,故刷新的是整个输出界面。

另外还有一个需要注意的小细节:由于整个过程是在输出界面作图和刷新,故需要使用plt进行作图

3、实现源码

import torch
import matplotlib.pyplot as plt
from IPython import display
x_list = []
y_list = []
for i in range(100):
    x_list.append(i)
    y_list.append(i**2)
    display.clear_output(wait=True)
    plt.plot(x_list, y_list)
    plt.xlabel('x')
    plt.ylabel('y')
    plt.show()
    plt.pause(0.1)

实现效果预览: