Использование Matplotlib с Tkinter

Question

Нужна помощь с matplotlib в tkinter, похоже, не получается получить запись для отображения правильного графика. Я хотел бы набрать 'x ** 2' и показать график для этой функции, но она не работает. Если бы кто-нибудь мог мне помочь, я был бы признателен. Заранее спасибо.

import tkinter as tk
from tkinter import * 
import matplotlib.pyplot as plt
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg, NavigationToolbar2Tk
import numpy as np
import matplotlib
from matplotlib.figure import Figure
matplotlib.use('TkAgg')

screen = tk.Tk()


screen.title('Function Graph')
screen.geometry('350x200')
function = Entry(screen, width = 20)
function.place(x=123, y=92)
f_label = Label(screen, text='Plot Function: ')
f_label.place(x=145, y= 70)

def plot_g():

    x = np.linspace(-3,3,100)

    y = function.get()

    new_w = tk.Toplevel(screen)


    fig = plt.figure()
    ax = fig.add_subplot(1, 1, 1)
    ax.spines['left'].set_position('center') 
    ax.spines['bottom'].set_position('zero') 
    ax.spines['right'].set_color('none') 
    ax.spines['top'].set_color('none') 
    ax.xaxis.set_ticks_position('bottom') 
    ax.yaxis.set_ticks_position('left')
    plt.grid()
    plt.plot(x,y)
    canvas = FigureCanvasTkAgg(fig, master=new_w)
    canvas.draw()
    canvas.get_tk_widget().pack(side = 'bottom', fill= BOTH, expand=True)
    toolbar = NavigationToolbar2Tk(canvas, new_w)
    toolbar.update()



f_button = Button(screen, text= 'Enter', command = plot_g)
f_button.place(x=160, y=120)


screen.mainloop()