From 8270cfc19eca1a07f7ffc4671db09ad4b39daf33 Mon Sep 17 00:00:00 2001 From: Brock Mendel Date: Tue, 7 Jan 2020 19:16:37 -0800 Subject: [PATCH] REF: move repeat to ExtensionIndex --- pandas/core/indexes/datetimelike.py | 6 ------ pandas/core/indexes/extension.py | 5 +++++ 2 files changed, 5 insertions(+), 6 deletions(-) diff --git a/pandas/core/indexes/datetimelike.py b/pandas/core/indexes/datetimelike.py index a10e0f63b841d..06678672c8126 100644 --- a/pandas/core/indexes/datetimelike.py +++ b/pandas/core/indexes/datetimelike.py @@ -460,12 +460,6 @@ def isin(self, values, level=None): return algorithms.isin(self.asi8, values.asi8) - @Appender(_index_shared_docs["repeat"] % _index_doc_kwargs) - def repeat(self, repeats, axis=None): - nv.validate_repeat(tuple(), dict(axis=axis)) - result = type(self._data)(self.asi8.repeat(repeats), dtype=self.dtype) - return self._shallow_copy(result) - @Appender(_index_shared_docs["where"] % _index_doc_kwargs) def where(self, cond, other=None): values = self.view("i8") diff --git a/pandas/core/indexes/extension.py b/pandas/core/indexes/extension.py index f0f407e9e8308..0178fe864801f 100644 --- a/pandas/core/indexes/extension.py +++ b/pandas/core/indexes/extension.py @@ -164,6 +164,11 @@ class ExtensionIndex(Index): _data: ExtensionArray + def repeat(self, repeats, axis=None): + nv.validate_repeat(tuple(), dict(axis=axis)) + result = self._data.repeat(repeats, axis=axis) + return self._shallow_copy(result) + def take(self, indices, axis=0, allow_fill=True, fill_value=None, **kwargs): nv.validate_take(tuple(), kwargs) indices = ensure_platform_int(indices)