pyspark dataframe memory usage