lists/contains.hpp
Go to the documentation of this file.
1 /*
2  * SPDX-FileCopyrightText: Copyright (c) 2021-2024, NVIDIA CORPORATION.
3  * SPDX-License-Identifier: Apache-2.0
4  */
5 #pragma once
6 
7 #include <cudf/column/column.hpp>
9 #include <cudf/utilities/export.hpp>
11 
12 namespace CUDF_EXPORT cudf {
13 namespace lists {
38 std::unique_ptr<column> contains(
39  cudf::lists_column_view const& lists,
40  cudf::scalar const& search_key,
43 
62 std::unique_ptr<column> contains(
63  cudf::lists_column_view const& lists,
64  cudf::column_view const& search_keys,
67 
85 std::unique_ptr<column> contains_nulls(
86  cudf::lists_column_view const& lists,
89 
94 enum class duplicate_find_option : int32_t {
95  FIND_FIRST = 0,
96  FIND_LAST
97 };
98 
127 std::unique_ptr<column> index_of(
128  cudf::lists_column_view const& lists,
129  cudf::scalar const& search_key,
130  duplicate_find_option find_option = duplicate_find_option::FIND_FIRST,
133 
164 std::unique_ptr<column> index_of(
165  cudf::lists_column_view const& lists,
166  cudf::column_view const& search_keys,
167  duplicate_find_option find_option = duplicate_find_option::FIND_FIRST,
170  // end of group
172 } // namespace lists
173 } // namespace CUDF_EXPORT cudf
A non-owning, immutable view of device data as a column of elements, some of which may be null as ind...
Given a column-view of lists type, an instance of this class provides a wrapper on this compound colu...
An owning class to represent a singular value.
Definition: scalar.hpp:40
Class definition for cudf::column.
rmm::cuda_stream_view const get_default_stream()
Get the current default stream.
std::unique_ptr< column > index_of(cudf::lists_column_view const &lists, cudf::column_view const &search_keys, duplicate_find_option find_option=duplicate_find_option::FIND_FIRST, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Create a column of values indicating the position of a search key row within the corresponding list r...
std::unique_ptr< column > contains(cudf::lists_column_view const &lists, cudf::column_view const &search_keys, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Create a column of bool values indicating whether the list rows of the first column contain the corre...
duplicate_find_option
Option to choose whether index_of() returns the first or last match of a search key in a list row.
std::unique_ptr< column > contains_nulls(cudf::lists_column_view const &lists, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Create a column of bool values indicating whether each row in the lists column contains at least one ...
@ FIND_FIRST
Finds first instance of a search key in a list row.
@ FIND_LAST
Finds last instance of a search key in a list row.
rmm::device_async_resource_ref get_current_device_resource_ref()
Get the current device memory resource reference.
detail::cccl_async_resource_ref< cuda::mr::resource_ref< cuda::mr::device_accessible > > device_async_resource_ref
Class definition for cudf::lists_column_view.
cuDF interfaces
Definition: host_udf.hpp:26