Use zero-copy slice when step=1

LucasG0 · Mar 25, 2024 · b9df1e6 · b9df1e6
1 parent 5385bc8
commit b9df1e6
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/python/pyarrow/array.pxi b/python/pyarrow/array.pxi
@@ -562,8 +562,13 @@ def _normalize_slice(object arrow_obj, slice key):
         Py_ssize_t n = len(arrow_obj)
 
     start, stop, step = key.indices(n)
-    indices = np.arange(start, stop, step)
-    return arrow_obj.take(indices)
+
+    if step != 1:
+        indices = np.arange(start, stop, step)
+        return arrow_obj.take(indices)
+    else:
+        length = max(stop - start, 0)
+        return arrow_obj.slice(start, length)
 
 
 cdef Py_ssize_t _normalize_index(Py_ssize_t index,