如何处理matplotlib按钮来调用函数并关闭当前图？

Question

我有一个代码，允许用户选择模型函数以适合某些数据。该模型函数定义了参数，在进行函数的实际拟合之前，需要对这些参数进行采样以获得初始估计值（见图）

单击“ Fit”按钮时，我希望开始进行拟合过程并关闭图形。

plt.show()
button_fit.on_clicked(final_fit(process)) # This calls the function that fits the data
plt.close('all')

但是，由于我使用的是plt.show（），因此需要关闭要评估的button_fit.on_clicked（final_fit（process））的图形。我怎样才能做到这一点？