diff options
| -rw-r--r-- | src/argaze/GazeFeatures.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/argaze/GazeFeatures.py b/src/argaze/GazeFeatures.py index e92e41c..574727c 100644 --- a/src/argaze/GazeFeatures.py +++ b/src/argaze/GazeFeatures.py @@ -166,14 +166,14 @@ class TimeStampedGazePositions(DataStructures.TimeStampedBuffer): precision: specific precision column label if exist. """ - # Select columns + # Copy columns if precision: - df = dataframe[[timestamp, x, y, precision]] + df = dataframe.loc[:, (timestamp, x, y, precision)] else: - df = dataframe[[timestamp, x, y]] + df = dataframe.loc[:, (timestamp, x, y)] # Merge x and y columns into one 'value' column df['value'] = tuple(zip(df[x], df[y])) |