PySpark Koans

Learn by fixing tests

Progress0/39

Spark Koans is an independent community tool.

Window FunctionsKoan 24

Window Functions - Row Number

Assign sequential row numbers within groups. Replace ___ with the correct code.

How it works: Replace the ___ blanks in the code editor with the correct PySpark code, then hit Run Code. Stuck? Try the Hint button.

Setup (read-only)

from pyspark.sql.window import Window
from pyspark.sql.functions import row_number, col

data = [
    ("Sales", "Alice", 5000),
    ("Sales", "Bob", 5500),
    ("Engineering", "Charlie", 6000),
    ("Engineering", "Diana", 6500),
    ("Engineering", "Eve", 5500)
]
df = spark.createDataFrame(data, ["dept", "name", "salary"])

Your CodeCtrl/Cmd+Enter to run

Output

Output will appear here...