列表元素#

group lists_elements

函数

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}

任何空输入元素将导致输出列中的相应条目为空。

Parameters:

  • input – 输入列表列

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

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

Returns:

新列，显示每行的元素数量