zero323 commented on a change in pull request #27406:
[SPARK-30681][PYSPARK][SQL] Add higher order functions API to PySpark
URL: https://github.com/apache/spark/pull/27406#discussion_r373487641
##########
File path: python/pyspark/sql/functions.py
##########
@@ -2840,6 +2840,367 @@ def from_csv(col, schema, options={}):
return Column(jc)
+def _invoke_higher_order_function(name, cols, funs):
+ """
+ Invokes expression identified by name,
+ (relative to ```org.apache.spark.sql.catalyst.expressions``)
+ and wraps the result with Column (first Scala one, then Python).
+
+ :param name: Name of the expression
+ :param cols: a list of columns
+ :param funs: a list of tuples ((*Column) -> Column, Iterable[int])
+ where the second element represent allowed arities
+
+ :return: a Column
+ """
+ sc = SparkContext._active_spark_context
+ expressions = sc._jvm.org.apache.spark.sql.catalyst.expressions
+ expr = getattr(expressions, name)
+
+ jcols = [_to_java_column(col).expr() for col in cols]
+ jfuns = [_create_lambda(f, a) for f, a in funs]
+
+ return Column(sc._jvm.Column(expr(*jcols + jfuns)))
+
+
+@since(3.0)
Review comment:
One could hope...
----------------------------------------------------------------
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]
With regards,
Apache Git Services
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
