REF: use _cast_pointwise_result in map

Author: NevroHelios

pandas/core/arrays/arrow/array.py

@@ -1594,7 +1594,8 @@ def to_numpy(
 
     def map(self, mapper, na_action: Literal["ignore"] | None = None):
         if is_numeric_dtype(self.dtype):
-            return map_array(self.to_numpy(), mapper, na_action=na_action)
+            result = map_array(self.to_numpy(), mapper, na_action=na_action)
+            return self._cast_pointwise_result(result)
         else:
             return super().map(mapper, na_action)
 

pandas/core/arrays/base.py

@@ -2530,7 +2530,8 @@ def map(self, mapper, na_action: Literal["ignore"] | None = None):
             If the function returns a tuple with more than one element
             a MultiIndex will be returned.
         """
-        return map_array(self, mapper, na_action=na_action)
+        results = map_array(self, mapper, na_action=na_action)
+        return self._cast_pointwise_result(results)
 
     # ------------------------------------------------------------------------
     # GroupBy Methods

pandas/core/arrays/masked.py

@@ -1395,7 +1395,8 @@ def max(self, *, skipna: bool = True, axis: AxisInt | None = 0, **kwargs):
         return self._wrap_reduction_result("max", result, skipna=skipna, axis=axis)
 
     def map(self, mapper, na_action: Literal["ignore"] | None = None):
-        return map_array(self.to_numpy(), mapper, na_action=na_action)
+        result = map_array(self.to_numpy(), mapper, na_action=na_action)
+        return self._cast_pointwise_result(result)
 
     @overload
     def any(