pyspark dataframe get column type