pyspark:聚合字段的新列名

Question

我有以下带有聚合函数的代码:

new_df = my_df.groupBy('id').agg({"id": "count", "money":"max"})

然后我的新列是 COUNT(id) 和 MAX(money)。我可以自己指定列名而不是使用默认列名吗？例如。我希望它们被称为 my_count_id 和 my_max_money。我怎么做？谢谢!

Answer 1

使用列而不是字典:

>>> from pyspark.sql.functions import *
>>> my_df.groupBy('id').agg(count("id").alias("some name"), max("money").alias("some other name"))