How to use aggregated data in a pyspark udf

Question

I am trying to calculate the speed of a vehicle using GPS records. I have original GPS points of and matched route for each trip in two separate data frames. The speed of th