pyspark create dataframe with two columns