Use 4 (or N) collections to yield only one value at a time (1xN) (i.e. zipped for tuple4+)

孤街醉人 提交于 2019-12-05 08:09:43

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).

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)
       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

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
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!