pyarrow.compute.partition_nth_indices#
- pyarrow.compute.partition_nth_indices(array, /, pivot, *, null_placement='at_end', options=None, memory_pool=None)#
返回将数组围绕枢轴（pivot）进行分区的索引。
此函数计算出一组索引，这些索引定义了输入数组的非稳定部分排序。
输出结果使得第 N 个索引指向输入数组按排序顺序排列后的第 N 个元素，且所有在第 N 个索引之前的索引所指向的输入元素，均小于或等于第 N 个及其之后位置的元素。
默认情况下，空值（null）被视为大于任何其他值，因此被分区到数组末尾。对于浮点类型，NaN 被视为大于任何其他非空值，但小于空值。
枢轴索引 N 必须在 PartitionNthOptions 中给出。空值和 NaN 的处理方式也可以在 PartitionNthOptions 中进行更改。
- 参数:
- array类数组对象 (Array-like)
计算函数的参数。
- pivot
int 等效排序数组中枢轴元素的索引。
- null_placement
str, 默认 “at_end” 输入中的空值应分区到的位置。接受的值为 “at_start”、“at_end”。
- options
pyarrow.compute.PartitionNthOptions, 可选 传递选项的替代方式。
- memory_pool
pyarrow.MemoryPool, 可选 如果不传递，将从默认内存池分配内存。