重复行计算器 - 计算文本中重复行的比例
输入列名 | 说明 |
|---|---|
texts | 待处理的文本列,要求元素类型为字符串 |
重复行比例列,元素为浮点数,表示重复行比例
下面的代码展示了如何使用 daft 运行算子计算文本中重复行的比例。
from __future__ import annotations import os import daft from daft import col from daft.las.functions.text.repeated_lines_calculator import RepeatedLinesCalculator from daft.las.functions.udf import las_udf if __name__ == "__main__": if os.getenv("DAFT_RUNNER", "native") == "ray": import logging import ray def configure_logging(): logging.basicConfig( level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s", datefmt="%Y-%m-%d %H:%M:%S.%s".format(), ) logging.getLogger("tracing.span").setLevel(logging.WARNING) logging.getLogger("daft_io.stats").setLevel(logging.WARNING) logging.getLogger("DaftStatisticsManager").setLevel(logging.WARNING) logging.getLogger("DaftFlotillaScheduler").setLevel(logging.WARNING) logging.getLogger("DaftFlotillaDispatcher").setLevel(logging.WARNING) ray.init(dashboard_host="0.0.0.0", runtime_env={"worker_process_setup_hook": configure_logging}) daft.context.set_runner_ray() daft.set_execution_config(actor_udf_ready_timeout=600) daft.set_execution_config(min_cpu_per_task=0) samples = { "text": [ "第一行内容\n第二行内容\n第一行内容\n第三行内容", "第一行内容\n第一行内容\n第一行内容", "第一行内容\n\n第二行内容\n \n第三行内容", ] } ds = daft.from_pydict(samples) ds = ds.with_column( "repeated_ratio", las_udf( RepeatedLinesCalculator, construct_args={}, )(col("text")), ) ds.show() # ╭────────────┬────────────────────╮ # │ text ┆ repeated_ratio │ # │ --- ┆ --- │ # │ String ┆ Float64 │ # ╞════════════╪════════════════════╡ # │ 第一行内容 ┆ 0.25 │ # │ 第二行内容 ┆ │ # │ 第一行内容 ┆ │ # │ 第三行内容 ┆ │ # ├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌┤ # │ 第一行内容 ┆ 0.6666666666666666 │ # │ 第一行内容 ┆ │ # │ 第一行内容 ┆ │ # ├╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌┤ # │ 第一行内容 ┆ 0 │ # │ ┆ │ # │ 第二行内容 ┆ │ # │ ┆ │ # │ 第三行内容 ┆ │ # ╰────────────┴────────────────────╯