object

Author: niruta25

doc/source/user_guide/categorical.rst

@@ -1178,3 +1178,32 @@ Use ``copy=True`` to prevent such a behaviour or simply don't reuse ``Categorica
     This also happens in some cases when you supply a NumPy array instead of a ``Categorical``:
     using an int array (e.g. ``np.array([1,2,3,4])``) will exhibit the same behavior, while using
     a string array (e.g. ``np.array(["a","b","c","a"])``) will not.
+
+.. note::
+
+    When constructing a :class:`pandas.Categorical` from a pandas :class:`Series` or
+     :class:`Index` with ``dtype='object'``, the dtype of the categories will be
+     preserved as ``object``. When constructing from a NumPy array
+     with ``dtype='object'`` or a raw Python sequence, pandas will infer the most
+     specific dtype for the categories (for example, ``str`` if all elements are strings).
+
+.. ipython:: python
+
+    pd.options.future.infer_string = True
+    ser = pd.Series(["foo", "bar", "baz"], dtype="object")
+    idx = pd.Index(["foo", "bar", "baz"], dtype="object")
+    arr = np.array(["foo", "bar", "baz"], dtype="object")
+    pylist = ["foo", "bar", "baz"]
+
+    cat_from_ser = pd.Categorical(ser)
+    cat_from_idx = pd.Categorical(idx)
+    cat_from_arr = pd.Categorical(arr)
+    cat_from_list = pd.Categorical(pylist)
+
+    # Series/Index with object dtype: preserve object dtype
+    assert cat_from_ser.categories.dtype == "object"
+    assert cat_from_idx.categories.dtype == "object"
+
+    # Numpy array or list: infer string dtype
+    assert cat_from_arr.categories.dtype == "str"
+    assert cat_from_list.categories.dtype == "str"

pandas/core/arrays/categorical.py

@@ -457,6 +457,11 @@ def __init__(
                 codes = arr.indices.to_numpy()
                 dtype = CategoricalDtype(categories, values.dtype.pyarrow_dtype.ordered)
             else:
+                # Check for pandas Series/ Index with object dtye
+                preserve_object_dtpe = False
+                if isinstance(values, (ABCSeries, ABCIndex)):
+                    if getattr(values.dtype, "name", None) == "object":
+                        preserve_object_dtpe = True
                 if not isinstance(values, ABCIndex):
                     # in particular RangeIndex xref test_index_equal_range_categories
                     values = sanitize_array(values, None)
@@ -465,15 +470,17 @@ def __init__(
                 except TypeError as err:
                     codes, categories = factorize(values, sort=False)
                     if dtype.ordered:
-                        # raise, as we don't have a sortable data structure and so
-                        # the user should give us one by specifying categories
                         raise TypeError(
                             "'values' is not ordered, please "
                             "explicitly specify the categories order "
                             "by passing in a categories argument."
                         ) from err
 
-                # we're inferring from values
+                # If we should prserve object dtype, force categories to object dtype
+                if preserve_object_dtpe:
+                    from pandas import Index
+
+                    categories = Index(categories, dtype=object, copy=False)
                 dtype = CategoricalDtype(categories, dtype.ordered)
 
         elif isinstance(values.dtype, CategoricalDtype):

pandas/tests/extension/test_categorical.py

@@ -180,6 +180,31 @@ def test_array_repr(self, data, size):
     def test_groupby_extension_agg(self, as_index, data_for_grouping):
         super().test_groupby_extension_agg(as_index, data_for_grouping)
 
+    def test_categorical_preserve_object_dtype_from_pandas(self):
+        import numpy as np
+
+        import pandas as pd
+
+        pd.options.future.infer_string = True
+
+        ser = pd.Series(["foo", "bar", "baz"], dtype="object")
+        idx = pd.Index(["foo", "bar", "baz"], dtype="object")
+        arr = np.array(["foo", "bar", "baz"], dtype="object")
+        pylist = ["foo", "bar", "baz"]
+
+        cat_from_ser = Categorical(ser)
+        cat_from_idx = Categorical(idx)
+        cat_from_arr = Categorical(arr)
+        cat_from_list = Categorical(pylist)
+
+        # Series/Index with object dtype: preserve object dtype
+        assert cat_from_ser.categories.dtype == "object"
+        assert cat_from_idx.categories.dtype == "object"
+
+        # Numpy array or list: infer string dtype
+        assert cat_from_arr.categories.dtype == "str"
+        assert cat_from_list.categories.dtype == "str"
+
 
 class Test2DCompat(base.NDArrayBacked2DTests):
     def test_repr_2d(self, data):