我希望添加一个整数元组列表,即给出一个arity k 元组的输入列表,生成一个arity元组 k ,其字段是对应的总和列表中元组的字段。
List( (1,2,3), (2,3,-3), (1,1,1))
(4, 6, 1)
我试图使用foldLeft
,但我无法将其编译。现在,我正在使用 for 循环,但我一直在寻找更简洁的解决方案。
答案 0 :(得分: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)
^
答案 1 :(得分: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)
答案 2 :(得分: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)