我编写了以下代码段来计算每个元素的出现次数。是否有可能以更短的方式实现这一目标?
int[] arr = {1, 6, 2, 8, 5, 4, 7, 7, 5, 7};
Arrays.stream(arr)
.collect(ArrayList::new, ArrayList::add, ArrayList::addAll)
.stream()
.collect(Collectors.groupingBy(s -> s))
.forEach((k, v) -> System.out.println(k+" "+v.size()));
另外,我想只显示超过1次的元素。所以我尝试修改如下导致错误。
.forEach((k, v) -> if(v.size() > 1) System.out.println(k+" "+v.size()));
这样做的正确方法是什么?
答案 0 :(得分:8)
对于后一个问题,你必须改变
.forEach((k, v) -> if(v.size() > 1) System.out.println(k+" "+v.size()));
到
.forEach((k, v) -> {if(v.size() > 1) System.out.println(k+" "+v.size());});
对于第一部分,不清楚为什么需要第一个collect
后跟第二个Stream
管道。
如果目的是将IntStream
转换为Stream<Integer>
,请使用boxed()
:
Arrays.stream(arr)
.boxed()
.collect(Collectors.groupingBy(s -> s))
.forEach((k, v) -> System.out.println(k+" "+v.size()));
正如Dici建议的那样,你也可以将收集者链接起来,将每个数字与其出现次数分组:
Map<Integer,Integer> occurrences =
Arrays.stream(arr)
.boxed()
.collect(Collectors.groupingBy(s -> s, Collectors.counting()));
答案 1 :(得分:3)
如果您愿意使用第三方库,Eclipse Collections的Bag
类型可以按以下方式使用:
Bags.mutable.with(1, 6, 2, 8, 5, 4, 7, 7, 5, 7)
.selectByOccurrences(count -> count > 1)
.forEachWithOccurrences((k, count) -> System.out.println(k+" "+count));
如果必须将int [] arr变量保存为int数组,则可以使用IntBag
。 更新:2016年9月9日:我已将selectByOccurrences
添加到原始Bags
,并在EC 8.0 release中提供。以下代码现在可以正常工作,而无需使用collect(i -> i)
:
IntBags.mutable.with(arr)
.selectByOccurrences(count -> count > 1)
.forEachWithOccurrences((k, count) -> System.out.println(k+" "+count));
注意:我是Eclipse Collections的提交者。
答案 2 :(得分:0)
我也希望分享我的解决方案!!!
// Solution 1 [Improved from Eran's solution & suggestion]
int[] arr = {1, 6, 2, 8, 5, 4, 7, 7, 5, 7};
Map<Integer, Long> counts = Arrays.stream(arr)
.boxed()
.collect(collectingAndThen(groupingBy(n -> n, counting()),
map -> map.entrySet().stream()
.filter(n -> n.getValue() > 1)
.collect(toMap(Entry::getKey, Entry::getValue))
));
System.out.println(counts.toString());
// Solution 2 [Improved from Dici's suggestion]
int[] arr = {1, 6, 2, 8, 5, 4, 7, 7, 5, 7};
Map<Object, Long> counts = Arrays.stream(arr)
.collect(ArrayList::new, ArrayList::add, ArrayList::addAll)
.stream()
.collect(groupingBy(Function.identity(), counting()));
counts.values().removeIf(count -> count < 2);
System.out.println(counts.toString());
答案 3 :(得分:0)
也可以使用频率来完成:
List<Integer> list = ImmutableList.of(1, 2, 3, 4, 5, 6, 3, 4, 5);
Map<Integer, Integer> result = list.stream().distinct().collect(Collectors.toMap(Function.identity(), token -> Collections.frequency(list, token)));
答案 4 :(得分:0)
List<Integer> numsList=Arrays.asList(1,2,3,5,2,4,3,1,2,2,4,4,5);
Map<Integer, Long> map=numsList.stream().collect(Collectors.groupingBy(Integer::intValue,Collectors.counting()));
map.forEach((k,v)->{System.out.println(k+" = "+v);});