Spark高級編程之TopN及分組TopN

1谈秫、獲取文本內(nèi)最大的前三個數(shù)字
輸入數(shù)據(jù):

3
54
44
2
67
32
133
54
23
1
35
23
73
32
16
78
21
56

1)Java版top3:

import org.apache.spark.SparkConf;
import org.apache.spark.api.java.JavaPairRDD;
import org.apache.spark.api.java.JavaRDD;
import org.apache.spark.api.java.JavaSparkContext;
import org.apache.spark.api.java.function.Function;
import org.apache.spark.api.java.function.PairFunction;
import scala.Tuple2;

import java.util.Arrays;
import java.util.List;

/**
 * 取最大的前3個數(shù)字
 */
public class Topn {
    public static void main(String[] args) {
        SparkConf conf = new SparkConf()
                .setAppName("Topn")
                .setMaster("local");
        JavaSparkContext sc = new JavaSparkContext(conf);
        JavaRDD<String> lines = sc.textFile("D:\\spark\\top3.txt");
        JavaPairRDD<Integer, String> pairs = lines.mapToPair(new PairFunction<String, Integer, String>() {
            @Override
            public Tuple2<Integer, String> call(String s) throws Exception {
                return new Tuple2<>(Integer.valueOf(s),s);
            }
        });
        JavaPairRDD<Integer, String> sortedPairs = pairs.sortByKey(false);
        JavaRDD<Integer> sortedNumbers = sortedPairs.map(new Function<Tuple2<Integer, String>, Integer>() {
            @Override
            public Integer call(Tuple2<Integer, String> v1) throws Exception {
                return v1._1;
            }
        });
        List<Integer> list = sortedNumbers.take(3);
        System.out.println(Arrays.toString(list.toArray()));

        sc.close();

    }
}

輸出結(jié)果:

[133, 78, 73]

2)Scala版top3

import org.apache.spark.{SparkConf, SparkContext}

object Topn {

  def main(args: Array[String]): Unit = {
    val inputPath = "D:\\spark\\top3.txt"
    val  conf = new SparkConf()
      .setAppName("Topn")
      .setMaster("local")
    val sc = new SparkContext(conf)
    val lines =sc.textFile(inputPath,1)
    val pairs = lines.map(line =>(line.toInt,line))
    val sortedPairs = pairs.sortByKey(false)
    val sortedNumbers = sortedPairs.map(_._1)
    val topNumber = sortedNumbers.take(3)
    for (num <- topNumber){
      println(num)
    }
    sc.stop()
  }

}

輸出結(jié)果:

133
78
73

2、對每個班級的學(xué)生成績判帮,取出前3名(分組取topn)
輸入數(shù)據(jù):

class1 90
class2 56
class1 87
class1 76
class2 88
class1 95
class1 74
class2 87
class2 67
class2 77

1)Java版分組取topn

/**
 * 分組取Top3
 */
public class GroupTop3 {
    public static void main(String[] args) {
        SparkConf conf = new SparkConf()
                .setAppName("Topn")
                .setMaster("local");
        JavaSparkContext sc = new JavaSparkContext(conf);
        JavaRDD<String> lines = sc.textFile("D:\\spark\\grouptop3.txt");
        JavaPairRDD<String, Integer> pairs = lines.mapToPair(new PairFunction<String, String, Integer>() {
            @Override
            public Tuple2<String, Integer> call(String s) throws Exception {
                String[] split = s.split(" ");
                return new Tuple2<>(split[0], Integer.valueOf(split[1]));
            }
        });

        JavaPairRDD<String, Iterable<Integer>> groupedPairs = pairs.groupByKey();
        JavaPairRDD<String, Iterable<Integer>> top3Scores = groupedPairs.mapToPair(new PairFunction<Tuple2<String, Iterable<Integer>>, String, Iterable<Integer>>() {
            @Override
            public Tuple2<String, Iterable<Integer>> call(Tuple2<String, Iterable<Integer>> classScores) throws Exception {
                Integer[] top3 = new Integer[3];
                String className = classScores._1;
                Iterator<Integer> scores = classScores._2.iterator();
                while (scores.hasNext()){
                    Integer score = scores.next();
                    for(int i=0;i<3;i++){
                        if(top3[i]==null){
                            top3[i] = score;
                            break;
                        }else if(score>top3[i]){
                            for(int j=top3.length-1;j>i;j--){
                                top3[j]=top3[j-1];
                            }
                            top3[i] = score;
                            break;
                        }
                    }
                }
                return new Tuple2<String, Iterable<Integer>>(className,Arrays.asList(top3));
            }
        });

        top3Scores.foreach(new VoidFunction<Tuple2<String, Iterable<Integer>>>() {
            @Override
            public void call(Tuple2<String, Iterable<Integer>> top3Score) throws Exception {
                System.out.println("class:"+top3Score._1);
                Iterator<Integer> integers = top3Score._2.iterator();
                while (integers.hasNext()){
                    System.out.println(integers.next());
                }
                System.out.println("=================================");

            }
        });
        sc.close();
    }
}

輸出結(jié)果:

class:class1
95
90
87
=================================
class:class2
88
87
77
=================================

2)scala版分組取topn

object GroupTop3 {

  def main(args: Array[String]): Unit = {
    val inputPath = "D:\\spark\\grouptop3.txt"
    val  conf = new SparkConf()
      .setAppName("GroupTop3")
      .setMaster("local")
    val sc = new SparkContext(conf)
    val lines =sc.textFile(inputPath,1)
    val pairs = lines.map(line =>(line.split(" ")(0),line.split(" ")(1).toInt))
    val groupedPairs = pairs.groupByKey()
    val groupscores = groupedPairs.map(classScore =>{
      val className = classScore._1;
      val scores = classScore._2.iterator
      val top3 = new Array[Integer](3)
      var flag = true
      for (score <- scores){
        var flag = true
        for (i <- 0 until top3.length if flag){
          if(top3(i)==null){
            top3(i) = score
            flag = false
          }else if(score>top3(i)){
            for(j <- top3.length - 1 to i+1 by -1){
              top3(j) = top3(j-1);
            }
            top3(i) = score;
            flag = false
          }
        }
      }
      (className,top3)
    })

    for(groupscore<-groupscores){
      println("class:"+groupscore._1)
      val scores = groupscore._2
      for(score<- scores){
        println(score)
      }
      println("=============================")
    }

    sc.stop()
  }

}

輸出結(jié)果:

class:class1
95
90
87
=============================
class:class2
88
87
77
=============================

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市献丑,隨后出現(xiàn)的幾起案子冗酿,更是在濱河造成了極大的恐慌,老刑警劉巖勾扭,帶你破解...
    沈念sama閱讀 211,948評論 6 492
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件矫钓,死亡現(xiàn)場離奇詭異要尔,居然都是意外死亡舍杜,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,371評論 3 385
  • 文/潘曉璐 我一進店門赵辕,熙熙樓的掌柜王于貴愁眉苦臉地迎上來既绩,“玉大人,你說我怎么就攤上這事还惠∷俏眨” “怎么了?”我有些...
    開封第一講書人閱讀 157,490評論 0 348
  • 文/不壞的土叔 我叫張陵蚕键,是天一觀的道長救欧。 經(jīng)常有香客問我,道長锣光,這世上最難降的妖魔是什么笆怠? 我笑而不...
    開封第一講書人閱讀 56,521評論 1 284
  • 正文 為了忘掉前任,我火速辦了婚禮誊爹,結(jié)果婚禮上骑疆,老公的妹妹穿的比我還像新娘。我一直安慰自己替废,他們只是感情好,可當(dāng)我...
    茶點故事閱讀 65,627評論 6 386
  • 文/花漫 我一把揭開白布泊柬。 她就那樣靜靜地躺著椎镣,像睡著了一般。 火紅的嫁衣襯著肌膚如雪兽赁。 梳的紋絲不亂的頭發(fā)上状答,一...
    開封第一講書人閱讀 49,842評論 1 290
  • 那天,我揣著相機與錄音刀崖,去河邊找鬼惊科。 笑死,一個胖子當(dāng)著我的面吹牛亮钦,可吹牛的內(nèi)容都是我干的馆截。 我是一名探鬼主播,決...
    沈念sama閱讀 38,997評論 3 408
  • 文/蒼蘭香墨 我猛地睜開眼蜂莉,長吁一口氣:“原來是場噩夢啊……” “哼蜡娶!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起映穗,我...
    開封第一講書人閱讀 37,741評論 0 268
  • 序言:老撾萬榮一對情侶失蹤窖张,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后蚁滋,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體宿接,經(jīng)...
    沈念sama閱讀 44,203評論 1 303
  • 正文 獨居荒郊野嶺守林人離奇死亡赘淮,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 36,534評論 2 327
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了睦霎。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片梢卸。...
    茶點故事閱讀 38,673評論 1 341
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖碎赢,靈堂內(nèi)的尸體忽然破棺而出低剔,到底是詐尸還是另有隱情,我是刑警寧澤肮塞,帶...
    沈念sama閱讀 34,339評論 4 330
  • 正文 年R本政府宣布襟齿,位于F島的核電站,受9級特大地震影響枕赵,放射性物質(zhì)發(fā)生泄漏猜欺。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 39,955評論 3 313
  • 文/蒙蒙 一拷窜、第九天 我趴在偏房一處隱蔽的房頂上張望开皿。 院中可真熱鬧,春花似錦篮昧、人聲如沸赋荆。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,770評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽窄潭。三九已至,卻和暖如春酵颁,著一層夾襖步出監(jiān)牢的瞬間嫉你,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 32,000評論 1 266
  • 我被黑心中介騙來泰國打工躏惋, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留幽污,地道東北人。 一個月前我還...
    沈念sama閱讀 46,394評論 2 360
  • 正文 我出身青樓簿姨,卻偏偏與公主長得像距误,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子款熬,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 43,562評論 2 349

推薦閱讀更多精彩內(nèi)容