假设我有一个包含每个类都有一个Map
的List
。
public class Test {
public Map<Long, Integer> map;
}
Map
中的Long
键是时间戳,Integer
值是分数。
我正在尝试创建一个Stream
,可以将所有对象的映射组合并输出一个具有唯一时间戳(即Long
)和平均分数的Map
。
我有这段代码,但它给我所有分数的总和而不是平均值(Integer
类没有平均方法)。
Test test1 = new Test();
test1.map = new HashMap() {{
put(1000L, 1);
put(2000L, 2);
put(3000L, 3);
}};
Test test2 = new Test();
test2.map = new HashMap() {{
put(1000L, 10);
put(2000L, 20);
put(3000L, 30);
}};
List<Test> tests = new ArrayList() {{
add(test1);
add(test2);
}};
Map<Long, Integer> merged = tests.stream()
.map(test -> test.map)
.map(Map::entrySet)
.flatMap(Collection::stream)
.collect(
Collectors.toMap(
Map.Entry::getKey,
Map.Entry::getValue,
Integer::sum
)
);
System.out.println(merged);
我认为这可能不是一个单一的Stream
可以轻松解决的问题,因此输出一个包含唯一时间戳和所有分数列表的Map
也是可以的。然后我可以自己计算平均值。
Map<Long, List<Integer>>
可以吗?
.flatMap(test -> test.map.entrySet().stream())
比.map(test -> test.map) .map(Map::entrySet) .flatMap(Collection::stream)
更简单。 - Holger