Pyspark：查找斜率：使用 groupBy 和 agg 获取 2 列或行中的日期和相关值的值来计算斜率

Question

我需要根据2列对数据进行分组，然后根据该数据获取与日期和对应值相关的数据，根据日期进行排序并计算斜率。 这就是我正在使用的：

df=df.groupBy('device','sv').agg(collect_list('date_col').alias('date_col'), collect_list('value').alias('value')).sort('date_col').withColumn('slope', get_slope(F.col("date_col"), F.col("value")))

我担心这不能正确计算斜率，因为我使用列表并分别收集数据并仅根据第一个列表进行排序。 任何人都可以帮我解决这个问题。 有没有办法根据两列收集数据然后排序？ 谢谢