LTIMindtree Interview Question

Write a code in pyspark?