BUG: Arrow setitem segfaults when len > 145 000 (pandas-dev#52075)

* BUG: Arrow setitem segfaults when len > 145 000

* Add gh ref

* Address review

* Restrict to bool type

Author: phofl

pandas/core/arrays/arrow/array.py

@@ -1634,6 +1634,10 @@ def _replace_with_mask(
                 indices = pa.array(indices, type=pa.int64())
                 replacements = replacements.take(indices)
             return cls._if_else(mask, replacements, values)
+        if isinstance(values, pa.ChunkedArray) and pa.types.is_boolean(values.type):
+            # GH#52059 replace_with_mask segfaults for chunked array
+            # https://github.com/apache/arrow/issues/34634
+            values = values.combine_chunks()
         try:
             return pc.replace_with_mask(values, mask, replacements)
         except pa.ArrowNotImplementedError:

pandas/tests/extension/test_arrow.py

@@ -2377,3 +2377,12 @@ def test_pickle_old_arrowextensionarray():
     tm.assert_extension_array_equal(result, expected)
     assert result._pa_array == pa.chunked_array(data)
     assert not hasattr(result, "_data")
+
+
+def test_setitem_boolean_replace_with_mask_segfault():
+    # GH#52059
+    N = 145_000
+    arr = ArrowExtensionArray(pa.chunked_array([np.ones((N,), dtype=np.bool_)]))
+    expected = arr.copy()
+    arr[np.zeros((N,), dtype=np.bool_)] = False
+    assert arr._pa_array == expected._pa_array