ignoring_gravity

Asked: 2024-09-11 16:07:12 +0800 CST2024-09-11 16:07:12 +0800 CST 2024-09-11 16:07:12 +0800 CST

pyarrow chunkedarray 获取给定索引处的项目

说我有

In [3]: import pyarrow as pa

In [4]: ca = pa.chunked_array([[1,2,3], [4,5,6]])

我想提取元素[1, 4, 2]并最终得到

<pyarrow.lib.Int64Array object at 0x7f6eb43c2d40>
[
  2,
  5,
  3
]

就像我在做 NumPy 风格的索引一样

Best Answer

0x26res

使用take函数

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

相关问题