Scala中的map與collect

在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 }

效果完全相同。

我很少在項目中調用collect方法,且這個方法命名的意圖也不是特別明顯,至少在函數式編程的語境下,map似乎更為通用。今天在閱讀Neal Ford的Functional Thinking時,看到書中給出的這樣一個案例:

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.

case語句其實是偏函數定義的語法糖,當我們編寫一個case語句時,其實等同于創建了一個具有applyisDefineAt方法的偏函數對象。由于偏函數實質是函數的一種實例,因此可以將case語句傳遞給map方法,但它此時扮演的是一個普通的匿名函數,而非偏函數。因此,map方法在調用該函數對象的apply方法之前,并沒有調用isDefineAt方法判斷參數值是否定義。

我們可以對比map方法和collect方法的實現:

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
}

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)
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,030評論 6 531
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 98,310評論 3 415
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 175,951評論 0 373
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 62,796評論 1 309
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,566評論 6 407
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,055評論 1 322
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,142評論 3 440
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,303評論 0 288
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 48,799評論 1 333
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 40,683評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,899評論 1 369
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,409評論 5 358
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,135評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,520評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,757評論 1 282
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,528評論 3 390
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 47,844評論 2 372

推薦閱讀更多精彩內容