import mdtraj as md
import numpy as np
def align_traj(traj, frame_index):
"""
This function aligns the given mdtraj trajectory to a specific frame
"""
[insert]
return traj
import mdtraj as md
import numpy as np
def align_traj(traj, frame_index):
"""
This function aligns the given mdtraj trajectory to a specific frame
"""
traj.superpose(traj, frame=frame_index)
return traj
Go to Google Form