字典搜索

组 搜索

函数

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

抛出:

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

参数:

• dictionary – 要搜索键的字典。

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

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

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

返回值:

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