grouped = data_file.groupBy(lambda r: r[0])
group_list = grouped.map(lambda x: list(x[1]))
max_list = group_list.map(reduce_by_max).collect()
pyspark: groupby, а затем получить максимальное значение каждой группы