如何使 ipywidget 按钮更新 matplotlib 图形的特定轴？

Question

如何让 Jupyter 笔记本中的 ipywidget 按钮更新特定轴上的绘图？

我已经知道如何在使用单轴时让按钮更新绘图，如下所示：

import ipywidgets as widgets
from IPython.display import display
import matplotlib.pyplot as plt
import numpy as np

btn = widgets.Button(description='Click')
display(btn)

output = widgets.Output()

def on_click_fn(obj):
    output.clear_output()
    values = np.random.rand(10)
    with output:
        plt.plot(values)
        plt.show()

btn.on_click(on_click_fn)
display(output)

在此示例中，单击按钮会更新绘图并显示一组新的 10 个随机点。我认为将其扩展到更新特定轴会很简单，并尝试了以下操作：

import ipywidgets as widgets
from IPython.display import display
import matplotlib.pyplot as plt
import numpy as np

btn = widgets.Button(description='Click')
display(btn)

output = widgets.Output()
fig, ax = plt.subplots(ncols=2)

def on_click_fn(obj):
    output.clear_output()
    values = np.random.rand(10)
    with output:
        ax[0].plot(values)
        plt.show()

btn.on_click(on_click_fn)
display(output)

但是，单击此示例中的按钮似乎没有任何作用。我尝试了添加/删除plt.show()调用的不同组合，改用fig.draw()，使用fig.canvas.draw_idle()等，但没有取得多大成功。完成此任务的正确、最少“hacky”方式是什么？

---

注意：This question 只是关于如何让按钮更新绘图，就像我的第一个示例一样，而不是让按钮仅更新特定轴。

Answer 1

使用此代码

import ipywidgets as widgets
from IPython.display import display
import matplotlib.pyplot as plt
import numpy as np
%matplotlib widget

btn = widgets.Button(description='Click')
display(btn)

output = widgets.Output()
fig, ax = plt.subplots(ncols=2)

def on_click_fn(obj):
    output.clear_output()
    values = np.random.rand(10)
    with output:
        ax[0].plot(values)
        plt.show()

btn.on_click(on_click_fn)
display(output)

我得到了这个输出