如何使用java8通过谓词对列表进行分区?

时间:2016-04-17 15:38:40

标签: java list java-8 java-stream

我有一个列表a,我希望将其分成几个小列表。

说出包含“aaa”的所有项目,包含“bbb”和更多谓词的所有项目。

我怎样才能使用java8?

我看到了这个post但它只拆分为2个列表。

public void partition_list_java8() {

    Predicate<String> startWithS = p -> p.toLowerCase().startsWith("s");

    Map<Boolean, List<String>> decisionsByS = playerDecisions.stream()
            .collect(Collectors.partitioningBy(startWithS));

    logger.info(decisionsByS);

    assertTrue(decisionsByS.get(Boolean.TRUE).size() == 3);
}

我看到了这个post,但在java 8之前它已经很老了。

2 个答案:

答案 0 :(得分:28)

就像在@RealSkeptic comment Predicate中解释的那样只能返回两个结果:true和false。这意味着您只能将数据拆分为两组 您需要的是某种Function,它允许您确定应该组合在一起的元素的一些常见结果。在你的情况下,这样的结果可能是小写的第一个字符(假设所有字符串都不是空的 - 至少有一个字符)。

现在使用Collectors.groupingBy(function),您可以将所有元素分组到单独的列表中,并将它们存储在Map中,其中key将是用于分组的常用结果(如第一个字符)。

所以你的代码看起来像

Function<String, Character> firstChar =  s -> Character.toLowerCase(s.charAt(0));

List<String> a = Arrays.asList("foo", "Abc", "bar", "baz", "aBc");
Map<Character, List<String>> collect = a.stream()
        .collect(Collectors.groupingBy(firstChar));

System.out.println(collect);

输出:

{a=[Abc, aBc], b=[bar, baz], f=[foo]}

答案 1 :(得分:5)

您可以使用Collectors.groupingBy来转换(分组) - >&gt; (该分组中的事物列表)。如果您不关心分组本身,请在该地图上调用values()以获取Collection<List<String>>分区。