Scala中for循环的yield用法

ゝ一世哀愁。 2023-06-17 05:54 85阅读 0赞

\*\*

# 概念 #

\*\*  
for循环中的yield会将for循环中的值保存下来，保存到一个集合中，在循环结束的时候会将集合全部输出，如果被循环的是map，则输出的就是map。如果被循环的是list，则输出的就是list，以此类推。

## 使用println进行输出 ##

scala> for (i <- 1 to 5)
         | println (i)
    1
    2
    3
    4
    5

## 使用yield输出 ##

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

## 循环过滤if判断，并返回值 ##

scala> for (e <- a if e % 2 == 0 )
         | yield e
    res5: scala.collection.immutable.IndexedSeq[Int] = Vector(2, 4)
    scala> a
    res6: scala.collection.immutable.IndexedSeq[Int] = Vector(1, 2, 3, 4, 5)
    
    scala> val b = 6 to 7
    b: scala.collection.immutable.Range.Inclusive = Range(6, 7)
    
    scala> for {
         | x <- a
         | y <- b
         | }
         | yield (x,y)
    res7: scala.collection.immutable.IndexedSeq[(Int, Int)] = Vector((1,6), (1,7), (2,6), (2,7), (3,6), (3,7), (4,6), (4,7), (5,6), (5,7))
    
    scala> for {
         | y <- b
         | x <- a
         | }
         | yield (x,y)
    res8: scala.collection.immutable.IndexedSeq[(Int, Int)] = Vector((1,6), (2,6), (3,6), (4,6), (5,6), (1,7), (2,7), (3,7), (4,7), (5,7))

## yield复杂点的使用 ##

//找出以固定字符串结尾的文件
    scala> def getTextFile(path:String) : Array[java.io.File] =
         | for {
         | file <- new java.io.File(path).listFiles
         | if file.isFile
         | if file.getName.endsWith(".txt")
         | }
         | yield file
    getTextFile: (path: String)Array[java.io.File]
    
    scala> getTextFile(".")
    res9: Array[java.io.File] = Array(./a.txt)

## 实际开发中的应用 ##

object Loop {
      def main(args: Array[String]): Unit = {
        val str = Array(1,2,3,4,5,6,7)
    
        for(i <- str ) {
          println(i)
        }
    
        val add  = for (i <- str) yield i +100
        println (add.toBuffer)
    
      }
    }

## 这是输出结果 ##

1
    2
    3
    4
    5
    6
    7
    ArrayBuffer(101, 102, 103, 104, 105, 106, 107)
    
    Process finished with exit code 0