Manage str.encode for pyarrow.lib.LargeStringScalar

Author: droussea2001

pandas/core/strings/object_array.py

@@ -239,8 +239,15 @@ def _str_fullmatch(
         return self._str_map(f, na_value=na, dtype=np.dtype(bool))
 
     def _str_encode(self, encoding, errors: str = "strict"):
-        f = lambda x: x.encode(encoding, errors=errors)
-        return self._str_map(f, dtype=object)
+        def encode_func(x):
+            if x is str:
+                return x.encode(encoding=encoding, errors=errors)
+            else:
+                # Manage AttributeError: 'pyarrow.lib.LargeStringScalar'
+                # object has no attribute 'encode'
+                return str(x).encode(encoding=encoding, errors=errors)
+
+        return self._str_map(encode_func, dtype=object)
 
     def _str_find(self, sub, start: int = 0, end=None):
         return self._str_find_(sub, start, end, side="left")

pandas/tests/strings/test_strings.py

@@ -561,7 +561,16 @@ def test_encode_errors_kwarg(any_string_dtype):
         ser.str.encode("cp1252")
 
     result = ser.str.encode("cp1252", "ignore")
-    expected = ser.map(lambda x: x.encode("cp1252", "ignore"))
+
+    def encode_func(x):
+        if x is str:
+            return x.encode("cp1252", "ignore")
+        else:
+            # Manage AttributeError: 'pyarrow.lib.LargeStringScalar'
+            # object has no attribute 'encode'
+            return str(x).encode("cp1252", "ignore")
+
+    expected = ser.map(encode_func).astype("object")
     tm.assert_series_equal(result, expected)