scala 两种方法实现单词计数

 val lines = List("hello world", "hello spark")
    val wordlist = lines.flatMap(line => line.split(" ")).map(word => (word, 1))
    //方法一:先groupBy再map
    wordlist.groupBy(_._1).map {
      case (word, list) => (word, list.size)
    }.foreach(println)

    //方法二:通过aggregate来实现map reduce,效率更高
    val seqop = (result: mutable.HashMap[String, Int], wordcount: (String, Int)) => {
      val addOne = (wordcount._1, result.getOrElse(wordcount._1, 0) + wordcount._2)
      result.+=(addOne)
    }
    val combop = (result1: mutable.HashMap[String, Int], result2: mutable.HashMap[String, Int]) => {
      result1 ++= result2
    }
    val result = wordlist.aggregate(mutable.HashMap[String, Int]())(seqop,combop)
    println(result)

从文件读取进行word count:

   val lines = Source.fromFile("test.txt").getLines()
    val seqop = (result: mutable.HashMap[String, Int], line: String) => {
      val wordcount = line.replace(",", " ").replace(".", " ").replace("(", " ").replace(")", " ").split(" ").filter(_.trim.length > 0).map(word => (word, 1))
      wordcount.foreach(wc => {
        val addOne = (wc._1, result.getOrElse(wc._1, 0) + wc._2)
        result += addOne
      })
      result
    }
    val combop = (result1: mutable.HashMap[String, Int], result2: mutable.HashMap[String, Int]) => {
      result1 ++= result2
    }

    val test = lines.aggregate(mutable.HashMap[String, Int]())(seqop, combop)
    println(test)

lines其实是一个迭代器,流式一行一行读取,不要toList,否则读取大文件时可能出现内存溢出问题

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值