您可以为此任务创建自己的收集器:
static Collector averagingWeighted(ToDoubleFunction valueFunction, ToIntFunction weightFunction) {
class Box {
double num = 0;
long denom = 0;
}
return Collector.of(
Box::new,
(b, e) -> {
b.num += valueFunction.applyAsDouble(e) * weightFunction.applyAsInt(e);
b.denom += weightFunction.applyAsInt(e);
},
(b1, b2) -> { b1.num += b2.num; b1.denom += b2.denom; return b1; },
b -> b.num / b.denom
);
}
此自定义收集器将两个函数作为参数:一个是返回用于给定流元素的值的函数(作为ToDoubleFunction),另一个函数返回权重(作为ToIntFunction).它使用一个辅助本地类,在收集过程中存储分子和分母.每次接受一个条目时,分子会随着值与其权重的乘积而增加,分母会随着权重而增加.然后整理器将两者的分割返回为Double.
示例用法是:
Map map = new HashMap<>();
map.put(0.7, 100);
map.put(0.5, 200);
double weightedAverage =
map.entrySet().stream().collect(averagingWeighted(Map.Entry::getKey, Map.Entry::getValue));