将Java用户定义的聚合函数注册为 SQL 函数。
Syntax
registerJavaUDAF(name, javaClassName)
参数
| 参数 | 类型 | Description |
|---|---|---|
name |
str | 用户定义的聚合函数的名称。 |
javaClassName |
str | Java类的完全限定名称。 |
退货
None
示例
spark.udf.registerJavaUDAF("javaUDAF", "test.org.apache.spark.sql.MyDoubleAvg")
df = spark.createDataFrame([(1, "a"), (2, "b"), (3, "a")], ["id", "name"])
df.createOrReplaceTempView("df")
q = "SELECT name, javaUDAF(id) as avg from df group by name order by name desc"
spark.sql(q).collect()
# [Row(name='b', avg=102.0), Row(name='a', avg=102.0)]