Обратное преобразование Лапласа

from mpmath import *
import time
mp.dps = 15#число членов, используемых в приближении
mp.pretty = True
start = time.time()
def  run_invertlaplace(tt,fp,ft):
         for i in range(0,len(tt)):
               print('Значение тестовой функции : %s'%ft(tt[i]))
               print('Полученное значение функции : %s'%invertlaplace(fp,tt[i],method='talbot'))
               print(' Значение h(t) : %s. Абсолютная погрешность :%s.'%(ft(tt[i]), ft(tt[i])-invertlaplace(fp,tt[i],method='talbot')))
         stop = time.time()
         print ("Время, затраченное на обратное преобразование Лапласа: %s"%(stop-start)) 
tt = [0.001, 0.01, 0.1, 1, 10]#список значений отсчётов времени
fp = lambda p: 1/(p+1)**2#передаточная функция частоты для тестирования программы
ft = lambda t: t*exp(-t)# переходная функция времени для тестирования программы
run_invertlaplace(tt,fp,ft)

# -*- coding: utf8 -*-    
import numpy as np
import matplotlib.pyplot as plt
import matplotlib as mpl
from mpmath import *
mpl.rcParams['font.family'] = 'fantasy'
mpl.rcParams['font.fantasy'] = 'Comic Sans MS, Arial'
mp.dps = 5; mp.pretty = True
def  run_invertlaplace(tt,fp,tau):
      y=[]
      for i in np.arange(0,len(tt)):
               if tt[i]<=tau:
                        y.append(0)
               else:
                        y.append(invertlaplace(fp,tt[i],method='talbot'))
      return y   
tau=5
tt = np.arange(0,100,0.05)
T1=5;T2=4;T3=4;K=1.5;tau=14
fp = lambda p: K*exp(-tau*p)/((T1*p+1)*(T2*p+1)*(T3*p+1)*p)
y=run_invertlaplace(tt,fp,tau)
plt.title('Переходная характеристика объекта управления \n, полученная через invertlaplace ')
plt.plot(tt,y,'r')
plt.grid(True)
plt.show()