Java map/collect 用于流对列表列表求和

Question

我正在使用 Java 8，以及包 java.util.stream 中的 Map、Reduce、Collect API 。 我有一个用例，我需要对列表的列表进行垂直求和。 我的列表的结构是 List<List<Integer>> 我只是想不出一种方法可以使用映射来压缩内部列表，并减少或收集流上可用的操作。 考虑以下代码。 假设我的列表名为 transactions 。 此交易存储的是内部列表中每个商品的每笔交易中售出的商品数量。内部 list 的长度对于所有交易都是相同的，因为它包含库存中每一项的一个条目。如果该项目未参与交易，则仅存储 0。

我在想一些类似的事情

transaction.stream().map(
      /*Try and put sum into an accumulator List*/
).collect(
      /*Collect entries into a list*/
)

我只是不知道如何在代码中实现这一点。 网络上的所有资源都是简单的示例，不适用于集合对象。任何指示/提示将不胜感激。

Answer 1

因为你有一个List<List<Integer>> ，看起来垂直总和应该产生 List<Integer> ，因为每一列都有自己的总和。假设 List<List<Integer>> 中的每一行包含相同数量的元素，最好首先找到 List<List<Integer>> 的转置，对每一行求和，并将其收集回 List<Integer> 。您可以使用以下方法来完成此操作:

List<List<Integer>> list = Arrays.asList(
    Arrays.asList(1, 2, 3, 4),
    Arrays.asList(5, 6, 7, 8)
);

List<Integer> sumList = IntStream.range(0, list.get(0).size())
                                 .mapToObj(i -> list.stream()
                                                    .mapToInt(l -> l.get(i))
                                                    .sum())
                                 .collect(Collectors.toList());

System.out.println(sumList);

输出:

[6, 8, 10, 12]