Skip to content

Commit ab0c761

Browse files
committed
REF: re-use ArrowEA methods in ArrowStringArray
1 parent 7b6e4f3 commit ab0c761

File tree

1 file changed

+13
-15
lines changed

1 file changed

+13
-15
lines changed

pandas/core/arrays/string_arrow.py

Lines changed: 13 additions & 15 deletions
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
)
@@ -295,6 +294,19 @@ def astype(self, dtype, copy: bool = True):
295294
_str_startswith = ArrowStringArrayMixin._str_startswith
296295
_str_endswith = ArrowStringArrayMixin._str_endswith
297296
_str_pad = ArrowStringArrayMixin._str_pad
297+
_str_match = ArrowExtensionArray._str_match
298+
_str_fullmatch = ArrowExtensionArray._str_fullmatch
299+
_str_lower = ArrowExtensionArray._str_lower
300+
_str_upper = ArrowExtensionArray._str_upper
301+
_str_strip = ArrowExtensionArray._str_strip
302+
_str_lstrip = ArrowExtensionArray._str_lstrip
303+
_str_rstrip = ArrowExtensionArray._str_rstrip
304+
_str_removesuffix = ArrowStringArrayMixin._str_removesuffix
305+
_str_get = ArrowStringArrayMixin._str_get
306+
_str_capitalize = ArrowStringArrayMixin._str_capitalize
307+
_str_title = ArrowStringArrayMixin._str_title
308+
_str_swapcase = ArrowStringArrayMixin._str_swapcase
309+
_str_slice_replace = ArrowStringArrayMixin._str_slice_replace
298310

299311
def _str_contains(
300312
self, pat, case: bool = True, flags: int = 0, na=np.nan, regex: bool = True
@@ -354,15 +366,6 @@ def _str_len(self):
354366
result = pc.utf8_length(self._pa_array)
355367
return self._convert_int_result(result)
356368

357-
_str_match = ArrowExtensionArray._str_match
358-
_str_fullmatch = ArrowExtensionArray._str_fullmatch
359-
_str_lower = ArrowExtensionArray._str_lower
360-
_str_upper = ArrowExtensionArray._str_upper
361-
_str_strip = ArrowExtensionArray._str_strip
362-
_str_lstrip = ArrowExtensionArray._str_lstrip
363-
_str_rstrip = ArrowExtensionArray._str_rstrip
364-
_str_removesuffix = ArrowStringArrayMixin._str_removesuffix
365-
366369
def _str_removeprefix(self, prefix: str):
367370
if not pa_version_under13p0:
368371
return ArrowExtensionArray._str_removeprefix(self, prefix)
@@ -471,8 +474,3 @@ def _cmp_method(self, other, op):
471474

472475
class ArrowStringArrayNumpySemantics(ArrowStringArray):
473476
_na_value = np.nan
474-
_str_get = ArrowStringArrayMixin._str_get
475-
_str_capitalize = ArrowStringArrayMixin._str_capitalize
476-
_str_title = ArrowStringArrayMixin._str_title
477-
_str_swapcase = ArrowStringArrayMixin._str_swapcase
478-
_str_slice_replace = ArrowStringArrayMixin._str_slice_replace

0 commit comments

Comments
 (0)