Add to_dataframe() to Record.

Author: Ivorforce

wfdb/io/record.py

@@ -851,6 +851,35 @@ def _arrange_fields(self, channels, sampfrom, smooth_frames):
         # Adjust date and time if necessary
         self._adjust_datetime(sampfrom=sampfrom)
 
+    def to_dataframe(self) -> pd.DataFrame:
+        """
+        Create a dataframe containing the same data as the record.
+
+
+        Returns
+        -------
+        A dataframe, with sig_name in the columns. The index is a DatetimeIndex
+        if both base_date and base_time were set, otherwise a TimedeltaIndex.
+        """
+        if self.base_datetime is not None:
+            index = pd.date_range(
+                start=self.base_datetime,
+                periods=self.sig_len,
+                freq=datetime.timedelta(seconds=1 / self.fs),
+            )
+        else:
+            index = pd.timedelta_range(
+                start=datetime.timedelta(),
+                periods=self.sig_len,
+                freq=datetime.timedelta(seconds=1 / self.fs),
+            )
+
+        return pd.DataFrame(
+            data=self.p_signal,
+            index=index,
+            columns=self.sig_name
+        )
+
 
 class MultiRecord(BaseRecord, _header.MultiHeaderMixin):
     """