列表元素

组 计数

函数

std::unique_ptr<column> count_elements(lists_column_view const &input, rmm::cuda_stream_view stream = cudf::get_default_stream(), rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref())

返回一个数值列，其中包含给定列表列中每个列表元素的行数。

输出列将与输入列表列具有相同的行数。每个 output[i] 将是 input[i].size()。

l = { {1, 2, 3}, {4}, {5, 6} }
r = count_elements(l)
r is now {3, 1, 2}

任何空输入元素将在输出列中产生相应的空条目。

参数:

• input – 输入列表列

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

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

返回:

包含每行元素数量的新列