Move offsets downcasting to MultiIndex._engine

pandas-dev · Apr 25, 2024 · c512c51 · c512c51
1 parent 3ab8601
commit c512c51
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/pandas/_libs/index.pyx b/pandas/_libs/index.pyx
@@ -712,8 +712,7 @@ cdef class BaseMultiIndexCodesEngine:
             Pre-calculated offsets, one for each level of the index.
         """
         self.levels = levels
-        # Downcast the type if possible, to prevent upcasting when shifting codes:
-        self.offsets = offsets.astype(np.min_scalar_type(offsets[0]), copy=False)
+        self.offsets = offsets
 
         # Transform labels in a single array, and add 2 so that we are working
         # with positive integers (-1 for NaN becomes 1). This enables us to

diff --git a/pandas/core/indexes/multi.py b/pandas/core/indexes/multi.py
@@ -1201,7 +1201,9 @@ def _engine(self):
         # equivalent to sorting lexicographically the codes themselves. Notice
         # that each level needs to be shifted by the number of bits needed to
         # represent the _previous_ ones:
-        offsets = np.concatenate([lev_bits[1:], [0]]).astype("uint64")
+        offsets = np.concatenate([lev_bits[1:], [0]])
+        # Downcast the type if possible, to prevent upcasting when shifting codes:
+        offsets = offsets.astype(np.min_scalar_type(int(offsets[0])))
 
         # Check the total number of bits needed for our representation:
         if lev_bits[0] > 64: