基于ID分组的条件聚合：数值列均值计算与非数值列保留规则

基础数据场景 #

原始数据：

ID           VAR       HSP     VAR2      CODE
00001           56        B       345       1       
00001           52        B       306       1       
00002           5         C       32        2       
00002           22        A       31        2       
00003           55        D       23        1       
00003           5         E       87        1       

处理需求 #

• 按ID分组，对重复ID的组执行以下操作：
  • 数值列（VAR、VAR2）计算均值，CODE不参与均值计算
  • HSP保留分组内排序后出现的第一个值
  • CODE保留分组内的第一个值

期望输出 #

ID           VAR       HSP     VAR2      CODE
00001           54        B       325.5     1       
00002           13.5      C       31.5      2       
00003           30        D       55        1       

含0值的补充场景 #

原始数据：

ID           VAR       HSP     VAR2      CODE
00001           56        B       345       1       
00001           0         B       306       1       
00002           5         C       32        2       
00002           22        A       0         2       
00003           55        D       23        1       
00003           5         E       87        1       

期望输出 #

ID           VAR       HSP     VAR2      CODE
00001           56        B       325.5     1       
00002           13.5      C       32        2       
00003           30        D       55        1       

补充规则 #

数值列中的0值不参与均值计算，仅对非0数值求平均。

内容的提问来源于stack exchange，提问作者NewUsr