Convert Spark cade to Python and functions

Related categories: Python / PySpark