scala的for循环和yield

最新推荐文章于 2025-03-03 10:39:30 发布

转载最新推荐文章于 2025-03-03 10:39:30 发布 · 925 阅读

scala 专栏收录该内容

2 篇文章

订阅专栏

本文详细介绍了Scala中for循环结合yield关键字的使用方法。通过多个示例展示了如何利用yield来收集循环过程中的值，并返回特定类型的集合。此外，还介绍了如何在for循环中加入条件判断（守卫）以过滤结果。

转载自https://www.cnblogs.com/sunfie/p/4983841.html

for循环中的 yield 会把当前的元素记下来，保存在集合中，循环结束后将返回该集合。Scala中for循环是有返回值的。如果被循环的是Map，返回的就是Map，被循环的是List，返回的就是List，以此类推。

例1：

1 scala> for (i <- 1 to 5) yield i
2 res10: scala.collection.immutable.IndexedSeq[Int] = Vector(1, 2, 3, 4, 5)

例2：

1 scala> for (i <- 1 to 5) yield i * 2
2 res11: scala.collection.immutable.IndexedSeq[Int] = Vector(2, 4, 6, 8, 10)

例3： for/yield 循环的求模操作:

1 scala> for (i <- 1 to 5) yield i % 2
2 res12: scala.collection.immutable.IndexedSeq[Int] = Vector(1, 0, 1, 0, 1)

例4：Scala 数组上的 for 循环 yield 的例子

 1 scala> val a = Array(1, 2, 3, 4, 5)
 2 a: Array[Int] = Array(1, 2, 3, 4, 5)
 3  
 4 scala> for (e <- a) yield e
 5 res5: Array[Int] = Array(1, 2, 3, 4, 5)
 6  
 7 scala> for (e <- a) yield e * 2
 8 res6: Array[Int] = Array(2, 4, 6, 8, 10)
 9  
10 scala> for (e <- a) yield e % 2
11 res7: Array[Int] = Array(1, 0, 1, 0, 1)

例5：for 循环, yield, 和守卫( guards) (for loop 'if' conditions)

假如你熟悉了 Scala 复杂的语法, 你就会知道可以在 for 循环结构中加上 'if' 表达式. 它们作为测试用，通常被认为是一个守卫，你可以把它们与 yield 语法联合起来用。参见：:

1 scala> val a = Array(1, 2, 3, 4, 5)
2 a: Array[Int] = Array(1, 2, 3, 4, 5)
3  
4 scala> for (e <- a if e > 2) yield e
5 res1: Array[Int] = Array(3, 4, 5)

加上了 "if e > 2" 作为守卫条件用以限制得到了只包含了三个元素的数组.

例6：Scala for 循环和 yield 的例子 - 总结

如果你熟悉 Scala 的 loop 结构, 就会知道在 for 后的圆括号中还可以许更多的事情. 你可以加入 "if" 表达式，或别的语句, 比如下面的例子，可以组合多个 if 语句:

1 def scalaFiles =
2   for {
3     file <- filesHere
4     if file.isFile
5     if file.getName.endsWith(".scala")
6   } yield file

yield 关键字的简短总结:

针对每一次 for 循环的迭代, yield 会产生一个值，被循环记录下来 (内部实现上，像是一个缓冲区).
当循环结束后, 会返回所有 yield 的值组成的集合.

返回集合的类型与被遍历的集合类型是一致的.