我有以下Map
:
HashMap<String, String> map1= new HashMap<String, String>();
map1.put("1", "One");
map1.put("2", "Two");
map1.put("3", "Three");
我有一个包含numbers
["1","2","3"]
我必须执行以下操作:
List<String> spelling= new ArrayList<>();
for (String num: numbers) {
if (map1.containsKey(num)){
spelling.add(map1.get(num))
}
}
如何使用lambda表达式编写上述代码?
答案 0 :(得分:14)
使用Stream
:
List<String> spelling = numbers.stream()
.map(map1::get)
.filter(Objects::nonNull)
.collect(Collectors.toList());
System.out.println (spelling);
请注意,我没有使用containsKey
检查某个密钥是否在地图中,而是使用了get
,然后过滤掉了null
。
输出:
[One, Two, Three]
答案 1 :(得分:5)
Eran解决方案的变体:
containsKey
代替null
值=&gt;如果map1
包含null
值,则检查null
值会产生错误的结果。代码片段:
List<String> spelling = numbers.stream()
.filter(map1::containsKey)
.map(map1::get)
.collect(Collectors.toList());
System.out.println (spelling);
答案 2 :(得分:2)
另一种选择是使用forEach
构造:
numbers.forEach(n -> {
if(map1.containsKey(n))
spelling.add(map1.get(n));
});
答案 3 :(得分:1)
试试这个
List<String> spelling = map1.keySet().stream()
.filter(numbers::contains)
.map(map1::get)
.collect(Collectors.toList());