Pergunta de entrevista da empresa Modak

I was asked to write pyspark code. And was tested on databricks as well