如何通过TableFunction在Snowpark中使用partition by

Question

我正在尝试在snowpark中使用UDTF，但不能按列进行分区。我想要的sql查询是这样的：

select mcount.result from CUSTOMER, table(map_count(name) over (partition by name)) mcount;

这里的"map_count“是我的JavaScript UDTF。以下是Snowpark中的代码片段：

val session = Session.builder.configs(configs).create
val df = session.table("CUSTOMER")
val window = Window.partitionBy(col("name"))
val result = df.join(TableFunction("map_count"), col("name"))
//result.show()

有什么建议如何通过表函数来使用窗口分区吗？雪地公园甚至支持这一点吗？

Answer 1

不幸的是，Snowpark目前不支持这一点。但我们正在努力。