LTM Limited interview question

Write a code in pyspark?