rwpenney commented on a change in pull request #30745:
URL: https://github.com/apache/spark/pull/30745#discussion_r583402905
##########
File path: python/pyspark/sql/functions.py
##########
@@ -222,6 +222,34 @@ def sum_distinct(col):
return _invoke_function_over_column("sum_distinct", col)
+def product(col):
+ """
+ Aggregate function: returns the product of the values in a group.
+
+ .. versionadded:: 3.2.0
+
+ Parameters
+ ----------
+ col : str, :class:`Column`
+ column containing values to be multiplied together
+
+ Examples
+ --------
+ >>> df = spark.range(1, 10).toDF('x').withColumn('mod3', col('x') % 3)
+ >>> prods = df.groupBy('mod3').agg(product('x').alias('product'))
+ >>> prods.orderBy('mod3').show()
+ +----+-------+
+ |mod3|product|
+ +----+-------+
+ | 0| 162.0|
+ | 1| 28.0|
+ | 2| 80.0|
+ +----+-------+
+
+ """
+ return _invoke_function("product", _to_java_column(col))
Review comment:
Thanks, @ueshin for spotting this, and your other "nits". Hopefully all
resolved now.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
