当你第一次来到Scala世界的时候,flatMap放方法看起来是很特别的,因此你须要理解他是如何工做的,还有它是应用在什么地方的。
函数
在你调用map方法后紧接着调用flatten方法的状况下,你就能够尝试调用flatMap来解决这个问题了。知足以下状况:scala
使用map方法来从原集合基础上建立一个新的集合code
map方法的返回结果是一个嵌套集合,或者元素为Some和Nonethree
你在map方法后紧接着调用了flatten方法字符串
若是你的程序正好符合上面的状况,那么你就是可使用flatMap来代替collection.map.flatten了
it
下面这个例子中,你会看到如何使用flatMap来处理一个Option。在这个例子中,咱们要计算集合中数字类型元素的和。又一个问题:数字都是字符串类型的,而且好多元素并非数字,也不能转换为Int型。先来看下这个集合:io
scala> val bag = List("1", "2", "three", "4", "one hundred seventy five") bag: List[String] = List(1, 2, three, 4, one hundred seventy five)
为了解决这个问题,你开始建立一个字符串到整形的转换函数来返回Some[Int]或者None:event
scala> def toInt(in: String): Option[Int] = { | try { | Some(Integer.parseInt(in.trim)) | } catch { | case e: Exception => None | } | } toInt: (in: String)Option[Int]
下面咱们来看看使用flatMap的效果:class
scala> bag.flatMap(toInt).sum res23: Int = 7 scala> bag.flatMap(toInt) res24: List[Int] = List(1, 2, 4)
为了看清楚flatMap是如何工做的,咱们吧这个问题拆分红几小步。首先,你要知道当你调用map方法会发生什么:基础
scala> bag.map(toInt) res26: List[Option[Int]] = List(Some(1), Some(2), None, Some(4), None)
map方法使toInt函数做用在了集合的每一个元素上,而且返回了一组Some[Int]和None的集合。可是sum方法要做用在List[Int]上,怎么办?上节踢桃flatten方法能够把集合中Some里的数据展开,而后抛弃None。
scala> bag.map(toInt).flatten res27: List[Int] = List(1, 2, 4)
此次咱们就可以调用sum方法了:
scala> bag.map(toInt).flatten.sum res28: Int = 7
哇,我看到了map.flatten的结构,我想到了“flat map”,因此我就可使用flatMap方法了:
scala> bag.flatMap(toInt).sum res23: Int = 7
一旦你得道了一个原始集合,而且想把它转换为List[Int]。你能够调用丰富的集合方法来获取你想要的结果:
scala> bag.flatMap(toInt).filter(_ > 1) res29: List[Int] = List(2, 4) scala> bag.flatMap(toInt).takeWhile(_ < 4) res30: List[Int] = List(1, 2) scala> bag.flatMap(toInt).partition(_ > 3) res31: (List[Int], List[Int]) = (List(4),List(1, 2))