From 41e6582c7609d4d052600308a15e75fc6c14fc06 Mon Sep 17 00:00:00 2001 From: Joris Van den Bossche Date: Tue, 23 Mar 2021 13:54:58 +0100 Subject: [PATCH] COMPAT: make Categorical._codes settable again --- pandas/core/arrays/categorical.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/pandas/core/arrays/categorical.py b/pandas/core/arrays/categorical.py index 1398db6960cc8..0c7ee2d43753e 100644 --- a/pandas/core/arrays/categorical.py +++ b/pandas/core/arrays/categorical.py @@ -1756,6 +1756,10 @@ def to_dense(self): def _codes(self) -> np.ndarray: return self._ndarray + @_codes.setter + def _codes(self, value: np.ndarray): + self._ndarray = value + def _from_backing_data(self, arr: np.ndarray) -> Categorical: assert isinstance(arr, np.ndarray) assert arr.dtype == self._ndarray.dtype