java - 在 Java 8+ 中对 map 进行分区

Question

我有一个 Map<String, String>和 List<String> .我想对 Map 进行分区根据条件

foreach(map.key -> list.contains(map.key))

并产生两个 Map (s)。这样做最优雅的方式是什么？我在 Java 11 上，所以你可以在答案中抛出你想要的一切。

我现在想到的是:

map.entrySet()
   .stream()
   .collect(partitioningBy(e -> list.contains(o.getKey())));

但这给出了 Map<Boolean, List<Entry<String, String>>> .

Answer 1

您可以使用 toMap(作为下游收集器)减少每个组:

Map<String, String> myMap = new HashMap<>();
myMap.put("d", "D");
myMap.put("c", "C");
myMap.put("b", "B");
myMap.put("A", "A");

List<String> myList = Arrays.asList("a", "b", "c");

Map<Boolean, Map<String, String>> result = myMap.entrySet()
        .stream()
        .collect(Collectors.partitioningBy(
                            entry -> myList.contains(entry.getKey()),
                            Collectors.toMap(Entry::getKey, Entry::getValue)
                    )
        );

对于这个例子，产生 {false={A=A, d=D}, true={b=B, c=C}}