计算scala中序列的后续元素的差异

时间:2013-06-28 19:52:45

标签: scala

我想在scala中执行almost exactly this。有优雅的方式吗?

具体来说,我只想要序列中相邻元素的区别。例如

input = 1,2,6,9
output = 1,4,3

2 个答案:

答案 0 :(得分:9)

这个怎么样?

scala> List(1, 2, 6, 9).sliding(2).map { case Seq(x, y, _*) => y - x }.toList
res0: List[Int] = List(1, 4, 3)

答案 1 :(得分:5)

这是一个使用递归并在列表中效果最佳的

def differences(l:List[Int]) : List[Int] = l match {
  case a :: (rest @ b :: _) => (b - a) :: differences(rest)
  case _ => Nil
}

这是一个应该在Vector或Array上非常快的一个:

def differences(a:IndexedSeq[Int]) : IndexedSeq[Int] = 
  a.indices.tail.map(i => a(i) - a(i-1))

当然总有这样:

def differences(a:Seq[Int]) : Seq[Int] = 
  a.tail.zip(a).map { case (x,y) => x - y }

请注意,只有递归版本处理空列表而没有例外。