From 6449eaaa930cd961e20056ca6b21832901b78ac5 Mon Sep 17 00:00:00 2001
From: Joris Van den Bossche <jorisvandenbossche@gmail.com>
Date: Thu, 31 Oct 2024 17:12:43 +0100
Subject: [PATCH] BUG: preserve (object) dtype in factorize (#60118)

* BUG: preserve (object) dtype in factorize

* add fallback for float16

(cherry picked from commit 13926e5e298acf328b0c1347f008ef3f9c4eb078)
---
 pandas/core/base.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/pandas/core/base.py b/pandas/core/base.py
index e98f1157572bb..a67003895d288 100644
--- a/pandas/core/base.py
+++ b/pandas/core/base.py
@@ -48,6 +48,7 @@
 from pandas.core.dtypes.generic import (
     ABCDataFrame,
     ABCIndex,
+    ABCMultiIndex,
     ABCSeries,
 )
 from pandas.core.dtypes.missing import (
@@ -1198,13 +1199,18 @@ def factorize(
         if uniques.dtype == np.float16:
             uniques = uniques.astype(np.float32)
 
-        if isinstance(self, ABCIndex):
-            # preserve e.g. MultiIndex
+        if isinstance(self, ABCMultiIndex):
+            # preserve MultiIndex
             uniques = self._constructor(uniques)
         else:
             from pandas import Index
 
-            uniques = Index(uniques)
+            try:
+                uniques = Index(uniques, dtype=self.dtype)
+            except NotImplementedError:
+                # not all dtypes are supported in Index that are allowed for Series
+                # e.g. float16 or bytes
+                uniques = Index(uniques)
         return codes, uniques
 
     _shared_docs[