Scala collect与map区别

首先讲一下map的作用：

map() 方法返回一个新数组，数组中的元素为原始数组元素调用函数处理后的值。
map() 方法按照原始数组元素顺序依次处理元素。
注意：1. map() 不会对空数组进行检测； 2.map() 不会改变原始数组。

在Scala中，当我需要对集合的元素进行转换时，自然而然会使用到map方法。而当我们在对Tuple类型的集合或者针对Map进行map操作时，通常更倾向于在map方法中使用case语句，这比直接使用_1与_2更加可读。例如：

val languageToCount = Map("Scala" -> 10, "Java" -> 20, "Ruby" -> 5)
languageToCount map { case (_, count) => count + 1 }

然而对于上述场景，其实我们也可以使用collect方法：

languageToCount collect { case (_, count) => count + 1 }

得到的结果都是相同的。

---

在函数式编程的语境下，map似乎更为通用，那什么时候要用到collect呢

那就是涉及到偏函数的时候。
我们来看一个例子：

List(1, 3, 5, "seven") map { case i: Int => i + 1 } //won't work
//scala.MatchError: seven (of class java.lang.String)
List(1, 3, 5, "seven") collect { case i: Int => i + 1 } //it works

为什么同样的case语句，放在collect中是正确的，放在map中就会抛出MatchError错误呢？查阅Scala的API文档，发现这个两个函数的定义存在本质上的区别：

def map[B](f: (A) ⇒ B): List[B]
def collect[B](pf: PartialFunction[A, B]): List[B]

两个方法的定义如出一辙，区别就在于前者接收的是一个函数类型的参数，而后者接收的是一个偏函数（partial function）类型的参数：

map: Builds a new collection by applying a function to all elements of this list.

colect: Builds a new collection by applying a partial function to all elements of this list on which the function is defined.

我们可以对比map方法和collect方法的实现：

//map接收的参数是函数类型
def map[B, That](f: A => B)(implicit bf: CanBuildFrom[Repr, B, That]): That = {
  def builder = {
      val b = bf(repr)
      b.sizeHint(this)
      b
  }
  val b = builder  
  for (x <- this) b += f(x)  
  b.result
}

collect接收的参数是偏函数类型
def collect[B, That](pf: PartialFunction[A, B])(implicit bf: CanBuildFrom[Repr, B, That]): That = {
   val b = bf(repr)
   //先过滤，再调用（把不符合偏函数中定义的类型的元素过滤掉）
   for (x <- this) if (pf.isDefinedAt(x)) b += pf(x)  
   b.result
}

在调用map方法时，一旦遍历到值"seven"，并调用f(x)，因为类型不符合模式匹配中的Int类型，导致抛出MatchError错误。
而collect方法在调用pf(x)之前，调用了pf的isDefinedAt(x)作了一次过滤。

---

如果在前面的map例子中再增加一个case子句，对String类型的值进行处理，则case语句就从偏函数变成了满足所有条件的“全”函数：

List(1, 3, 5, "seven") map {
  case i: Int => i + 1
  case s: String => s.length
}

得到的结果为：

List[Int] = List(2, 4, 6, 5)

由于collect方法接收的是一个偏函数类型，所以它并不能接收一个lambda表达式（匿名函数）：

List(1, 3, 5, "seven").collect(i => i + 1)

会抛出：

error: missing parameter type

另外

我们在使用collect时，可以利用偏函数的原理，同时实现filter与map的特性。例如：

List(1, 2, 3, 4, 5, 6) collect { case i if i % 2 == 0 => i * i }

这段代码相当于：

List(1, 2, 3, 4, 5, 6).filter(i => i % 2 == 0).map(i => i * i)

偏函数的小结：

参考：https://www.cnblogs.com/sjj1022/p/6946836.html
Scala学习文档： https://pan.baidu.com/s/1BO5z6uaPE2yZfVmZ0PMohw
提取码: s9xy