pyarrow：查找分块数组的差异

Question

如果我有一个分块数组，如何找到它的差异（类似于 pandas.Series.diff 或 Polars.Series.diff）？

例如如果我从

开始

import pyarrow as pa
ca = pa.chunked_array([[1,3, 2], [5, 2, 1]])

我想最终得到一个带有值的数组（或分块数组）：

[null, 2, -1, 3, -1, -1]

Answer 1

import pyarrow.compute as pc

pc.pairwise_diff(ca.combine_chunks())