Scala中的map方法和flatMap方法的区别

map方法和flatMap方法都是根据我们传入的集合，取出里面的内容变成一个新的集合返回

但是他们的用法有些区别

map方法的返回值是B，也就是自定义的，你想返回啥都可以，可以是一个String,一个数值，一个集合...

flatMap方法返回值必须是GenTraversableOnce[B]类型，也就是集合类型，因为flatMap方法有一个扁平化操作，如果返回值不是集合类型，最终结果看起来会很奇怪，可以把返回值设置成一个字符串看看结果是什么样的

我写了一个map，里面的key是一个字符串，value是一个list,用这个例子可以看出他们到底有什么区别

map和flatMap方法里面传入是相同的函数，关键是看一下返回值我设置的都是map的value值，也就是返回这个list两者的区别

object mapANDfloatMap {
  def main(args: Array[String]): Unit = {
    val list1 = List[String]("v1", "vv1")
    val list2 = List[String]("v2", "vv2")
    println(list2.mkString(","))

    val map: Map[String, List[String]] = Map("k1" -> list1, "k2" -> list2)
    map.flatMap (kv => {
       kv._2
    }).foreach(println)
    
    map.map (kv => {
      kv._2
    }).foreach(println)
  }
}

  可以看见flatMap将list直接展开了，map返回了完整的list

flatMap就是用于这样在一个集合中还有一个小集合，展开里面的内容

map方法是返回了完整的小集合，不需要展开

词频统计的时候我们会经常用到flatMap,我拿出wordcount中的一部分来说

split切分每一行数据得到一个数组，flatMap将这个返回值展开了，

也就是说当最终遍历结束，我们得到的返回集合中，是一个一个的单词

    //首先获取文件的连接
    val source: BufferedSource = Source.fromFile("Scalalearning/data/words.txt")

    //获取到每一行数据
    val iterator: Iterator[String] = source.getLines()
        
    //通过flatMap展开
    iterator.flatMap(line => {
      line.split(",")
    }).foreach(println)

 

如果是用map,我们返回值是split切分出来的数组，返回值还是一个数组，我们最终得到的集合里是5个数组元素，这样达不到做词频统计的需求

object WordCountDemo {
  def main(args: Array[String]): Unit = {
    //首先获取文件的连接
    val source: BufferedSource = Source.fromFile("Scalalearning/data/words.txt")

    //获取到每一行数据
    val iterator = source.getLines()

    iterator.map(line => {
      line.split(",").mkString(",")
    }).foreach(println)
  }
}