theta_intersection

使用 Datasketches 交集对象返回 Datasketches ThetaSketch 对象的两个二进制表示形式的交集。

Syntax

from pyspark.sql import functions as sf

sf.theta_intersection(col1, col2)

参数

参数	类型	Description
col1	pyspark.sql.Column 或 str	第一个 Theta 草图。
col2	pyspark.sql.Column 或 str	第二个 Theta 草图。

退货

pyspark.sql.Column：相交 ThetaSketch 的二进制表示形式。

例子

示例 1：获取两个 Theta 草图的交集

from pyspark.sql import functions as sf
df = spark.createDataFrame([(1,1),(2,2),(3,2),(3,3)], "struct<v1:int,v2:int>")
df = df.agg(
    sf.theta_sketch_agg("v1").alias("sketch1"),
    sf.theta_sketch_agg("v2").alias("sketch2")
)
df.select(sf.theta_sketch_estimate(sf.theta_intersection(df.sketch1, "sketch2"))).show()

+-----------------------------------------------------------+
|theta_sketch_estimate(theta_intersection(sketch1, sketch2))|
+-----------------------------------------------------------+
|                                                          3|
+-----------------------------------------------------------+