java - 如何使用lambda收集java8中多个列表中具有相同索引的元素

Question

我现在正在使用 Java8，

我想做的是根据元素在每个列表中的索引从不同的列表中挑选元素

我有以下代码:

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

我想要这样的输出:

[[1,2,3],[4,5,6],[7,8,9]]

我试过了，但没有得到结果。那么你能给我一些提示吗？

Answer 1

您可以创建列表索引的 IntStream 并映射它们。下面的代码解决了你的问题(假设输入列表是非空的并且每一行都有相同数量的值):

List<List<Integer>> transposed = IntStream.range(0, lst.get(0).size())
    .mapToObj(idx -> lst.stream().map(l -> l.get(idx)).collect(Collectors.toList()))
    .collect(Collectors.toList());