我如何用groupBy计算发生的事件?

我想将流中的项收集到一个映射中,该映射将相等的对象分组在一起,并映射到出现的数量。

List<String> list = Arrays.asList("Hello", "Hello", "World");
Map<String, Long> wordToFrequency = // what goes here?

所以在这种情况下,我想地图由这些条目组成:

Hello -> 2
World -> 1

我该怎么做呢?

189910 次浏览

我认为你只是在寻找过载,它需要另一个Collector来指定对每个组做什么…然后Collectors.counting()来进行计数:

import java.util.*;
import java.util.stream.*;


class Test {
public static void main(String[] args) {
List<String> list = new ArrayList<>();


list.add("Hello");
list.add("Hello");
list.add("World");


Map<String, Long> counted = list.stream()
.collect(Collectors.groupingBy(Function.identity(), Collectors.counting()));


System.out.println(counted);
}
}

结果:

{Hello=2, World=1}

(也有可能使用groupingByConcurrent来提高效率。如果在您的上下文中安全的话,那么在您的实际代码中需要记住这一点。)

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


list.add("Hello");
list.add("Hello");
list.add("World");


Map<String, List<String>> collect = list.stream()
.collect(Collectors.groupingBy(o -> o));
collect.entrySet()
.forEach(e -> System.out.println(e.getKey() + " - " + e.getValue().size()));

下面是StreamEx的简单解决方案:

StreamEx.of(list).groupingBy(Function.identity(), MoreCollectors.countingInt());

这样做的好处是减少Java流样板代码:collect(Collectors.

下面是一个对象列表的例子

Map<String, Long> requirementCountMap = requirements.stream().collect(Collectors.groupingBy(Requirement::getRequirementType, Collectors.counting()));

这里有一些稍微不同的选项来完成手头的任务。

使用toMap:

list.stream()
.collect(Collectors.toMap(Function.identity(), e -> 1, Math::addExact));

使用Map::merge:

Map<String, Integer> accumulator = new HashMap<>();
list.forEach(s -> accumulator.merge(s, 1, Math::addExact));

如果你愿意使用第三方库,你可以使用Eclipse集合中的Collectors2类使用StreamList转换为BagBag是一个数据结构,即为计数而建

Bag<String> counted =
list.stream().collect(Collectors2.countBy(each -> each));


Assert.assertEquals(1, counted.occurrencesOf("World"));
Assert.assertEquals(2, counted.occurrencesOf("Hello"));


System.out.println(counted.toStringOfItemToCount());

输出:

{World=1, Hello=2}

在这种特殊情况下,你可以简单地将collectList直接转换为Bag

Bag<String> counted =
list.stream().collect(Collectors2.toBag());

你也可以在不使用Stream的情况下创建Bag,方法是在Eclipse Collections协议中调整List

Bag<String> counted = Lists.adapt(list).countBy(each -> each);

或者在这个特殊情况下:

Bag<String> counted = Lists.adapt(list).toBag();

您也可以直接创建Bag。

Bag<String> counted = Bags.mutable.with("Hello", "Hello", "World");

Bag<String>类似于Map<String, Integer>,因为它在内部跟踪键及其计数。但是,如果你向Map对象请求一个它不包含的键,它将返回null。如果你使用occurrencesOfBag对象请求一个它不包含的键,它将返回0。

注意:我是Eclipse集合的提交者。