add take function including frequency for Timedelta and Datetime Arrays

Author: [Annika Rudolph]

pandas/core/arrays/datetimelike.py

@@ -20,6 +20,7 @@
 
 import numpy as np
 
+from pandas.core.algorithms import take
 from pandas._config.config import get_option
 
 from pandas._libs import (
@@ -68,6 +69,7 @@
     SequenceIndexer,
     TimeAmbiguous,
     TimeNonexistent,
+    TakeIndexer,
     npt,
 )
 from pandas.compat.numpy import function as nv
@@ -1792,7 +1794,7 @@ def strftime(self, date_format: str) -> npt.NDArray[np.object_]:
     ----------
     freq : str or Offset
         The frequency level to {op} the index to. Must be a fixed
-        frequency like 's' (second) not 'ME' (month end). See
+        frequency like 'S' (second) not 'ME' (month end). See
         :ref:`frequency aliases <timeseries.offset_aliases>` for
         a list of possible `freq` values.
     ambiguous : 'infer', bool-ndarray, 'NaT', default 'raise'
@@ -1830,11 +1832,6 @@ def strftime(self, date_format: str) -> npt.NDArray[np.object_]:
     ------
     ValueError if the `freq` cannot be converted.
 
-    See Also
-    --------
-    DatetimeIndex.floor : Perform floor operation on the data to the specified `freq`.
-    DatetimeIndex.snap : Snap time stamps to nearest occurring frequency.
-
     Notes
     -----
     If the timestamps have a timezone, {op}ing will take place relative to the
@@ -2363,6 +2360,35 @@ def interpolate(
         if not copy:
             return self
         return type(self)._simple_new(out_data, dtype=self.dtype)
+    
+    def take(
+        self,
+        indices: TakeIndexer,
+        *,
+        allow_fill: bool = False,
+        fill_value: Any = None,
+        axis: AxisInt = 0,
+    ) -> Self:
+        
+        if allow_fill:
+            fill_value = self._validate_scalar(fill_value)
+
+        new_data = take(
+            self._ndarray,
+            indices,
+            allow_fill=allow_fill,
+            fill_value=fill_value,
+            axis=axis,
+        )    
+        result = self._from_backing_data(new_data)
+        indices = np.asarray(indices, dtype=np.intp)
+        maybe_slice = lib.maybe_indices_to_slice(indices, len(self))
+        
+        if isinstance(maybe_slice, slice):
+            freq = self._get_getitem_freq(maybe_slice)
+            result.freq = freq
+        
+        return result 
 
     # --------------------------------------------------------------
     # Unsorted