Count Elements In List Pyspark