Create new column with max value based on filtered rows with groupby in pyspark

Question

I have a spark dataframe

import pandas as pd
foo = pd.DataFrame({\'id\': [1,1,2,2,2], \'col\': [\'a\',\'b\',\'a\',\'a\',\'b\'], \'value\': [1,5,2