Display Function In Pyspark