如何使用地图上相同的第一个项目整理列表项

时间:2012-05-10 20:29:07

标签: list scala

我知道这可能是Scala中一个非常简单的List操作,但我是一个新手,无法弄明白。我有一个查询返回一个结果集,其中包含一系列值,按公共ID分组。例如:

结果集:

[{ 1, "a", 30 },
{ 1, "b", 20 },
{ 1, "c", 22 },
{ 2, "a", 32 },
{ 2, "c", 10 }]

我想做的就是将其放入地图中:

1 -> [{"a", 30}, {"b", 20}, {"c", 22}]
2 -> [{"a", 32}, {"c", 10}]

我认为收集方法可用于此但无法弄明白。

1 个答案:

答案 0 :(得分:10)

我不确定数据结构中的类型是什么,但也许您可以对此进行调整。假设您有一组元组:

val items = 
  List((1, "a", 30),
       (1, "b", 20),
       (1, "c", 22),
       (2, "a", 32),
       (2, "c", 10))

items
  .groupBy{ case (a,b,c) => a }
  .mapValues(_.map{ case (a,b,c) => (b,c) })

// Map(1 -> List((a,30), (b,20), (c,22)), 2 -> List((a,32), (c,10)))

或者,更简洁:

items.groupBy(_._1).mapValues(_.map(t => (t._2, t._3)))

collect方法完全不同(基本上,它是map,它会丢弃不匹配的值)。 groupBy方法就是您真正想要的方法。