pyspark dataframe get column types