在Java 8 Streams中使用Streams的Hashmap来收集Map的值

时间:2015-04-14 10:54:20

标签: java hashmap java-8 collectors

考虑一个hashmap

Map<Integer, List> id1 = new HashMap<Integer,List>();

我在两个hashmap中插入了一些值。

例如,

    List<String> list1 = new ArrayList<String>();

    list1.add("r1");
    list1.add("r4");

    List<String> list2 = new ArrayList<String>();
    list2.add("r2");
    list2.add("r5");

    List<String> list3 = new ArrayList<String>();
    list3.add("r3");
    list3.add("r6");

    id1.put(1,list1);
    id1.put(2,list2);
    id1.put(3,list3);
    id1.put(10,list2);
    id1.put(15,list3);

Q1)现在我想对hashmap中的键应用一个过滤条件并检索相应的值(List)。

例如:我的查询是key = 1,输出应该是&#39; list1&#39;

我写了

id1.entrySet().stream().filter( e -> e.getKey() == 1);

但我不知道如何检索列表作为此流操作的输出。

Q2)我想再次对hashmap中的键应用过滤条件并检索相应的列表列表。

例如:这里我的查询是key = 1%(即密钥可以是1,10,15),输出应该是&#39; list1&#39;,&#39; list2&#39;,&#39; ; list3&#39;(列表清单)。

6 个答案:

答案 0 :(得分:57)

如果您确定最多只能获得一个通过过滤器的元素(由过滤器保证),您可以使用findFirst

Optional<List> o = id1.entrySet()
                      .stream()
                      .filter( e -> e.getKey() == 1)
                      .map(Map.Entry::getValue)
                      .findFirst();

在一般情况下,如果过滤器可能匹配多个列表,您可以将它们收集到列表列表中:

List<List> list = id1.entrySet()
                     .stream()
                     .filter(.. some predicate...)
                     .map(Map.Entry::getValue)
                     .collect(Collectors.toList());

答案 1 :(得分:23)

您需要做的是从Stream Map中创建.entrySet()

// Map<K, V> --> Set<Map.Entry<K, V>> --> Stream<Map.Entry<K, V>>
map.entrySet().stream()

从上面开始,您可以.filter()对这些条目进行操作。例如:

// Stream<Map.Entry<K, V>> --> Stream<Map.Entry<K, V>>
.filter(entry -> entry.getKey() == 1)

要从中获取值.map()

// Stream<Map.Entry<K, V>> --> Stream<V>
.map(Map.Entry::getValue)

最后,您需要收集到List

// Stream<V> --> List<V>
.collect(Collectors.toList())

如果您只有一个条目,请改用它(注意:此代码假定有一个值;否则,请使用.orElse();有关详细信息,请参阅javadoc of Optional

// Stream<V> --> Optional<V> --> V
.findFirst().get()

答案 2 :(得分:4)

对于您的Q2,您的问题已有答案。对于你的Q1,更常见的是当你知道密钥的过滤应该给出一个唯一的值时,根本不需要使用Streams。

只需使用getgetOrDefault,即:

List<String> list1 = id1.getOrDefault(1, Collections.emptyList());

答案 3 :(得分:1)

您也可以这样做

public Map<Boolean, List<Student>> getpartitionMap(List<Student> studentsList) {

    List<Predicate<Student>> allPredicates = getAllPredicates();
    Predicate<Student> compositePredicate =  allPredicates.stream()
                             .reduce(w -> true, Predicate::and)
     Map<Boolean, List<Student>> studentsMap= studentsList
                .stream()
                .collect(Collectors.partitioningBy(compositePredicate));
    return studentsMap;
}

public List<Student> getValidStudentsList(Map<Boolean, List<Student>> studentsMap) throws Exception {

    List<Student> validStudentsList =  studentsMap.entrySet()
             .stream()
             .filter(p -> p.getKey() == Boolean.TRUE)
             .flatMap(p -> p.getValue().stream())
             .collect(Collectors.toList());

    return validStudentsList;
}

public List<Student> getInValidStudentsList(Map<Boolean, List<Student>> studentsMap) throws Exception {

    List<Student> invalidStudentsList = 
             partionedByPredicate.entrySet()
             .stream()
             .filter(p -> p.getKey() == Boolean.FALSE)
             .flatMap(p -> p.getValue().stream())
             .collect(Collectors.toList());

    return invalidStudentsList;

}

使用flatMap,您只需List<Student>而不是List<List<Student>>

由于

答案 4 :(得分:1)

使用keySet-

id1.keySet().stream()
        .filter(x -> x == 1)
        .map(x -> id1.get(x))
        .collect(Collectors.toList())

答案 5 :(得分:0)

也许示例过于简单,但您不需要此处的 Java 流 API。直接使用地图即可。

path