как сгруппировать столбцы без функции агрегации в пандах?

Question

У меня есть следующий код, написанный в pyspark sql, который выглядит следующим образом:

dep_cus = sc_sql.sql("select cid, sum(trans_amnt) as run_bal, max(ds) as last_tran_date, mark_seg, province, local_body, branch, acc_type, int_rate from dep_sdf_temp group by cid, mark_seg, province, local_body, branch, acc_type, int_rate")

Как я могу написать этот код в пандах?

Answer 1

Это то, что вы ищете?

dep_cus = dep_sdf_temp.groupby(['cid', 'mark_seg', 'province', 'local_body', 'branch', 'acc_type', 'int_rate'])
                      .agg({'ds':'max', 'trans_amnt':'sum'})
                      .rename(columns={'ds':'last_tran_date', 'trans_amnt': 'run_bal'})