Use 4 (or N) collections to yield only one value a

2019-04-28 20:52发布

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

scala> val b = List(3,4)
b: List[Int] = List(3, 4)

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

scala> val d = List(7,8)
d: List[Int] = List(7, 8)

scala> (a,b,c).zipped.toList
res6: List[(Int, Int, Int)] = List((1,3,5), (2,4,6))

Now:

scala> (a,b,c,d).zipped.toList
<console>:12: error: value zipped is not a member of (List[Int], List[Int], List[Int],     List[Int])
              (a,b,c,d).zipped.toList
                       ^

I've searched for this elsewhere, including this one and this one, but no conclusive answer.

I want to do the following or similar:

for((itemA,itemB,itemC,itemD) <- (something)) yield itemA + itemB + itemC + itemD

Any suggestions?

4条回答
Summer. ? 凉城
2楼-- · 2019-04-28 20:55

Short answer:

for (List(w,x,y,z) <- List(a,b,c,d).transpose) yield (w,x,y,z)
  // List[(Int, Int, Int, Int)] = List((1,3,5,7), (2,4,6,8))

Why you want them as tuples, I'm not sure, but a slightly more interesting case would be when your lists are of different types, and for example, you want to combine them into a list of objects:

case class Person(name: String, age: Int, height: Double, weight: Double)
val names =   List("Alf", "Betty")
val ages =    List(22, 33)
val heights = List(111.1, 122.2)
val weights = List(70.1, 80.2)
val persons: List[Person] = ???

Solution 1: using transpose, as above:

for { List(name: String, age: Int, height: Double, weight: Double) <- 
        List(names, ages, heights, weights).transpose
    } yield Person(name, age, height, weight)

Here, we need the type annotations in the List extractor, because transpose gives a List[List[Any]].

Solution 2: using iterators:

val namesIt   = names.iterator
val agesIt    = ages.iterator
val heightsIt = heights.iterator
val weightsIt = weights.iterator

for { name <- names } 
  yield Person(namesIt.next, agesIt.next, heightsIt.next, weightsIt.next)

Some people would avoid iterators because they involve mutable state and so are not "functional". But they're easy to understand if you come from the Java world and might be suitable if what you actually have are already iterators (input streams etc).

查看更多
Evening l夕情丶
3楼-- · 2019-04-28 20:58

Shameless plug-- product-collections does something similar:

 a flatZip b flatZip c flatZip d
 res0: org.catch22.collections.immutable.CollSeq4[Int,Int,Int,Int] = 
 CollSeq((1,3,5,7),
         (2,4,6,8))

 scala> res0(0) //first row
 res1: Product4[Int,Int,Int,Int] = (1,3,5,7)

 scala> res0._1 //first column
 res2: Seq[Int] = List(1, 2)
查看更多
Root(大扎)
4楼-- · 2019-04-28 21:07
       val g = List(a,b,c,d)
   val result = ( g.map(x=>x(0)), g.map(x=>x(1) ) )

result : (List(1, 3, 5, 7),List(2, 4, 6, 8))

basic, zipped assit tuple2 , tuple3 http://www.scala-lang.org/api/current/index.html#scala.runtime.Tuple3Zipped so, You want 'tuple4zippped' you make it

gool luck

查看更多
我欲成王,谁敢阻挡
5楼-- · 2019-04-28 21:12

found a possible solution, although it's very imperative to my taste:

  val a = List(1,2)
  val b = List(3,4)
  val c = List(5,6)
  val d = List(7,8)

  val g : List[Tuple4[Int,Int,Int,Int]] = {
    a.zipWithIndex.map { case (value,index) => (value, b(index), c(index), d(index))}  
  }

zipWithIndex would allow me to go through all the other collections. However, i'm sure there's a better way to do this. Any suggestions?

Previous attempts included:

Ryan LeCompte's zipMany or transpose.

however, it a List, not a tuple4. this is not as convenient to work with since i can't name the variables.

Tranpose it's already built in in the standard library and doesn't require higher kinds imports so it's preferrable, but not ideal.

I also, incorrectly, tried the following example with Shapeless

scala> import Traversables._
import Tuples._
import Traversables._
import Tuples._
import scala.language.postfixOps

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

scala> val b = List(3,4)
b: List[Int] = List(3, 4)

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

scala> val d = List(7,8)
d: List[Int] = List(7, 8)

scala> val x = List(a,b,c,d).toHList[Int :: Int :: Int :: Int :: HNil] map tupled
x: Option[(Int, Int, Int, Int)] = None
查看更多
登录 后发表回答