python计算程序怎么改出图

做了一个输入数，才能出结果的程序，现在想根据计算过程出个图，能在原程序上改么？我直接在计算方程式后面加上plt.plot显示错误

可以，你想要什么样图？你说的图是不是界面？