java 加权平均_使用Java 8流计算加权平均值

您可以为此任务创建自己的收集器：

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));