如何用Java流api过滤地图？

Question

Map<Integer, String> map = new HashMap<>();
  map.put(1, "f");
  map.put(2, "I");
  map.put(3, "a");
  map.put(4, "c");....etc

现在我有一个列表：

List<Integer> picks = {1,3}

我想得到一个字符串列表，即从map中匹配键值的值，在' pick '列表中找到。所以，我期望得到结果{"f "，" a"}。有没有办法用java stream api优雅地做到？

当有一个值时，我会这样做：

map.entrySet().stream()
   .filter(entry -> "a".equals(entry.getValue()))
   .map(entry -> entry.getValue())
   .collect(Collectors.toList())

但是，当有一个键/选择列表要过滤时，就很难了。

Answer 1

您不需要遍历整个地图来收集选择的值。而是遍历所需的键并从地图中获取相关的值。如果地图比您需要选择的值大得多（通常情况下），那么这种方法应该优于其他方法。此外，这种解决方案更紧凑，对我来说视觉混乱更少。看起来是这样的。

List<String> pickedValues = picks.stream().map(map::get)
    .filter(Objects::nonNull)
    .collect(Collectors.toList());

Answer 2

您可以通过使用这样的东西来实现这一点：

List<String> values = map.entrySet()
                .stream()
                .filter(entry -> picks.contains(entry.getKey()))
                .map(Map.Entry::getValue)
                .collect(Collectors.toList());

values.forEach(System.out::println);

输出：

f
a

但是，由于< code>List::contains是O(N ),所以它可能效率不高。考虑对< code>picks使用< code>Set(例如< code>HashSet)而不是< code>List，因为< code>HashSet::contains是O(1)。

Answer 3

您可以在Stream#filter中使用List.contains（）来仅接受列表中存在的那些值：

List<String> result = map.entrySet()
   .stream()
   .filter(ent -> picks.contains(ent.getKey()))
   .map(Map.Entry::getValue)     
   .collect(Collectors.toList());