pyspark dataframe column type