一起使用 PartioningBy 和 groupingBy 时无法获得所需的输出答案

【问题标题】：Not able to get desired output while working with PartioningBy and groupingBy together一起使用 PartioningBy 和 groupingBy 时无法获得所需的输出
【发布时间】：2016-12-05 11:25:59
【问题描述】：

我正在使用收集器的 groupingBy 和 partitioningBy 函数。我与人员列表一起工作，人员列表如下：

List<Person> persons =
        Arrays.asList(
            new Person("Max", 18),
            new Person("Peter", 23),
            new Person("Pamela", 23),
            new Person("David", 12),
            new Person("Pam", 12));

我想要的是根据名字以字母“P”开头的人来划分列表，然后根据他们的年龄对他们进行分组。这是我执行上述过滤的代码：

Map<Boolean, Map<Object, List<Person>>> rr = persons.stream()
                                            .collect(Collectors.partitioningBy(p -> p.name.startsWith("P"), 
                                                    Collectors.groupingBy(p -> p.age > 20)));

我得到的输出是：

rr = {false={false=[Max, David]}, true={false=[Pam], true=[Peter, Pamela]}}

现在，我的要求是仅从上述结果中获取内部地图。也就是说，我想将返回值更改为：

{false=[Pam], true=[Peter, Pamela]}

也就是说，我想要partioningBy 函数返回的布尔值为真的结果（或分区映射）。我怎样才能做到这一点？

【问题讨论】：

为什么你想要这个？只需执行filter(p -> p.name.startsWith("P")) 之前 .collect(Collectors.groupingBy(p -> p.age > 20))。否则，如果您想收集以后不使用的数据，只需按照您的问题中发布的方法进行操作，然后在结果上致电get(true)。
原因不详。只是看看这种特殊情况是否有任何方法。
嗯，你可以先过滤，也可以后提取。两者都很简单。你还有什么想法？

标签： java java-8 java-stream partitioning collectors

【解决方案1】：

您可以创建一个自定义收集器（我只是将其作为练习，请照此处理）：

 static class MyCustom<T, U> implements Collector<Person, List<Person>, Map<T, List<U>>> {

    private final Function<Person, T> function;

    private final Predicate<Person> predicate;

    private final Function<Person, U> transformingFunction;

    public MyCustom(Predicate<Person> predicate, Function<Person, T> function,
            Function<Person, U> transformingFunction) {
        this.predicate = predicate;
        this.function = function;
        this.transformingFunction = transformingFunction;
    }

    @Override
    public Supplier<List<Person>> supplier() {
        return ArrayList::new;
    }

    @Override
    public BiConsumer<List<Person>, Person> accumulator() {
        return (list, person) -> {
            if (predicate.test(person)) {
                list.add(person);
            }
        };
    }

    @Override
    public BinaryOperator<List<Person>> combiner() {
        return (l1, l2) -> {
            l1.addAll(l2);
            return l1;
        };
    }

    @Override
    public Function<List<Person>, Map<T, List<U>>> finisher() {
        return list -> {
            return list.stream().collect(
                    Collectors.groupingBy(function, Collectors.mapping(transformingFunction, Collectors.toList())));
        };
    }

    @Override
    public Set<java.util.stream.Collector.Characteristics> characteristics() {
        return EnumSet.of(Characteristics.UNORDERED);
    }
}

然后像这样应用它：

 MyCustom<Integer, String> custom = new MyCustom<>((Person p) -> p.getName().startsWith("P"),
            (Person p) -> p.getAge(), Person::getName);

 System.out.println(persons.stream().collect(custom)); // {23=[Peter, Pamela], 12=[Pam]}

【讨论】：

【解决方案2】：

按name过滤

按age应用分区

    Map<Boolean, List<Person>> p1 = persons.stream().filter(p -> p.name.startsWith("P")).collect(Collectors.partitioningBy(p -> p.getAge() > 20));

【讨论】：