Row_number

Funkcja okna: zwraca sekwencyjną liczbę rozpoczynającą się od 1 w partycji okna.

Składnia

from pyspark.sql import functions as sf

sf.row_number()

Parametry

Ta funkcja nie pobiera żadnych parametrów.

Zwraca

pyspark.sql.Column: kolumna do obliczania liczb wierszy.

Przykłady

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|
+---+----------+