Github user HyukjinKwon commented on a diff in the pull request:
https://github.com/apache/spark/pull/21820#discussion_r203924999
--- Diff: python/pyspark/sql/functions.py ---
@@ -2551,6 +2551,27 @@ def map_concat(*cols):
return Column(jc)
+@since(2.4)
+def sequence(start, stop, step=None):
+ """
+ Generate a sequence of integers from start to stop, incrementing by
step.
+ If step is not set, incrementing by 1 if start is less than or equal
to stop, otherwise -1.
+
+ >>> df1 = spark.createDataFrame([(-2, 2)], ('C1', 'C2'))
+ >>> df1.select(sequence('C1', 'C2').alias('r')).collect()
+ [Row(r=[-2, -1, 0, 1, 2])]
+ >>> df2 = spark.createDataFrame([(4, -4, -2)], ('C1', 'C2', 'C3'))
--- End diff --
What happens if the `step` is positive in this case?
---
---------------------------------------------------------------------
To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org
For additional commands, e-mail: reviews-h...@spark.apache.org
