假设我们有以下不同大小的列表:
val list1 = ("a", "b", "c")
val list2 = ("x", "y")
现在我想合并这两个列表并创建一个新的列表,其中包含串联的字符串元素:
val desiredResult = ("ax", "by", "c")
我试过
val wrongResult = (list1, list2).zipped map (_ + _)
建议使用here,但这不能按预期工作,因为zip会丢弃那些无法匹配的较长列表中的元素。
我该如何解决这个问题?有没有办法压缩列表并给出一个“默认元素”(如本例中的空字符串),如果一个列表更长?
答案 0 :(得分:36)
您正在寻找的方法是.zipAll
:
scala> val list1 = List("a", "b", "c")
list1: List[String] = List(a, b, c)
scala> val list2 = List("x", "y")
list2: List[String] = List(x, y)
scala> list1.zipAll(list2, "", "")
res0: List[(String, String)] = List((a,x), (b,y), (c,""))
.zipAll
有3个参数:
this
(调用集合.zipAll
)更短答案 1 :(得分:4)
基于API的zipAll
是可行的方法,但您可以实现它(作为练习),例如,如下所示,
implicit class OpsSeq[A,B](val xs: Seq[A]) extends AnyVal {
def zipAll2(ys: Seq[B], xDefault: A, yDefault: B) = {
val xs2 = xs ++ Seq.fill(ys.size-xs.size)(xDefault)
val ys2 = ys ++ Seq.fill(xs.size-ys.size)(yDefault)
xs2.zip(ys2)
}
}
因此,例如
Seq(1,2).zipAll2(Seq(3,4,5),10,20)
List((1,3), (2,4), (10,5))
和
list1.zipAll2(list2, "", "")
List((a,x), (b,y), (c,""))
递归版,
def zipAll3[A,B](xs: Seq[A], ys: Seq[B], xd: A, yd: B): Seq[(A,B)] = {
(xs,ys) match {
case (Seq(), Seq()) => Seq()
case (x +: xss, Seq()) => (x,yd) +: zipAll3(xss, Seq(), xd, yd)
case (Seq(), y +: yss) => (xd,y) +: zipAll3(Seq(), yss, xd, yd)
case (x +: xss, y +: yss) => (x,y) +: zipAll3(xss, yss, xd, yd)
}
}
默认xd
和默认yd
值。