Aggregation and feature extraction at the same time using pyspark

Question

i have this dataset

+---------+------+------------------+--------------------+-------------+
|    LCLid|season|       sum(KWH/hh)|         avg(KWH/hh)|Acorn_g