欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-06-08 22:19:22
...

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)
  }
}

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

  可以看见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)

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

 

如果是用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)
  }
}

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