Lat Long to UTM coversion on spark data frame using pyspark

Question

I have a spark dataframe with Lat/long field. I have a function to do the UTM conversion in python. I am trying to register the function in spark and then use it in my spark