max_by_group = list(data_file.map(lambda x: (x[0], x)).reduceByKey(lambda x1, x2:
    max(x1, x2, key=lambda x: x[-1])).values())
max_by_group.collect()
pyspark: groupby, а затем получить максимальное значение каждой группы