Skip to content
Snippets Groups Projects
Commit f844d87d authored by Olivier Bertrand's avatar Olivier Bertrand
Browse files

Fix to_hdf saving and loading in Trajectory

parent 0a46d705
Branches
Tags
No related merge requests found
...@@ -233,6 +233,7 @@ class Trajectory(pd.DataFrame): ...@@ -233,6 +233,7 @@ class Trajectory(pd.DataFrame):
def q_3(self, q_3): def q_3(self, q_3):
self.__set_q_i(3, q_3) self.__set_q_i(3, q_3)
# overload of save/load function
def read_csv(self, filename, sep=',', header=[0, 1], index_col=0): def read_csv(self, filename, sep=',', header=[0, 1], index_col=0):
""" Load from a hdf file """ Load from a hdf file
""" """
...@@ -242,8 +243,19 @@ class Trajectory(pd.DataFrame): ...@@ -242,8 +243,19 @@ class Trajectory(pd.DataFrame):
return self return self
def read_hdf(self, filename): def read_hdf(self, filename):
raise NameError('Not implemented') df = pd.read_hdf(filename)
self.from_dataframe(df)
return self
def to_hdf(self, filename):
df = pd.DataFrame(self)
df.to_hdf(filename, key='posorients')
def to_csv(self, filename):
df = pd.DataFrame(self)
df.to_csv(filename)
# initialisation from variables
def from_array(self, nparray, rotconv): def from_array(self, nparray, rotconv):
""" Assign trajectory from a numpy array """ Assign trajectory from a numpy array
N x 6 (rotconv = Euler angles) N x 6 (rotconv = Euler angles)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment