def __getitem__(self, key): result = self._data[key] if isinstance(result, type(self._data)): return type(self)(result, name=self.name) # Includes cases where we get a 2D ndarray back for MPL compat deprecate_ndim_indexing(result) return result
def __getitem__(self, key): result = self._data[key] if isinstance(result, type(self._data)): if result.ndim == 1: return type(self)(result, name=self.name) # Unpack to ndarray for MPL compat # pandas\core\indexes\extension.py:220: error: "ExtensionArray" has # no attribute "_data" [attr-defined] result = result._data # type: ignore[attr-defined] # Includes cases where we get a 2D ndarray back for MPL compat deprecate_ndim_indexing(result) return result