BUG: using .replace() on a Series containing np.nan raising ValueErro… · pandas-dev/pandas@4e139f6 (original) (raw)

`@@ -499,3 +499,16 @@ def test_replace_with_compiled_regex(self):

`

499

499

`result = s.replace({regex: "z"}, regex=True)

`

500

500

`expected = pd.Series(["z", "b", "c"])

`

501

501

`tm.assert_series_equal(result, expected)

`

``

502

+

``

503

`+

def test_pandas_replace_na(self):

`

``

504

`+

GH#43344

`

``

505

`+

ser = pd.Series(["AA", "BB", "CC", "DD", "EE", "", pd.NA], dtype="string")

`

``

506

`+

regex_mapping = {

`

``

507

`+

"AA": "CC",

`

``

508

`+

"BB": "CC",

`

``

509

`+

"EE": "CC",

`

``

510

`+

"CC": "CC-REPL",

`

``

511

`+

}

`

``

512

`+

result = ser.replace(regex_mapping, regex=True)

`

``

513

`+

exp = pd.Series(["CC", "CC", "CC-REPL", "DD", "CC", "", pd.NA], dtype="string")

`

``

514

`+

tm.assert_series_equal(result, exp)

`