PySpark Koans

Learn by fixing tests

Progress0/39

Spark Koans is an independent community tool.

Window FunctionsKoan 23

Window Functions - Running Total

Use window functions to calculate running totals. 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 sum as spark_sum, col

data = [
    ("2024-01-01", 100),
    ("2024-01-02", 150),
    ("2024-01-03", 200),
    ("2024-01-04", 175)
]
df = spark.createDataFrame(data, ["date", "sales"])

Your CodeCtrl/Cmd+Enter to run

Output

Output will appear here...