PySpark Koans

Learn by fixing tests

Progress0/39

Spark Koans is an independent community tool.

BasicsKoan 2

Selecting Columns

Learn how to select specific columns from a DataFrame. 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)

data = [("Alice", 34, "NYC"), ("Bob", 45, "LA"), ("Charlie", 29, "Chicago")]
df = spark.createDataFrame(data, ["name", "age", "city"])

Your CodeCtrl/Cmd+Enter to run

Output

Output will appear here...