Intro
map、flatten、flatMap是scala中常用的函数,举几个例子,简单的总结下。
Map
map操作是针对集合的变换操作,它将函数应用到集合中的每个元素,并产生一个结果集合。
import scala.collection.mutable.{ListBuffer, ArrayBuffer}
import scala.collection.mutable.{ListBuffer, ArrayBuffer}
val arr=Array(ArrayBuffer(1),ArrayBuffer(2),ArrayBuffer(3),ArrayBuffer(4),ArrayBuffer(5))
arr: Array[scala.collection.mutable.ArrayBuffer[Int]] = Array(ArrayBuffer(1), ArrayBuffer(2), ArrayBuffer(3), ArrayBuffer(4), ArrayBuffer(5))
生成一个ArrayBuffer的嵌套数组,里面的元素也是ArrayBuffer
通过map,给每个元素ArrayBuffer增加10
arr.map(x=>x+=(10)).foreach(println)
ArrayBuffer(1, 10)
ArrayBuffer(2, 10)
ArrayBuffer(3, 10)
ArrayBuffer(4, 10)
ArrayBuffer(5, 10)
flatten
flatten可以把嵌套的结构展开。比如上面的case中,把所有子元素ArrayBuffer展开,拼接成一个大的数组
val arr1 = arr.flatten
arr1: Array[Int] = Array(1, 10, 2, 10, 3, 10, 4, 10, 5, 10)
arr1.getClass.getSimpleName
res37: String = int[]
flatMap
如果想一步完成上面map和flatten的操作,咋整,flatMap可以做到
val arr=Array(ArrayBuffer(1),ArrayBuffer(2),ArrayBuffer(3),ArrayBuffer(4),ArrayBuffer(5))
arr: Array[scala.collection.mutable.ArrayBuffer[Int]] = Array(ArrayBuffer(1), ArrayBuffer(2), ArrayBuffer(3), ArrayBuffer(4), ArrayBuffer(5))
arr.flatMap(_+=(10))
res38: Array[Int] = Array(1, 10, 2, 10, 3, 10, 4, 10, 5, 10)
Ref
[1] http://dblab.xmu.edu.cn/blog/961-2/
2021-03-11 于南京市江宁区九龙湖