Feb-05-2022, 03:12 AM
something like:
def create_group(col1x, coly): return HAVE.groupBy(colx, coly), \ .agg(F.sum('col5').alias('col5'), \ F.sum('col6').alias('col6'), \ F.sum('col7').alias('col7'), \ F.sum('col8').alias('col8')) \ .sort(colx, coly) def groups(): NEED1 = create_group(col1, col2) NEED2 = create_group(col3, col4)