Adding specific from_dataframe classmethod for TimeStampedGazePositions.

1 files changed, 10 insertions, 0 deletions

diff --git a/src/argaze/GazeFeatures.py b/src/argaze/GazeFeatures.py
index ff561fb..faf1093 100644
--- a/src/argaze/GazeFeatures.py
+++ b/src/argaze/GazeFeatures.py
@@ -147,6 +147,16 @@ class TimeStampedGazePositions(DataStructures.TimeStampedBuffer):
 
             return TimeStampedGazePositions({ast.literal_eval(ts_str): json_buffer[ts_str] for ts_str in json_buffer})
 
+    @classmethod
+    def from_dataframe(self, dataframe: pandas.DataFrame, exclude=[]) -> TimeStampedBufferType:
+        """Create a TimeStampedGazePositions from [Pandas DataFrame](https://pandas.pydata.org/docs/reference/api/pandas.DataFrame.html)."""
+
+        dataframe.drop(exclude, inplace=True, axis=True)
+
+        assert(dataframe.index.name == 'timestamp')
+
+        return TimeStampedGazePositions(dataframe.to_dict('index'))
+
 GazeMovementType = TypeVar('GazeMovement', bound="GazeMovement")
 # Type definition for type annotation convenience