add time_to_ts method

pyneon/stream.py

@@ -83,7 +83,8 @@ def is_uniformly_sampled(self) -> bool:
 
     def time_to_ts(self, time: Union[Number, np.ndarray]) -> np.ndarray:
         """Convert time(s) in seconds to timestamp(s) in nanoseconds."""
-        time
+        time = np.array([time])
+        return np.array([self.ts[np.absolute(self.times - t).argmin()] for t in time])
 
     def crop(
         self,

pyneon/utils/__init__.py

@@ -1,4 +1,3 @@
 from .sample_data import get_sample_data
-from .utils import time_to_ts
 
-__all__ = ["get_sample_data", "time_to_ts"]
+__all__ = ["get_sample_data"]

pyneon/utils/utils.py

@@ -1,10 +1,3 @@
 import pandas as pd
 from typing import Union
 from numbers import Number
-
-
-def time_to_ts(time: Number, data: pd.DataFrame):
-    if "time [s]" not in data.columns or "timestamp [ns]" not in data.columns:
-        raise ValueError('Dataframe must have columns "time [s]" and "timestamp [ns]"')
-    ts = data.loc[(data["time [s]"] - time).abs().idxmin(), "timestamp [ns]"]
-    return ts