LTIMindtree interview question

Write a code in pyspark?