有条件地合并列表[List[_]]

Question

我想根据内部列表中元素的值合并List[List[Double]]。到目前为止，这是我所拥有的：

// inner Lists are (timestamp, ID, measurement)
val data = List(List(60, 0, 3.4),  List(60, 1, 2.5),  List(120, 0, 1.1),
                List(180, 0, 5.6), List(180, 1, 4.4), List(180, 2, 6.7))

data
  .foldLeft(List[List[Double]]())(
    (ret, ll) =>

      // if this is the first list, just add it to the return val
      if (ret.isEmpty){
        List(ll)

      // if the timestamps match, add a new (ID, measurement) pair to this inner list
      } else if (ret(0)(0) == ll(0)){
        {{ret(0) :+ ll(1)} :+ ll(2)} :: ret.drop(1)

      // if this is a new timestamp, add it to the beginning of the return val
      } else {
        ll :: ret
      }
  )

这行得通，但它对我来说并不是最佳选择（尤其是右侧添加“:+”）。对于我的用例，我有一个相当大的（约 25,000 个内部列表）元素列表，它们本身都是长度为 3 的列表。最多会有四重退化，因为内部列表是List(timestamp, ID, measurement) 组，并且只有四个唯一ID。本质上，我想将所有具有相同时间戳的测量值混合在一起。

有没有人看到更优化的方法？

如果有更好的方法从那一点开始的话，我实际上是从时间戳List[Double] 和四个 ID 中的每一个的测量值 List[Double] 开始的。

Answer 1

这里有一个稍微短一点的方法：

data.
  groupBy(_(0)).
  mapValues(_.flatMap(_.tail)).
  toList.
  map(kv => kv._1 :: kv._2)

结果看起来与您的算法产生的结果 1:1 完全相同：

List(List(180.0, 0.0, 5.6, 1.0, 4.4, 2.0, 6.7), List(120.0, 0.0, 1.1), List(60.0, 0.0, 3.4, 1.0, 2.5))
List(List(180.0, 0.0, 5.6, 1.0, 4.4, 2.0, 6.7), List(120.0, 0.0, 1.1), List(60.0, 0.0, 3.4, 1.0, 2.5))

解释：

• 按时间戳分组
• 在分组值中，删除多余的时间戳，并展平为单个列表
• 将时间戳重新添加到 ids-&-measurements 的平面列表中

Answer 2

这是一种可能性：

input
  .groupBy(_(0))
  .map { case (tstp, values) => tstp :: values.flatMap(_.tail) }

这个想法只是将内部列表按其第一个元素分组，然后将结果值展平。

返回：

List(List(180.0, 0.0, 5.6, 1.0, 4.4, 2.0, 6.7), List(120.0, 0.0, 1.1), List(60.0, 0.0, 3.4, 1.0, 2.5))

Answer 3

用案例类表示您的测量结果怎么样？

case class Measurement(timestamp: Int, id: Int, value: Double)

val measurementData = List(Measurement(60, 0, 3.4),  Measurement(60, 1, 2.5),  
    Measurement(120, 0, 1.1), Measurement(180, 0, 5.6), 
    Measurement(180, 1, 4.4), Measurement(180, 2, 6.7))

measurementData.foldLeft(List[Measurement]())({
    case (Nil, m) => List(m)
    case (x :: xs, m) if x.timestamp == m.timestamp => m :: xs
    case (xs, m) => m :: xs
  })