当前位置：首页 > article >正文

scala之全文单词统计

article 2024/12/3 2:46:35

1.读取文件，得到很长的字符串

2.把字符串拆分成一个一个的单词

3.统计每个单词出现的次数

4.排序

5.把结果保存到一个新文件中

完整代码

import java.io.PrintWriter
import scala.io.Source
import scala.reflect.api.Printers

object test2 {
  def main(args: Array[String]): Unit = {


    //1.读入文件内容
    val content = Source.fromFile("1.txt").mkString
    println(content)

    //2.把字符串中的每个单词拆出，并保存到数组
    //W:表示一个非字（不是一个字 eg:空格，逗号，句号……）
    //W+:多个非字
    val arr = content.split("\\W+")


    //统计
    val m1 = scala.collection.mutable.Map[String, Int]()
    arr.foreach(word => {
      //检查是否出现过
      if (m1.contains(word)) {
        //把票数
        m1(word) += 1
      } else {
        m1(word) = 1
      }
    })

    //对于m1,他是一个Map,(thank:10,is:5),现在需要对它进行排序，把出现最多次数的放在最前面，然后再输出
    //只要排序之前的结果的前15个？在list中取出前15个？
    val sortedM1 = m1.toList.sortWith((a, b) => a._2 > b._2).filter(a => a._1.length > 2).slice(0, 15)
    //输出排序后的Map
    sortedM1.foreach { case (word, cishu) => println(s"${word}:${cishu}") }

    //把结果保存到一个新文件中:
    val writer = new PrintWriter("2.txt")
    writer.println("统计的结果是:")
    //输出排序后的Map
    //把结果保存到一个新的文件中
    sortedM1.foreach { case (word, cishu) => writer.println(s"${word}:${cishu}") }
    writer.close()
  }

}

运行结果