scala:产生两个循环的成对组合

时间:2012-07-30 12:42:28

标签: scala

我想创建一个包含元组的集合,其中包含两个列表的所有成对组合。类似的东西:

for ( x <- xs )
  for ( y <- ys ) 
    yield (x,y)

在Python中,这可行,在Scala中显然for只产生最后一个循环(因此评估为Unit

在Scala中实现它的最简洁方法是什么?

2 个答案:

答案 0 :(得分:30)

你快到了那里:

scala> val xs = List (1,2,3)
xs: List[Int] = List(1, 2, 3)

scala> val ys = List (4,5,6)
ys: List[Int] = List(4, 5, 6)

scala> for (x <- xs; y <- ys) yield (x,y)
res3: List[(Int, Int)] = List((1,4), (1,5), (1,6), (2,4), (2,5), (2,6), (3,4), (3,5), (3,6))

答案 1 :(得分:9)

根据尼古拉斯的说法更明确一点:
在Scala中,您可以在一个单一的理解中使用多个生成器。

val xs = List(1,2,3)
val ys = List(4,5)

for {
  x <- xs
  y <- ys
} yield (x,y)

res0: List[(Int, Int)] = List((1,4), (1,5), (2,4), (2,5), (3,4), (3,5))

你甚至可以在理解中进行评估。

for {
  x <- xs
  y <- ys
  if (x + y == 6)
} yield (x,y)

res1: List[(Int, Int)] = List((1,5), (2,4))

或者进行作业。

for {
  x <- xs
  y <- ys
  val z = x + y
} yield (x,y,z)

res2: List[(Int,Int,Int)] = List((1,4,5), (1,5,6), (2,4,6), (2,5,7), (3,4,7), (3,5,8))