pyspark dataframe column types