In [ ]:

    

import numpy as np
import matplotlib.pyplot as plt

th = 1500
t0 = 2500 


xh = 20

a = np.log(2)/(th - t0)**2
b = np.log(2)/(xh)**2

g_t = lambda t: np.exp(-a*(t-t0)**2)
g_s = lambda x: np.exp(-b*(x)**2)

k = 2*np.pi/(10000)
w = k*137

sint = lambda x,t: np.sin(w*t-k*x)

x = np.linspace(-30, 30, 1e5)

t = 1300
dt = 0.1

tval = sint(x,t)**2*g_s(x)*g_t(t)


x_p = plt.plot(x,tval)


plt.ylim([-1, 1])
plt.xlabel(r"$x \ (a.u.)$",size=20)
plt.ylabel(r"$V_{eff}(x,t) \ (a.u.)$",size=20)
for i in range(10000, 50000):
    
    t += dt
    tval = 1.2*sint(x,t)**2*g_s(x)*g_t(t)-1/np.sqrt(x**2+1)
    plt.title("Time t = " +str(t)+"$(a.u.)$")
    plt.setp(x_p[0], ydata=tval)
    plt.pause(0.001)
    plt.draw()


plt.show()


    

    




---------------------------------------------------------------------------
KeyboardInterrupt                         Traceback (most recent call last)
/usr/local/lib/python3.5/site-packages/matplotlib/artist.py in draw_wrapper(artist, renderer, *args, **kwargs)
     59     def draw_wrapper(artist, renderer, *args, **kwargs):
     60         before(artist, renderer)
---> 61         draw(artist, renderer, *args, **kwargs)
     62         after(artist, renderer)
     63 

/usr/local/lib/python3.5/site-packages/matplotlib/figure.py in draw(self, renderer)
   1157         dsu.sort(key=itemgetter(0))
   1158         for zorder, a, func, args in dsu:
-> 1159             func(*args)
   1160 
   1161         renderer.close_group('figure')

/usr/local/lib/python3.5/site-packages/matplotlib/artist.py in draw_wrapper(artist, renderer, *args, **kwargs)
     59     def draw_wrapper(artist, renderer, *args, **kwargs):
     60         before(artist, renderer)
---> 61         draw(artist, renderer, *args, **kwargs)
     62         after(artist, renderer)
     63 

/usr/local/lib/python3.5/site-packages/matplotlib/axes/_base.py in draw(self, renderer, inframe)
   2322 
   2323         for zorder, a in dsu:
-> 2324             a.draw(renderer)
   2325 
   2326         renderer.close_group('axes')

/usr/local/lib/python3.5/site-packages/matplotlib/artist.py in draw_wrapper(artist, renderer, *args, **kwargs)
     59     def draw_wrapper(artist, renderer, *args, **kwargs):
     60         before(artist, renderer)
---> 61         draw(artist, renderer, *args, **kwargs)
     62         after(artist, renderer)
     63 

/usr/local/lib/python3.5/site-packages/matplotlib/axis.py in draw(self, renderer, *args, **kwargs)
   1109 
   1110         for tick in ticks_to_draw:
-> 1111             tick.draw(renderer)
   1112 
   1113         # scale up the axis label box to also find the neighbors, not

/usr/local/lib/python3.5/site-packages/matplotlib/artist.py in draw_wrapper(artist, renderer, *args, **kwargs)
     59     def draw_wrapper(artist, renderer, *args, **kwargs):
     60         before(artist, renderer)
---> 61         draw(artist, renderer, *args, **kwargs)
     62         after(artist, renderer)
     63 

/usr/local/lib/python3.5/site-packages/matplotlib/axis.py in draw(self, renderer)
    245                 self.gridline.draw(renderer)
    246             if self.tick1On:
--> 247                 self.tick1line.draw(renderer)
    248             if self.tick2On:
    249                 self.tick2line.draw(renderer)

/usr/local/lib/python3.5/site-packages/matplotlib/artist.py in draw_wrapper(artist, renderer, *args, **kwargs)
     59     def draw_wrapper(artist, renderer, *args, **kwargs):
     60         before(artist, renderer)
---> 61         draw(artist, renderer, *args, **kwargs)
     62         after(artist, renderer)
     63 

/usr/local/lib/python3.5/site-packages/matplotlib/lines.py in draw(self, renderer)
    720             self._transform_path(subslice)
    721 
--> 722         transf_path = self._get_transformed_path()
    723 
    724         if self.get_path_effects():

/usr/local/lib/python3.5/site-packages/matplotlib/lines.py in _get_transformed_path(self)
    684         """
    685         if self._transformed_path is None:
--> 686             self._transform_path()
    687         return self._transformed_path
    688 

/usr/local/lib/python3.5/site-packages/matplotlib/lines.py in _transform_path(self, subslice)
    676         else:
    677             _path = self._path
--> 678         self._transformed_path = TransformedPath(_path, self.get_transform())
    679 
    680     def _get_transformed_path(self):

/usr/local/lib/python3.5/site-packages/matplotlib/transforms.py in __init__(self, path, transform)
   2674         self._path = path
   2675         self._transform = transform
-> 2676         self.set_children(transform)
   2677         self._transformed_path = None
   2678         self._transformed_points = None

/usr/local/lib/python3.5/site-packages/matplotlib/transforms.py in set_children(self, *children)
    168         """
    169         for child in children:
--> 170             child._parents[id(self)] = self
    171 
    172     if DEBUG:

/usr/local/Cellar/python3/3.5.2_1/Frameworks/Python.framework/Versions/3.5/lib/python3.5/weakref.py in __setitem__(self, key, value)
    156         if self._pending_removals:
    157             self._commit_removals()
--> 158         self.data[key] = KeyedRef(value, self._remove, key)
    159 
    160     def copy(self):

/usr/local/Cellar/python3/3.5.2_1/Frameworks/Python.framework/Versions/3.5/lib/python3.5/weakref.py in __init__(self, ob, callback, key)
    309 
    310     def __init__(self, ob, callback, key):
--> 311         super().__init__(ob, callback)
    312 
    313 

KeyboardInterrupt: 

In [ ]: