多个对象属性Stream的累加和

问题描述 投票:0回答:4

我有一个按月年字符串属性排序的对象列表。 我的对象类定义看起来像

Public class Obj{
    String year;
    Long membercount;
    Long nonmembercount;
    Double memberpayment;
    Double nonmemberpayment;
}

new Obj("9-2015",100,20,10,5)
new Obj("10-2015",220,40,20,55)
new Obj("11-2015",300,60,30,45)
new Obj("12-2015",330,30,50,6)
new Obj("1-2016",100,10,10,4)

我想对

membercount
,
nonmembercount
,
memberpayment
,
nonmemberpayment

进行累计求和

所以我的新对象列表如下所示

new Obj("9-2015",100,20,10,5)
new Obj("10-2015",320,60,30,60)
new Obj("11-2015",620,120,60,105)
new Obj("12-2015",950,150,110,111)
new Obj("1-2016",1050,160,120,115)

我尝试使用

Collectors.summingDouble
,但它给了我所有的总和而不是累积的。

非常感谢您的指点。

java java-8 java-stream
4个回答
4
投票

Stream API 中不直接支持累积操作,尽管可以通过自定义实现此类操作

Collector
。但值得注意的是,已经有对数组上此类操作的直接支持,这可能足以满足您的情况:

Obj
的草图扩展为

public class Obj {
    String year;
    Long membercount;
    Long nonmembercount;
    Double memberpayment;
    Double nonmemberpayment;

    public Obj(String year, long membercount, long nonmembercount,
            double memberpayment, double nonmemberpayment) {
        this.year = year;
        this.membercount = membercount;
        this.nonmembercount = nonmembercount;
        this.memberpayment = memberpayment;
        this.nonmemberpayment = nonmemberpayment;
    }

    @Override
    public String toString() {
        return "Obj("+year+", "+membercount+", "+nonmembercount
            +", "+memberpayment+", "+nonmemberpayment+')';
    }
}

解决方案可能如下所示:

// test data
List<Obj> list=Arrays.asList(
    new Obj("9-2015", 100, 20, 10, 5),
    new Obj("10-2015", 220, 40, 20, 55),
    new Obj("11-2015", 300, 60, 30, 45),
    new Obj("12-2015", 330, 30, 50, 6),
    new Obj("1-2016", 100, 10, 10, 4));

// creating an array as need for the operation, it will contain the
// result afterwards, whereas the source list is not modified
Obj[] array = list.toArray(new Obj[0]);

// the actual operation
Arrays.parallelPrefix(array, (a,b) -> new Obj(b.year,
    a.membercount + b.membercount,
    a.nonmembercount + b.nonmembercount,
    a.memberpayment + b.memberpayment,
    a.nonmemberpayment + b.nonmemberpayment
));

// just print the result
Arrays.asList(array).forEach(System.out::println);

最后一行将打印

Obj(9-2015, 100, 20, 10.0, 5.0)
Obj(10-2015, 320, 60, 30.0, 60.0)
Obj(11-2015, 620, 120, 60.0, 105.0)
Obj(12-2015, 950, 150, 110.0, 111.0)
Obj(1-2016, 1050, 160, 120.0, 115.0)

虽然此操作不太可能从如此少量元素的并行处理中受益,但遗憾的是此操作没有顺序版本。所以你可能会考虑使用普通的循环解决方案......


为了完整起见,这里是一个基于流收集器的累积操作解决方案。与

Arrays.parallelPrefix
一样,更新函数必须无副作用且具有关联性,函数返回具有汇总属性的新对象就是这种情况。

public static <T> Collector<T,?,List<T>> cumulative(BinaryOperator<T> update) {
    return Collector.of(ArrayList::new,
        (l,o) -> {
            if(!l.isEmpty()) o=update.apply(l.get(l.size()-1), o);
            l.add(o);
        },
        (l,m) -> {
            if(l.isEmpty()) return m;
            if(!m.isEmpty()) {
                T a = l.get(l.size()-1);
                for(T b: m) l.add(update.apply(a, b));
            }
            return l;
        });
}

按照上述设置使用它:

List<Obj> result = list.stream().collect(cumulative((a,b) -> new Obj(b.year,
    a.membercount + b.membercount,
    a.nonmembercount + b.nonmembercount,
    a.memberpayment + b.memberpayment,
    a.nonmemberpayment + b.nonmemberpayment
)));

0
投票

我的解决方案:

List<Obj> objects = ...;
int cumulSum[] = {0};

objects.stream().map(obj -> {
    cumulSum[0] += obj.membercount;
    return new Obj(obj.year, cumulSum[0], ...);
});

0
投票

根据我对你问题的理解,这就是我能想到的。

我使用

map(o->o.membercount)
collect(Collectors.summingDouble(n->n))
来对每个字段进行计算。看看吧。

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

public class DoubleSummation {

    public static void main(String[] args) {
        List<Obj> list = new ArrayList<>();
        list.add(new Obj("9-2015",100,20,10,5));
        list.add(new Obj("10-2015",220,40,20,55));
        list.add(new Obj("11-2015",300,60,30,45));
        list.add(new Obj("12-2015",330,30,50,6));
        list.add(new Obj("1-2016",100,10,10,4));

        Double sumMemberCount = list.stream().mapToDouble(o->o.membercount).sum();
        Double sumNonmembercount = list.stream().mapToDouble(o->o.nonmembercount).sum();
        Double sumMemberpayment = list.stream().mapToDouble(o->o.memberpayment).sum();
        Double sumNonmemberpayment = list.stream().mapToDouble(o->o.nonmemberpayment).sum();

        System.out.println(sumMemberCount);
        System.out.println(sumNonmembercount);
        System.out.println(sumMemberpayment);
        System.out.println(sumNonmemberpayment);
    }
}

class Obj{

    String year;
    long membercount;
    long nonmembercount;
    double memberpayment;
    double nonmemberpayment;

    public Obj(String year, long membercount, long nonmembercount, double memberpayment, double nonmemberpayment) {
        this.year = year;
        this.membercount = membercount;
        this.nonmembercount = nonmembercount;
        this.memberpayment = memberpayment;
        this.nonmemberpayment = nonmemberpayment;
    }

}

希望这有帮助!


0
投票

这里是使用 JEP 461:Stream Gatherers Java 22 预览语言功能的解决方案,它添加了对此类累积积累的内置支持:

List<Obj> summed = list.stream()
        .gather(Gatherers.scan(
                () -> new Obj("", 0, 0, 0, 0),
                (o1, o2) -> new Obj(
                        o2.year,
                        o1.membercount + o2.membercount,
                        o1.nonmembercount + o2.nonmembercount,
                        o1.memberpayment + o2.memberpayment,
                        o1.nonmemberpayment + o2.nonmemberpayment
                )))
        .toList();

这使用新的

Stream.gather
方法和新的内置
Gatherers.scan
收集器将初始流转换为累积和流。

Java文档

Gatherer

将输入元素流转换为输出元素流的中间操作,可以选择在到达上游末尾时应用最终操作。 […]

[…]

聚集操作的例子有很多,包括但不限于:将元素分组(窗口函数);对连续相似的元素进行去重;增量累加功能(前缀扫描);增量重新排序功能等。类

Gatherers
提供了常见收集操作的实现。

Stream.gather

返回一个流,其中包含将给定收集器应用于该流的元素的结果。

Gatherers.scan

返回一个使用提供的函数执行前缀扫描(增量累积)的收集器。从从

Supplier
获得的初始值开始,通过将
BiFunction
应用于当前值和下一个输入元素来获得每个后续值,之后在下游生成结果值。

示例:

// will contain: ["1", "12", "123", "1234", "12345", "123456", "1234567", "12345678", "123456789"]
List<String> numberStrings =
    Stream.of(1,2,3,4,5,6,7,8,9)
          .gather(
              Gatherers.scan(() -> "", (string, number) -> string + number)
           )
          .toList();
© www.soinside.com 2019 - 2024. All rights reserved.