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)
}
}
可以看见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)
}
}
上一篇: jdk8中map新增的merge方法
推荐阅读
-
浅谈Series和DataFrame中的sort_index方法
-
smarty模板引擎中内建函数if、elseif和else的使用方法
-
JSz中的静态方法和实例方法的分析
-
Android中判断listview是否滑动到顶部和底部的实现方法
-
Javascript中类式继承和原型式继承的实现方法和区别之处
-
SQLSERVER全文目录全文索引的使用方法和区别讲解
-
numpy中的ndarray方法和属性详解
-
晨祥seo心法-网站中的无效页面的危害和处理方法
-
python中读取文件的read、readline、readlines方法区别
-
[python] 在 python2和3中关于类继承的 super方法简要说明