Skip to content

Commit

Permalink
Backport PR #56184 on branch 2.1.x (BUG: mode not preserving object d…
Browse files Browse the repository at this point in the history
…type for string option) (#56383)
  • Loading branch information
phofl committed Dec 7, 2023
1 parent aacdf61 commit d1764d4
Show file tree
Hide file tree
Showing 4 changed files with 17 additions and 2 deletions.
1 change: 1 addition & 0 deletions doc/source/whatsnew/v2.1.4.rst
Expand Up @@ -31,6 +31,7 @@ Bug fixes
- Fixed bug in :meth:`DataFrame.to_hdf` raising when columns have ``StringDtype`` (:issue:`55088`)
- Fixed bug in :meth:`Index.insert` casting object-dtype to PyArrow backed strings when ``infer_string`` option is set (:issue:`55638`)
- Fixed bug in :meth:`Series.__ne__` resulting in False for comparison between ``NA`` and string value for ``dtype="string[pyarrow_numpy]"`` (:issue:`56122`)
- Fixed bug in :meth:`Series.mode` not keeping object dtype when ``infer_string`` is set (:issue:`56183`)
- Fixed bug in :meth:`Series.str.split` and :meth:`Series.str.rsplit` when ``pat=None`` for :class:`ArrowDtype` with ``pyarrow.string`` (:issue:`56271`)
- Fixed bug in :meth:`Series.str.translate` losing object dtype when string option is set (:issue:`56152`)
-
Expand Down
6 changes: 5 additions & 1 deletion pandas/core/series.py
Expand Up @@ -2218,7 +2218,11 @@ def mode(self, dropna: bool = True) -> Series:

# Ensure index is type stable (should always use int index)
return self._constructor(
res_values, index=range(len(res_values)), name=self.name, copy=False
res_values,
index=range(len(res_values)),
name=self.name,
copy=False,
dtype=self.dtype,
).__finalize__(self, method="mode")

def unique(self) -> ArrayLike: # pylint: disable=useless-parent-delegation
Expand Down
10 changes: 10 additions & 0 deletions pandas/tests/series/test_reductions.py
Expand Up @@ -29,6 +29,16 @@ def test_mode_extension_dtype(as_period):
tm.assert_series_equal(res, ser)


def test_mode_infer_string():
# GH#56183
pytest.importorskip("pyarrow")
ser = Series(["a", "b"], dtype=object)
with pd.option_context("future.infer_string", True):
result = ser.mode()
expected = Series(["a", "b"], dtype=object)
tm.assert_series_equal(result, expected)


def test_reductions_td64_with_nat():
# GH#8617
ser = Series([0, pd.NaT], dtype="m8[ns]")
Expand Down
2 changes: 1 addition & 1 deletion pandas/tests/test_algos.py
Expand Up @@ -2102,7 +2102,7 @@ def test_timedelta_mode(self):
tm.assert_series_equal(ser.mode(), exp)

def test_mixed_dtype(self):
exp = Series(["foo"])
exp = Series(["foo"], dtype=object)
ser = Series([1, "foo", "foo"])
tm.assert_numpy_array_equal(algos.mode(ser.values), exp.values)
tm.assert_series_equal(ser.mode(), exp)
Expand Down

0 comments on commit d1764d4

Please sign in to comment.