How to apply a custom function to grouped data in PySpark

Question

I am workig with PySpark and have a dataframe looking like this example below:

I want to group by req and apply a function on each group by. My function looks som