Does row_number over a window cause a shuffle?

classic Classic list List threaded Threaded
1 message Options
Reply | Threaded
Open this post in threaded view
|

Does row_number over a window cause a shuffle?

JayeshLalwani
I have some code that adds a column that contains a row_number over a window. It looks somewhat like this

val sortColumns: List[Column] = r.sortFields.map(sf => sf.map(col(_))).getOrElse(List(col(s"defaultSortCol")))
val partitionWindow = Window.partitionBy(s"groupByCol")
val window = partitionWindow.orderBy(sortColumns:_*)
df.withColumn(s"MyRank", row_number().over(window))

Will this cause a shuffle because I am partitioning by groupByCol? What happens if groupByCol is a boolean? Will this cause the dataframe to be partition into 2?

Or does Spark do something like using an accumulator to assign the row numbers? 


The information contained in this e-mail is confidential and/or proprietary to Capital One and/or its affiliates and may only be used solely in performance of work or services for Capital One. The information transmitted herewith is intended only for use by the individual or entity to which it is addressed. If the reader of this message is not the intended recipient, you are hereby notified that any review, retransmission, dissemination, distribution, copying or other use of, or taking of any action in reliance upon this information is strictly prohibited. If you have received this communication in error, please contact the sender and delete the material from your computer.