python spark agg function sum code example Example: Pyspark Aggregation on multiple columns df.groupBy("year", "sex").agg(avg("percent"), count("*"))