Scala111-map、flatten、flatMap

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 于南京市江宁区九龙湖