Write Pyspark That Creates A Dataframe