spark structured streaming 上可以基于 groupBy window 的结果自定义处理方法吗

官方的例子是这样的，最后用了一个 count()d 的方法来做统计：

words = ... # streaming DataFrame of schema { timestamp: Timestamp, word: String } # Group the data by window and word and compute the count of each group windowedCounts = words.groupBy( window(words.timestamp, "10 minutes", "5 minutes"), words.word ).count() 

现在我想这样做，基于 groupBy window 出来的 GroupedData 数据，使用自定义的方式来处理, 比如在 g()中，增加一些自定义逻辑。

schema = StructType( [StructField("key", StringType()), StructField("avg_min", DoubleType())] ) @panda_udf(schema, functiOnType=PandasUDFType.GROUPED_MAP) def g(df): #whatever user-defined code words = ... # streaming DataFrame of schema { timestamp: Timestamp, word: String } windowedCounts = words.groupBy( window(words.timestamp, "10 minutes", "5 minutes"), words.word ).apply(g) 

我尝试过，但是没成功。不知道是我用法不对，还是说不能将用户自定义方法作用于 groupBy window 后的数据？