我该怎么办“if..else”在 for 理解中？

Question

我问一个非常基本的问题，最近让我很困惑。 我想编写一个 Scala For 表达式来执行如下操作：

for (i <- expr1) {
  if (i.method) {
    for (j <- i) {
      if (j.method) {
        doSomething()
      } else {
        doSomethingElseA()
      }
    }
  } else {
    doSomethingElseB()
  }
}

问题是，在多个 For 表达式生成器中，我不知道可以将每个 for 表达式主体放在哪里。

for {i <- expr1
  if(i.method) // where can I write the else logic ?
  j <- i 
  if (j.method)
} doSomething()

如何以 Scala 风格重写代码？

Answer 1

您编写的第一个代码完全有效，因此无需重写它。在其他地方，您说过您想知道如何以 Scala 风格进行操作。并没有真正的“Scala 风格”，但我会假设一种更实用的风格并解决它。

for (i <- expr1) {
  if (i.method) {
    for (j <- i) {
      if (j.method) {
        doSomething()
      } else {
        doSomethingElseA()
      }
    }
  } else {
    doSomethingElseB()
  }
}

第一个担心是这不会返回任何值。它所做的只是副作用，也是应该避免的。所以第一个改变是这样的：

val result = for (i <- expr1) yield {
  if (i.method) {
    for (j <- i) yield {
      if (j.method) {
        returnSomething()
        // etc

之间有很大的区别

for (i <- expr1; j <- i) yield ...

现在，和

for (i <- expr1) yield for (j <- i) yield ...

，它们返回不同的东西，有时你想要后者，而不是前者。不过，我假设你想要前者。现在，在继续之前，让我们修复代码。它很丑陋、难以理解并且信息量不大。让我们通过提取方法来重构它。

def resultOrB(j) = if (j.method) returnSomething else returnSomethingElseB
def nonCResults(i) = for (j <- i) yield resultOrB(j)
def resultOrC(i) = if (i.method) nonCResults(i) else returnSomethingC
val result = for (i <- expr1) yield resultOrC(i)

它已经干净多了，但它并没有达到我们的预期。让我们看看区别：

trait Element
object Unrecognized extends Element
case class Letter(c: Char) extends Element
case class Punct(c: Char) extends Element
val expr1 = "This is a silly example." split "\\b"

def wordOrPunct(j: Char) = if (j.isLetter) Letter(j.toLower) else Punct(j)
def validElements(i: String) = for (j <- i) yield wordOrPunct(j)
def classifyElements(i: String) = if (i.nonEmpty) validElements(i) else Unrecognized
val result = for (i <- expr1) yield classifyElements(i)

result 的类型是 Array[AnyRef]，而使用多个生成器将产生 Array[Element]。修复的简单部分是这样的：

val result = for {
  i <- expr1
  element <- classifyElements(i)
} yield element

但仅此一点是行不通的，因为classifyElements 本身返回AnyRef，而我们希望它返回一个集合。现在，validElements 返回一个集合，因此这不是问题。我们只需要修复else部分。由于 validElements 返回一个 IndexedSeq，因此我们也在 else 部分返回它。最终结果是：

trait Element
object Unrecognized extends Element
case class Letter(c: Char) extends Element
case class Punct(c: Char) extends Element
val expr1 = "This is a silly example." split "\\b"

def wordOrPunct(j: Char) = if (j.isLetter) Letter(j.toLower) else Punct(j)
def validElements(i: String) = for (j <- i) yield wordOrPunct(j)
def classifyElements(i: String) = if (i.nonEmpty) validElements(i) else IndexedSeq(Unrecognized)
val result = for {
  i <- expr1
  element <- classifyElements(i)
} yield element

它执行与您所呈现的循环和条件完全相同的组合，但它更具可读性并且易于更改。

关于产量

我认为重要的是要注意所提出的问题的一件事。让我们简化一下：

for (i <- expr1) {
  for (j <- i) {
    doSomething
  }
}

现在，这是使用 foreach 实现的（请参阅此处< /a>，或其他类似的问题和答案）。这意味着上面的代码与此代码执行完全相同的操作：

for {
  i <- expr1
  j <- i
} doSomething

完全相同。当使用 yield 时，情况就完全不正确了。以下表达式不会产生相同的结果：

for (i <- expr1) yield for (j <- i) yield j

for (i <- expr1; j <- i) yield j

第一个代码段将通过两个 map 调用来实现，而第二个代码段将使用一个 flatMap 和一个 map< /代码>。

因此，只有在 yield 的上下文中，担心嵌套 for 循环或使用多个生成器才有意义。而且，事实上，生成器代表着正在生成某些东西的事实，这仅适用于真正的 for 理解（yield荷兰国际集团的东西）。

Answer 2

该部分

for (j <- i) {
   if (j.method) {
     doSomething(j)
   } else {
     doSomethingElse(j)
   }
 }

可以重写为

for(j <- i; e = Either.cond(j.method, j, j)) {
  e.fold(doSomething _, doSomethingElse _)  
}  

（当然，如果您的 do.. 方法返回某些内容，您可以使用yield 代替）

这里它并不是那么有用，但如果您有更深的嵌套结构，它可以...

Answer 3

import scalaz._; import Scalaz._

val lhs = (_ : List[X]) collect { case j if j.methodJ => doSomething(j) } 
val rhs = (_ : List[X]) map doSomethingElse
lhs <-: (expr1 partition methodI) :-> rhs

Answer 4

你不能。 for(expr; if) 构造仅过滤必须在循环中处理的元素。

Answer 5

如果顺序对于调用 doSomething() 和 doSomethingElse() 并不重要，那么您可以像这样重新排列代码。

val (tmp, no1) = expr1.partition(_.method)
val (yes, no2) = tmp.partition(_.method)

yes.foreach(doSomething())
no1.foreach(doSomethingElse())
no2.foreach(doSomethingElse())

为了回答您原来的问题，我认为对于特定用例来说，理解可能非常好，但您的示例不太适合。

Answer 6

Scala 中指定的操作条件用于过滤生成器中的元素。不满足条件的元素将被丢弃，并且不会呈现给yield /代码块。

这意味着，如果您想根据条件表达式执行替代操作，则需要将测试推迟到yield /代码块。

另请注意，for 操作的计算成本相对较高（目前），因此也许更简单的迭代方法可能更合适，也许类似于：

expr1 foreach {i =>
  if (i.method) {
    i foreach {j =>
      if (j.method)
        doSomething()
      else
        doSomethingElseA()
    }
  }
  else
    doSomethingElseB()
}

更新：

如果您必须使用 for 理解并且可以接受一些限制，那么这可能会起作用:

for (i <- expr1; j <- i) {
  if (i.method) {if (j.method) doSomething() else doSomethingElseA()} else doSomethingElseB()
}