spark 按某几列distinct去重、topN、窗口计算的应用

        在spark计算当中,我们日常有些简单需求,比如去重,比如有四列(appid、date、type、platform),我们只需要对appid,date进行去重,另外两列不关注。在spark中,distinct会对所有列执行去重操作,两行比较,只要有个一列有差异,就不算重复。

        要解决,只对某几列进行去重,可以使用Top N 的思路,也就是先分组,分组后,我们只取row_num=1第一行就可以了。

具体实现代码如下:

//选出某几列,去掉空值
Dataset dataset = input.select(
                Consts.Fields.APPID,
                Consts.Fields.EXT$ADID
        ).na().drop(dropNulCol);
        dataset.persist();
//设置窗口计算,可以指定多个列为partitionBy,其实就是多个groupby 分组列
        WindowSpec w=Window.partitionBy(Consts.Fields.APPID).orderBy(col("count").desc());
//只取row num=1,那就是top N,如果window里有orderby排序的话。
        Dataset top = count.withColumn("rn",row_number().over(w)).where(col("rn").$eq$eq$eq(1)).drop("rn");
        

 

你可能感兴趣的:(Spark)