Skip to content

Commit 9e2c11d

Browse files
use super() in take function
1 parent fd7b27c commit 9e2c11d

File tree

2 files changed

+13
-23
lines changed

2 files changed

+13
-23
lines changed

pandas/core/arrays/datetimelike.py

Lines changed: 8 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,6 @@
118118
from pandas.core.algorithms import (
119119
isin,
120120
map_array,
121-
take,
122121
unique1d,
123122
)
124123
from pandas.core.array_algos import datetimelike_accumulations
@@ -2355,7 +2354,7 @@ def interpolate(
23552354
if not copy:
23562355
return self
23572356
return type(self)._simple_new(out_data, dtype=self.dtype)
2358-
2357+
23592358
def take(
23602359
self,
23612360
indices: TakeIndexer,
@@ -2364,27 +2363,18 @@ def take(
23642363
fill_value: Any = None,
23652364
axis: AxisInt = 0,
23662365
) -> Self:
2367-
2368-
if allow_fill:
2369-
fill_value = self._validate_scalar(fill_value)
2370-
2371-
new_data = take(
2372-
self._ndarray,
2373-
indices,
2374-
allow_fill=allow_fill,
2375-
fill_value=fill_value,
2376-
axis=axis,
2377-
)
2378-
result = self._from_backing_data(new_data)
2379-
2366+
result = super().take(
2367+
indices=indices, allow_fill=allow_fill, fill_value=fill_value, axis=axis
2368+
)
2369+
23802370
indices = np.asarray(indices, dtype=np.intp)
23812371
maybe_slice = lib.maybe_indices_to_slice(indices, len(self))
2382-
2372+
23832373
if isinstance(maybe_slice, slice):
23842374
freq = self._get_getitem_freq(maybe_slice)
23852375
result.freq = freq
2386-
2387-
return result
2376+
2377+
return result
23882378

23892379
# --------------------------------------------------------------
23902380
# Unsorted

pandas/tests/indexes/multi/test_get_level_values.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -126,10 +126,10 @@ def test_values_loses_freq_of_underlying_index():
126126

127127
def test_get_level_values_gets_frequency_correctly():
128128
# GH#57949 GH#58327
129-
datetime_index = pd.date_range(start=pd.to_datetime("1/1/2018"),
130-
periods = 4,
131-
freq = 'YS')
129+
datetime_index = pd.date_range(
130+
start=pd.to_datetime("1/1/2018"), periods=4, freq="YS"
131+
)
132132
other_index = ["A"]
133-
multi_index = pd.MultiIndex.from_product([datetime_index, other_index])
133+
multi_index = MultiIndex.from_product([datetime_index, other_index])
134134

135-
assert multi_index.get_level_values(0).freq is datetime_index.freq
135+
assert multi_index.get_level_values(0).freq == datetime_index.freq

0 commit comments

Comments
 (0)