Skip to content

Commit 0360c9a

Browse files
committed
REF: re-use ArrowEA methods in ArrowStringArray
1 parent b50610c commit 0360c9a

File tree

1 file changed

+13
-15
lines changed

1 file changed

+13
-15
lines changed

pandas/core/arrays/string_arrow.py

+13-15
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,6 @@
5656
ArrayLike,
5757
AxisInt,
5858
Dtype,
59-
Scalar,
6059
Self,
6160
npt,
6261
)
@@ -293,6 +292,19 @@ def astype(self, dtype, copy: bool = True):
293292
_str_startswith = ArrowStringArrayMixin._str_startswith
294293
_str_endswith = ArrowStringArrayMixin._str_endswith
295294
_str_pad = ArrowStringArrayMixin._str_pad
295+
_str_match = ArrowExtensionArray._str_match
296+
_str_fullmatch = ArrowExtensionArray._str_fullmatch
297+
_str_lower = ArrowExtensionArray._str_lower
298+
_str_upper = ArrowExtensionArray._str_upper
299+
_str_strip = ArrowExtensionArray._str_strip
300+
_str_lstrip = ArrowExtensionArray._str_lstrip
301+
_str_rstrip = ArrowExtensionArray._str_rstrip
302+
_str_removesuffix = ArrowStringArrayMixin._str_removesuffix
303+
_str_get = ArrowStringArrayMixin._str_get
304+
_str_capitalize = ArrowStringArrayMixin._str_capitalize
305+
_str_title = ArrowStringArrayMixin._str_title
306+
_str_swapcase = ArrowStringArrayMixin._str_swapcase
307+
_str_slice_replace = ArrowStringArrayMixin._str_slice_replace
296308

297309
def _str_contains(
298310
self, pat, case: bool = True, flags: int = 0, na=np.nan, regex: bool = True
@@ -348,15 +360,6 @@ def _str_len(self):
348360
result = pc.utf8_length(self._pa_array)
349361
return self._convert_int_result(result)
350362

351-
_str_match = ArrowExtensionArray._str_match
352-
_str_fullmatch = ArrowExtensionArray._str_fullmatch
353-
_str_lower = ArrowExtensionArray._str_lower
354-
_str_upper = ArrowExtensionArray._str_upper
355-
_str_strip = ArrowExtensionArray._str_strip
356-
_str_lstrip = ArrowExtensionArray._str_lstrip
357-
_str_rstrip = ArrowExtensionArray._str_rstrip
358-
_str_removesuffix = ArrowStringArrayMixin._str_removesuffix
359-
360363
def _str_removeprefix(self, prefix: str):
361364
if not pa_version_under13p0:
362365
return ArrowExtensionArray._str_removeprefix(self, prefix)
@@ -461,8 +464,3 @@ def _cmp_method(self, other, op):
461464

462465
class ArrowStringArrayNumpySemantics(ArrowStringArray):
463466
_na_value = np.nan
464-
_str_get = ArrowStringArrayMixin._str_get
465-
_str_capitalize = ArrowStringArrayMixin._str_capitalize
466-
_str_title = ArrowStringArrayMixin._str_title
467-
_str_swapcase = ArrowStringArrayMixin._str_swapcase
468-
_str_slice_replace = ArrowStringArrayMixin._str_slice_replace

0 commit comments

Comments
 (0)