Backport PR pandas-dev#57454 on branch 2.2.x (Release the gil in take…

… for axis=1) (pandas-dev#57484) Backport PR pandas-dev#57454: Release the gil in take for axis=1 Co-authored-by: Patrick Hoefler <[email protected]>
meeseeksmachine · Feb 18, 2024 · ab8541c · ab8541c
1 parent 32d2b99
commit ab8541c
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/pandas/_libs/algos_take_helper.pxi.in b/pandas/_libs/algos_take_helper.pxi.in
@@ -184,6 +184,17 @@ def take_2d_axis1_{{name}}_{{dest}}(ndarray[{{c_type_in}}, ndim=2] values,
 
     fv = fill_value
 
+    {{if c_type_in == c_type_out != "object"}}
+    with nogil:
+        for i in range(n):
+            for j in range(k):
+                idx = indexer[j]
+                if idx == -1:
+                    out[i, j] = fv
+                else:
+                    out[i, j] = values[i, idx]
+
+    {{else}}
     for i in range(n):
         for j in range(k):
             idx = indexer[j]
@@ -195,6 +206,7 @@ def take_2d_axis1_{{name}}_{{dest}}(ndarray[{{c_type_in}}, ndim=2] values,
                 {{else}}
                 out[i, j] = values[i, idx]
                 {{endif}}
+    {{endif}}
 
 
 @cython.wraparound(False)