Scala基础应用(7)- map/flatMap和filter

时间:2023-02-04 23:13:07

Scala基础应用(7)- map/flatMap和filter

本章主要介绍集合的map, flatten, flatMap, filter zip, fold, reduce等常用方法

map

map接受一个函数作为参数, 而此函数会接受一个参数为集合中元素, 所以此函数用于对集合中每一个元素进行处理后,产出其结果的集合。

val test = List("a", "b", "c")
test.map(x => x.toUpperCase)
或者 test.map(_.toUpperCase)

flatten

flatten 是将嵌套集合展开一层,注意Any类型的集合是不是展开的,但集合类型是可以展开的,即使是Any类型的集合类型。

List(List(List(1,2),List(3,4)), List(5,6)).flatten 可以展开一层,元法再次调用flatten进行再次展开,因为List(5,6)在第一次flatten时,已经展开,这个List会变成含有List和常数的List[Any]类型,下面可以进行再次展开

List(List(List(1,2),List(3,4)), List(List(5,6), List(7,8))).flatten.flatten // 可以展开第二层

flatMap

flatMap实际上是对集合进行map, 然后将map函数处理的结果再flatten

List(List("a", "b"), List("c", "d")).flatMap (x => x)

List(List("a", "b"), List("c", "d")).flatten 返回的结果一样

只是前面一个式子中,x 参数是一个List集合

List(List(1,2),List(3,4)).flatMap(x=>x.map(x=>x*2))
通过上面这个式子,也可以看出x是集合,它有map功能,同时我们也能看出map返回的也是集合,而结果是展开的,所以可以理解为flatMap在执行完函数后,执行了flatten

filter

filter通过函数参数过滤掉函数返回false的元素

List("a", "b", "c").filter(x => x == "a")

zip

zip将两个集合进行合并,最终结果为较短集合的长度

val list1 = List("a", "b", "c", "d")
val list2 = List(1, 2, 3)
list1.zip(list2) 或者 list2.zip(list1)的长度都是3

可用zipAll来处理长度不同的两个集合,那如何处理较短的集合的填充呢?

list1.zipAll(list2, "填充1", "填充2")
如果list1较短,则用"填充1"填充,如果list2较短,则用"填充2"进行填充
上例将以"填充2"填充, 因为list2较短

fold

fold将集合元素进行处理,然后产生你想要的结果类型。

  • 样式

    集合变量.fold(初始值) { (累集结果变量,元素) => 处理函数体}

    如:
    List(1,2,3,4,5).fold(0) { (sum, item) => sum + item }
    返回15
    List(1,2,3,4,5).fold("a") { (sum, item) => sum + item.toString }
    返回 a12345

    注意:如果初始值类型和元素类型一至,返回类型也是该类型,否则为Any类型
  • foldLeft和foldRight分别从左开始fold和从右往左开始fold

    class Foo(val name: String, val age: Int, val genda: Symbol)
    object Foo {
    def apply(name: String, age: Int, genda: Symbol) = new Foo(name, age, genda)
    }

    val fooList = Foo("张三", 15, '男) :: Foo("李四", 20, '男) :: Foo("阿花", 18, '女) :: Nil

    val lst = fooList.foldLeft(List[String]()) {
    (z, f) =>
    val title = f.genda match {
    case '男 => "Mr."
    case '女 => "Ms."
    }
    println(z)
    z :+ s"$title ${f.name}, ${f.age}"
    }

    注意: foldLeft和foldRight产生的结果与初始值的类型相同

    另外:
    foldLeft的简写为 /:
    foldRight的简写为 :\

    (0/:(1 to 100)) (_+_)
    ((1 to 100) :\0) (_+_)

reduce

可以将reduce认为是fold的一种特殊情况,reduce的返回值类型必须与元素的类型相同

val list1 = List(1,2,3)
list1.reduce((sum,item) => sum + item)