python - 雪花Snowpark Python : Group By and Concat

Question

我在雪花中有这个

<表类=“s-表”> <标题> id_request 警报代码 <正文> 100 R70 100 R69 100 R54 101 R24 101 R93

我想把它变成这样

<表类=“s-表”> <标题> id_request alert_all <正文> 100 R70，R69，R54 101 R24，R93

我尝试写这个，但似乎是错误的

df_alerts_3 = df_alerts_2.groupBy('id_request')\
.agg(concat_ws(lit(','), array_agg('alert_code')).alias('alert_all'))

非常感谢您的帮助

Answer 1

CONCAT_WS在单行上下文中连接字符串。要跨多行连接字符串，您需要使用聚合函数 LISTAGG .

Snowpark 等效项是 snowflake.snowpark.functions.listagg :

Returns the concatenated input values, separated by delimiter string

df.group_by(df.col1).agg(listagg(df.col2. ",")).within_group(df.col2.asc())
df.select(listagg(df["col2"], ",", False)