Простой сценарий построения

Question

Я пытаюсь создать простой сценарий быстрого построения для любой функции x. Я хотел бы передать буквальный возврат f (x) в форме алгебры c, чтобы его можно было использовать в качестве заголовка сюжета. например: Если у меня есть функция def f1(x): return x**2, я хотел бы передать (x ** 2) моей функции построения графика в виде строки, чтобы ее можно было использовать для плитки графика.

пока что у меня есть:

import matplotlib.pyplot as plt
import numpy as np

def f1(x): return x**2 

def create_plot_data(f,xmin,xmax,n):
    xs = [xmin + (xmax - xmin )*i/float(n-1) for i in range(n)]
    ys = [f(x) for x in xs]
    return[xs,ys]

def plot_f(func):
    Plot_title = 'Plot of ' +str(func())#would like the algebraic form of the return of given function.
    plot_label = func.__name__
    xmin = -5
    xmax = 5
    n = 11
    plot_data = create_plot_data(func,xmin,xmax,n)
    xs = plot_data[0]
    ys = plot_data[1]
    plt.plot(xs,ys, label = plot_label)    
    plt.title(Plot_title)
    plt.legend()
    plt.xlabel('x')
    plt.ylabel('f(x)')
    plt.savefig(plot_label + '.png')
    plt.show()
    return

plot_f(f1)

поэтому для def f1(x): return x**2 он должен выглядеть следующим образом: «plot of: x**2 for def f2(x): return (((b-a)**2)**0.5) заголовок должен читать (((b-a)**2)**0.5) введите здесь описание изображения