Big pyspark dataframe with longitude and latitude columns,how to get a new column with State name based on lon and lat columns quickly?

Question

I have a pyspark dataframe with longitude and latitude columns,how to get a new column with the State name based on longitude and latitude columns, I use UDF and reverse_geocode