PySpark code from article https://towardsdatascience.com/a-brief-introduction-to-pyspark-ff4284701873