使用Java 8流计算平均值

Question

我是一名Java 8的新手，正在学习阶段。我需要使用Java 8编写以下逻辑。我有一个Json Array数据，我的json结构如下。

{"list":[{"core":{"min":281.556,"max":286.67,"top":972.73},"dt":"2017-02-16 12:00:00"},{"core":{"min":281.821,"max":285.66,"top":970.91},"dt":"2017-02-16 15:00:00"},{"core":{"min":274.498,"max":277.05,"top":970.44},"dt":"2017-02-16 18:00:00"},{"core":{"min":271.503,"max":272.78,"top":969.32},"dt":"2017-02-16 21:00:00"}]}

这个jsonArray中包含了大约100个jsonObject，每3小时就会有一组数据。我需要使用Java 8 stream或其他Java 8特性来分别找到最小值和最大值的平均值，并且还需要分别计算顶部的平均值。另一个条件是最小值和最大值的平均值应该基于白天或黑夜。如果dt包含06:00、09:00、12:00或15:00这些时间，那么它应该是日间平均值(白天最小值+最大值)，否则就应该是夜间平均值(黑夜最小值+最大值)。顶部不受白天或黑夜的影响。非常感谢您的帮助。

以下是使用Java 7进行登录。需要使用Java 8特性重新实现。

int dayavg=0;
        int nightavg = 0;
        int topavg=0;
        int day=0;
        int night = 0;
        int top=0;
    for(int i=0;i<50;i++){
        JsonNode node = list.get(i);
        String dt = node.get("dt").textValue();
        if(dt.contains("06:00:00")|| dt.contains("09:00:00") || dt.contains("12:00:00") || dt.contains("15:00:00")){
            int val = node.get("core").get("min").asInt() + node.get("core").get("max").asInt();
            day = day + val;
        }
        if(dt.contains("18:00:00")|| dt.contains("21:00:00") || dt.contains("00:00:00") || dt.contains("03:00:00")){
            int val = node.get("core").get("min").asInt() + node.get("core").get("max").asInt();
            night = night + val;
        }
        int val = node.get("core").get("top").asInt();
        top = top + val;
    }
    topavg = top/50;
    dayavg = day/100;
    nightavg = night/100;

- RK3

---

IntStream.range(0, 50).forEach(index -> list.get(index).get("dt_txt").textValue().concat("06:00:00") ); 我试图做类似的事情。使用通常的循环，我可以轻松地完成这个任务。只需从JSON中读取数据并进行平均即可。但需要使用Java 8功能来完成。 - RK3

已经更新问题并添加所需的逻辑。 - RK3

Answer 1

制作自己的收集器

如果您只想拥有一个流，则可以自己创建收集器。

使用这样的两个POJO：

class Values {
    int top,day,night;
}

class Avg {
    int topAvg,dayAvg,nightAvg;
}

收集器将是这样的：

    BinaryOperator<Values> combiner = (left,right) -> {
        left.day+=right.day;
        left.night+=right.night;
        left.top+=right.top;
        return left;
    };

    Function<Values,Avg> finisher =(value)->{
        Avg avg = new Avg();
        avg.topAvg = value.top/50;
        avg.dayAvg = value.day/100;
        avg.nightAvg = value.night/100;
        return avg;
    };


    BiConsumer<Values,JsonNode> accumulator= (values, node) ->{
        String dt = node.get("dt").textValue();
        if(dt.contains("06:00:00")|| dt.contains("09:00:00") || dt.contains("12:00:00") || dt.contains("15:00:00")){
            int val = node.get("core").get("min").asInt() + node.get("core").get("max").asInt();
            values.day +=  val;
        }
        if(dt.contains("18:00:00")|| dt.contains("21:00:00") || dt.contains("00:00:00") || dt.contains("03:00:00")){
            int val = node.get("core").get("min").asInt() + node.get("core").get("max").asInt();
            values.night += val;
        }
        int val = node.get("core").get("top").asInt();
        values.top += val;
    };

    Avg avg = list.stream()
        .collect(Collector.of(Values::new, accumulator,combiner,finisher));

按值使用一个流

你也可以使用多个流来计算dayAvg，类似以下这样：

   OptionalDouble dayAvg = list.stream()
        .filter(node -> {
            String dt = node.get("dt").textValue();
            return dt.contains("06:00:00") || dt.contains("09:00:00") || dt.contains("12:00:00") || dt.contains("15:00:00");
        })
        .mapToInt(node -> node.get("core").get("min").asInt() + node.get("core").get("max").asInt())
        .average();

我认为这样更清晰，但可能会降低方法的性能。

- Matthieu Gabin

Answer 2

您可以使用自定义的enum枚举类型：

static enum Type { DAY, NIGHT, TOP }

然后收集两倍的列表：

// Day and night
Map<Type, Double> result = list.stream()
.collect(
    Collectors.groupingBy(node -> {
        String dt = node.get("dt").textValue();
        if(dt.contains("06:00:00")|| dt.contains("09:00:00") || dt.contains("12:00:00") || dt.contains("15:00:00")){
            return Type.DAY;
        }
        if(dt.contains("18:00:00")|| dt.contains("21:00:00") || dt.contains("00:00:00") || dt.contains("03:00:00")){
            return Type.NIGHT;
        }
        // Should not happen
        return null;
    },
    Collectors.averagingInt(node -> node.get("core").get("min").asInt() + node.get("core").get("max").asInt()))
);

// Top
result.putAll(
     list.stream().collect(
         Collectors.groupingBy(node -> Type.TOP,
         Collectors.averagingInt(node -> node.get("core").get("top").asInt()))
     )
);

- ToYonos

---

我认为这是一个非常好的解决方案，我想我更愿意为Top值做一些类似于：result.put(Type.TOP,list.stream().collect(Collectors.averagingInt(node -> node.get("core").get("top").asInt()))); 因为Group By并不是真正有用的。 - Matthieu Gabin

是的，我同意，出于演示的目的，我想保持相同的逻辑两次。 - ToYonos