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 obtém itens em índices fornecidos

Diga que eu tenho

In [3]: import pyarrow as pa

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

Gostaria de extrair elementos [1, 4, 2]e terminar com

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

como se eu estivesse fazendo indexação no estilo NumPy

Best Answer

0x26res

Use a takefunção

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

relate perguntas