pyarrow.compute.select_k_unstable#
- pyarrow.compute.select_k_unstable(input, /, k, sort_keys, *, options=None, memory_pool=None)#
从输入中选择前 k 个有序元素的索引。
此函数从由列键（options.sort_keys）指定的 input 数组、记录批处理或表中，选择前 k 个有序元素的索引数组。输出不保证稳定性。空值（Null）被视为大于任何其他值，因此被排在最后。对于浮点类型，NaN 被视为大于任何其他非空值，但小于空值。
- 参数:
- input类似数组或类似标量的数据
计算函数的参数。
- k
int 按排序顺序选择的前导值数量（即，如果排序顺序为“descending”（降序），则为最大值，否则为最小值）。
- sort_keys序列，包含 (
名称,顺序)元组 用于对输入进行排序的字段/列键名称，以及每个字段/列的排序顺序。order 的可选值为 “ascending”（升序）和 “descending”（降序）。字段名称可以是字符串形式的列名或表达式。
- options
pyarrow.compute.SelectKOptions，可选 传递选项的替代方式。
- memory_pool
pyarrow.MemoryPool, 可选 如果不传递，将从默认内存池分配内存。