row_number

Función Window: devuelve un número secuencial a partir de 1 dentro de una partición de ventana.

Syntax

from pyspark.sql import functions as sf

sf.row_number()

Parámetros

Esta función no toma ningún parámetro.

Devoluciones

pyspark.sql.Column: columna para calcular números de fila.

Examples

from pyspark.sql import functions as sf
from pyspark.sql import Window
df = spark.range(3)
w = Window.orderBy(df.id.desc())
df.withColumn("desc_order", sf.row_number().over(w)).show()

+---+----------+
| id|desc_order|
+---+----------+
|  2|         1|
|  1|         2|
|  0|         3|
+---+----------+