&res created ARROW-17951: ---------------------------- Summary: Support ChunkedArray mask in replace operations Key: ARROW-17951 URL: https://issues.apache.org/jira/browse/ARROW-17951 Project: Apache Arrow Issue Type: Improvement Environment: pyarrow==9.0.0 python 3.9 Reporter: &res
Replace operations (more specifically replace_with_mask) don't support chunked array for the mask argument. But one would expect that ReplaceMaskChunked, which support chunked array for its input data (the data to be replaced), should also support chunked array for its mask. {code:java} pc.replace_with_mask(pa.array(["a", "b"]), pa.array([True, False]), pa.scalar("c", pa.string())) pc.replace_with_mask( pa.chunked_array([pa.array(["a", "b"])]), pa.array([True, False]), pa.scalar("c", pa.string()), ) with pytest.raises(pa.ArrowInvalid, match=r"Mask must be array or scalar, not ChunkedArray"): pc.replace_with_mask( pa.chunked_array([pa.array(["a", "b"])]), pa.chunked_array([pa.array([True, False])]), pa.scalar("c", pa.string()), ) {code} -- This message was sent by Atlassian Jira (v8.20.10#820010)