Ejemplo n.º 1
0
def load_discrete_trajectory(filename):
    r"""Read discrete trajectory form binary file.   

    Parameters
    ---------- 
    filename : str 
        The filename of the discrete state trajectory file. 
        The filename can either contain the full or the 
        relative path to the file.
    
    Returns
    -------
    dtraj : (M,) ndarray of int
        Discrete state trajectory

    See also
    --------
    save_discrete_trajectory

    Notes
    -----
    The binary file is a one dimensional numpy array
    of integers stored in numpy .npy format.

    Examples
    --------

    >>> from tempfile import NamedTemporaryFile    
    >>> from pyemma.msm.io import load_discrete_trajectory, save_discrete_trajectory

    Use temporary file
    
    >>> tmpfile = NamedTemporaryFile()

    Discrete trajectory
    
    >>> dtraj = np.array([0, 1, 0, 0, 1, 1, 0])

    Write to disk (as ascii file)
    
    >>> save_discrete_trajectory(tmpfile.name, dtraj)

    Read from disk

    >>> X = load_discrete_trajectory(tmpfile.name)
    >>> X
    array([0, 1, 0, 0, 1, 1, 0])       
    
    """
    return trajectory.load_discrete_trajectory(filename)
Ejemplo n.º 2
0
def load_discrete_trajectory(filename):
    r"""Read discrete trajectory form binary file.   

    Parameters
    ---------- 
    filename : str 
        The filename of the discrete state trajectory file. 
        The filename can either contain the full or the 
        relative path to the file.
    
    Returns
    -------
    dtraj : (M,) ndarray of int
        Discrete state trajectory

    See also
    --------
    save_discrete_trajectory

    Notes
    -----
    The binary file is a one dimensional numpy array
    of integers stored in numpy .npy format.

    Examples
    --------

    >>> from tempfile import NamedTemporaryFile    
    >>> from pyemma.msm.io import load_discrete_trajectory, save_discrete_trajectory

    Use temporary file
    
    >>> tmpfile = NamedTemporaryFile()

    Discrete trajectory
    
    >>> dtraj = np.array([0, 1, 0, 0, 1, 1, 0])

    Write to disk (as ascii file)
    
    >>> save_discrete_trajectory(tmpfile.name, dtraj)

    Read from disk

    >>> X = load_discrete_trajectory(tmpfile.name)
    >>> X
    array([0, 1, 0, 0, 1, 1, 0])       
    
    """
    return trajectory.load_discrete_trajectory(filename)
Ejemplo n.º 3
0
 def test_load_discrete_trajectory(self):
     dtraj_n = np.load(self.filename)
     dtraj = dt.load_discrete_trajectory(self.filename)
     self.assertTrue(np.all(dtraj_n == dtraj))
 def test_load_discrete_trajectory(self):
     dtraj_n=np.load(self.filename)
     dtraj=dt.load_discrete_trajectory(self.filename)
     self.assertTrue(np.all(dtraj_n==dtraj))