snowflake-cloud-data-platform - 如何在带有 TableFunction 的 Snowpark 中使用分区依据

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()

关于如何使用表函数进行窗口分区有什么建议吗？ snowpark 甚至支持这个吗？

Answer 1

不幸的是，Snowpark 目前不支持此功能。但我们正在努力。