In [1]:
%matplotlib notebook
In [2]:
import matplotlib.pyplot as plt
from pytraj import view
import pytraj as pt
from nglview import link
import nglview
fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111)
ax.set_xlabel('frame number')
ax.set_ylabel('End to end distance (Angstrom)')
traj = pt.datafiles.load_tz2()
data = pt.radgyr(traj, '@CA')
ax.plot(data)
# convert pytraj's Trajectory to nglview.Trajectory
ntraj = pt.view.to_nglview(traj)
tv = nglview.TrajectoryViewer(ntraj)
link_mpl(tv, ax, data)
tv