字典搜索#

group dictionary_search

函数

std::unique_ptr<scalar> get_index(dictionary_column_view const &dictionary, scalar const &key, rmm::cuda_stream_view stream = cudf::get_default_stream(), rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref())#

返回给定键的索引值。

如果字典中不存在该键，返回的标量将具有 is_valid()==false

Throws:

cudf::logic_error – 如果 key.type() != dictionary.keys().type()

Parameters:

• dictionary – 用于搜索键的字典。

• key – 在字典键集中搜索的值。

• stream – 用于设备内存操作和内核启动的CUDA流。

• mr – 用于分配返回标量的设备内存的设备内存资源。

Returns:

字典中键的数值标量索引值。