Java 8:在Stream上执行reduce操作

时间:2017-06-05 23:33:58

标签: java java-8 java-stream reduce

我有一个包含键值对的java.util.stream.Stream,如:

<1,3> <1,5> <3,1> <4,2> <4,7> <4,8>

现在我想合并所有具有相同密钥的条目:

<1,[3,5]>  <3,[1]> <4,[2,7,8]>

数据已经排序,因此只需要合并连续的数据集。

现在,我正在寻找一种方法来转换上面的流内容,而不是将所有数据集都加载到内存中。

我更喜欢使用包含值列表而不是单个值的不同对象类型来获取java.util.stream.Stream作为结果。

我唯一的方法是执行合并的自定义迭代器,但是转换为迭代器并返回流似乎非常难看。

最好的方法是什么?

1 个答案:

答案 0 :(得分:4)

以下是SteamEx的解决方案。

int[][] datasets = { { 1, 3 }, { 1, 5 }, { 3, 1 }, { 4, 2 }, { 4, 7 }, { 4, 8 } };

StreamEx.of(datasets) //
        .collapse((a, b) -> a[0] == b[0], groupingBy(a -> a[0], mapping(a -> a[1], toList()))) //
        .forEach(System.out::println);

您可以使用int[]对象替换dataset。我们可以添加peek来验证它是否是延迟加载/计算:

StreamEx.of(datasets) //
        .peek(System.out::println) //
        .collapse((a, b) -> a[0] == b[0], groupingBy(a -> a[0], mapping(a -> a[1], toList()))) //
        .limit(1) //
        .forEach(System.out::println);