使用默认元素压缩两个不同长度的列表以进行填充

时间:2016-02-15 09:27:19

标签: scala functional-programming

假设我们有以下不同大小的列表:

val list1 = ("a", "b", "c")
val list2 = ("x", "y")

现在我想合并这两个列表并创建一个新的列表,其中包含串联的字符串元素:

val desiredResult = ("ax", "by", "c")

我试过

val wrongResult = (list1, list2).zipped map (_ + _)

建议使用here,但这不能按预期工作,因为zip会丢弃那些无法匹配的较长列表中的元素。

我该如何解决这个问题?有没有办法压缩列表并给出一个“默认元素”(如本例中的空字符串),如果一个列表更长?

2 个答案:

答案 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值。