Red Ventures interview question

Databricks Assignment - Spark (Scala and Python)