第十章 Scala 容器基础(十六)：flatMap一个把flatten和map结合起来的方法

最新推荐文章于 2021-09-29 20:42:25 发布

weixin_33957648

最新推荐文章于 2021-09-29 20:42:25 发布

阅读量238

点赞数

文章标签： scala python

原文链接：https://my.oschina.net/nenusoul/blog/657946

版权

本文详细介绍了Scala中flatMap方法的工作原理及应用场景，通过一个具体例子展示了如何使用flatMap处理Option，以及flatMap与map、flatten之间的关系。文章还探讨了flatMap在数据处理中的优势，并提供了几个实用的代码示例。

摘要由CSDN通过智能技术生成

2019独角兽企业重金招聘Python工程师标准>>>

Problem

当你第一次来到Scala世界的时候，flatMap放方法看起来是很特别的，所以你需要理解他是如何工作的，还有它是应用在什么地方的。

Solution

在你调用map方法后紧接着调用flatten方法的情况下，你就可以尝试调用flatMap来解决这个问题了。满足如下情况：

使用map方法来从原集合基础上创建一个新的集合
map方法的返回结果是一个嵌套集合，或者元素为Some和None
你在map方法后紧接着调用了flatten方法

如果你的程序正好符合上面的情况，那么你就是可以使用flatMap来代替collection.map.flatten了

下面这个例子中，你会看到如何使用flatMap来处理一个Option。在这个例子中，我们要计算集合中数字类型元素的和。又一个问题：数字都是字符串类型的，并且好多元素并不是数字，也不能转换为Int型。先来看下这个集合：

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：

scala> def toInt(in: String): Option[Int] = {
     |   try {
     |     Some(Integer.parseInt(in.trim))
     |   } catch {
     |     case e: Exception => None
     |   }
     | }
toInt: (in: String)Option[Int]

下面我们来看看使用flatMap的效果：

scala> bag.flatMap(toInt).sum
res23: Int = 7

scala> bag.flatMap(toInt)
res24: List[Int] = List(1, 2, 4)

Discussion

为了看清楚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))

转载于:https://my.oschina.net/nenusoul/blog/657946