Function SelectK

cpp/src/arrow/compute/kernels/select_k_test.cc:57–63

template <SortOrder order>
Result<std::shared_ptr<Array>> SelectK(const Datum& values, int64_t k) {
  if (order == SortOrder::Descending) {
    return SelectKUnstable(values, SelectKOptions::TopKDefault(k));
  } else {
    return SelectKUnstable(values, SelectKOptions::BottomKDefault(k));
  }
}

void ValidateSelectK(const Datum& datum, Array& select_k_indices, SortOrder order) {
  ASSERT_TRUE(datum.is_arraylike());