5

我希望添加一个整数元组列表，即给定一个元组 k 的输入列表，生成一个元组k的元组，其字段是列表中元组的相应字段的总和。

输入

List( (1,2,3), (2,3,-3), (1,1,1))

输出

(4, 6, 1)

我试图使用foldLeft，但我无法编译它。现在，我正在使用for循环，但我正在寻找更简洁的解决方案。

4

3 回答 3

7

这可以使用shapeless安全且非常简洁地完成输入，

scala> import shapeless._, syntax.std.tuple._
import shapeless._
import syntax.std.tuple._

scala> val l = List((1, 2, 3), (2, 3, -1), (1, 1, 1))
l: List[(Int, Int, Int)] = List((1,2,3), (2,3,-1), (1,1,1))

scala> l.map(_.toList).transpose.map(_.sum)
res0: List[Int] = List(4, 6, 3)

请注意，与依赖强制转换的解决方案不同，这种方法是类型安全的，并且在编译时而不是在运行时检测到任何类型错误，

scala> val l = List((1, 2, 3), (2, "foo", -1), (1, 1, 1))
l: List[(Int, Any, Int)] = List((1,2,3), (2,foo,-1), (1,1,1))

scala> l.map(_.toList).transpose.map(_.sum)
<console>:15: error: could not find implicit value for parameter num: Numeric[Any]
              l.map(_.toList).transpose.map(_.sum)
                                          ^
于 2013-09-16T21:28:36.557 回答
3 
scala> val tuples = List( (1,2,3), (2,3,-3), (1,1,1))
tuples: List[(Int, Int, Int)] = List((1,2,3), (2,3,-3), (1,1,1))

scala> tuples.map(t => t.productIterator.toList.map(_.asInstanceOf[Int])).transpose.map(_.sum)
res0: List[Int] = List(4, 6, 1)

productIterator调用时类型信息会丢失，Tuple3因此您必须从Any回转换为Int.

如果元组总是包含相同的类型，我建议使用另一个集合，例如List. Tuple更适合不同的类型。当您具有相同的类型并且通过使用该解决方案不会丢失类型信息时productIterator会更加优雅。

scala> val tuples = List(List(1,2,3), List(2,3,-3), List(1,1,1))
tuples: List[List[Int]] = List(List(1, 2, 3), List(2, 3, -3), List(1, 1, 1))

scala> tuples.transpose.map(_.sum)
res1: List[Int] = List(4, 6, 1)
于 2013-09-16T19:39:24.887 回答
2 
scala> val list = List( (1,2,3), (2,3,-3), (1,1,1))
list: List[(Int, Int, Int)] = List((1,2,3), (2,3,-3), (1,1,1))

scala> list.foldRight( (0, 0, 0) ){ case ((a, b, c), (a1, b1, c1)) => (a + a1, b + b1, c + c1) }
res0: (Int, Int, Int) = (4,6,1)
于 2013-09-16T19:33:06.023 回答