Capgemini interview question

Coding assessment on the pyspark in Azure databricks