You are viewing a plain text version of this content. The canonical link for it is here.
Posted to user@spark.apache.org by Jayesh Lalwani <ja...@capitalone.com> on 2018/08/03 15:15:25 UTC

Does row_number over a window cause a shuffle?

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.