Scala实现WordCount

Question

我听说WordCount用scala不是只需要两行吗,为什么我…

import java.io.FileWriter
import scala.collection.mutable
object MyWCApp extends App{
	val contents=
		"""hello,world,hello,world,count,world
		  |xixixi,hehehe,hahaha,world,count,count
		  |hello""".stripMargin
	//val contents =scala.io.Source.fromFile("input/hell.txt").mkString
	//println(contents)
	val tmp = mutable.ListBuffer[(String,Int)]()
	for(line<-contents.split("\r\n")){
		tmp++=List(line.trim).flatMap(_.split(",")).map(x=>(x,1))
	}
	val mp = mutable.Map[String,Int]()
	for((k,v)<-tmp){
		if (mp.getOrElse(k, -1) != -1) {
			mp(k) = v + mp(k)
		} else {
			mp(k) = 1
		}
	}
	val b= new FileWriter("output/ans.txt")
	for ((k,v)<-mp){
		b.write(s"$k\t$v")
		b.write(10)
	}
	b.close()
}

Answer 1

Michael_PK 2019-05-26 18:31:00

你这写的什么鬼。。。这么复杂

• 提问者 Witness2017 #1

  我也觉得太麻烦了，，该从哪里改它呢。。

• Michael_PK 回复 提问者 Witness2017 #2

  这种写法百度一搜一堆的呢，自己搜下

• 提问者 Witness2017 #3

  非常感谢！

Answer 2

提问者 Witness2017 2019-05-26 23:03:17

import java.io.FileWriter
import scala.collection.mutable
object MyWCApp extends App{
  val contents=
     """hello,world,hello,world,count,world
       |xixixi,hehehe,hahaha,world,count,count
       |hello""".stripMargin
  //val contents =scala.io.Source.fromFile("input/hell.txt").mkString
  val a = List(contents).flatMap(_.split("\r\n")).flatMap(_.split(",")).map((_,1))
        .groupBy(_._1).map(t=>(t._1,t._2.map(_._2).reduce(_+_)))//.foreach(println)
  val fd = new FileWriter("output/ans.txt")
  println(a)
  a.foreach(t=>fd.write(s"${t._1}\t${t._2}\n"))
  fd.flush()
  fd.close()

最终版!