函数
std::unique_ptr< column >	cudf::lists::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())
	返回一个数字列，其中包含给定列表列中每个列表元素的行数。更多...

详细描述

函数文档

std::unique_ptr<column> cudf::lists::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 现在是 {3, 1, 2}

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

参数