使用Java 8流来聚合列表对象

时间:2017-05-06 07:36:55

标签: java-8

我们使用3个列表ListA,ListB,ListC来保留3个科目(A,B,C)中10名学生的分数。

受试者B和C是可选的,因此只有少数学生中有10名在这些科目中有分数

Class Student{
String studentName;
int marks;
}

ListA有10名学生的记录,ListB为5,ListC为3(也是名单的大小)

想知道我们如何使用java 8 steam总结学生对其科目的分数。

我尝试了以下

List<Integer> list = IntStream.range(0,listA.size() -1).mapToObj(i -> listA.get(i).getMarks() + 
listB.get(i).getMarks() + 
listC.get(i).getMarks()).collect(Collectors.toList());;

这个

有两个问题

a)它会将IndexOutOfBoundsException作为listB,而listC没有10个元素

b)返回的列表,如果是Integer类型,我希望它是Student类型。

任何输入都会非常有用

2 个答案:

答案 0 :(得分:5)

您可以制作3个列表的流,然后调用flatMap将所有列表的元素放入单个流中。该流将为每个学生每个标记包含一个元素,因此您必须按学生姓名汇总结果。有点像:

Map<String, Integer> studentMap = Stream.of(listA, listB, listC)
        .flatMap(Collection::stream)
        .collect(groupingBy(student -> student.name, summingInt(student -> student.mark)));

或者,如果您的Student类的字段包含getter,则可以更改最后一行以使其更具可读性:

Map<String, Integer> studentMap = Stream.of(listA, listB, listC)
        .flatMap(Collection::stream)
        .collect(groupingBy(Student::getName, summingInt(Student::getMark)));

然后打印出studentMap

来检查结果
studentMap.forEach((key, value) -> System.out.println(key + " - " + value));

如果要创建Student对象列表,可以使用第一个映射的结果并从其条目创建新流(此特定示例假定您的Student类具有all-args构造函数,所以你可以单行它:)

List<Student> studentList = Stream.of(listA, listB, listC)
        .flatMap(Collection::stream)
        .collect(groupingBy(Student::getName, summingInt(Student::getMark)))
        .entrySet().stream()
            .map(mapEntry -> new Student(mapEntry.getKey(), mapEntry.getValue()))
            .collect(toList());

答案 1 :(得分:1)

我会这样做:

Map<String, Student> result = Stream.of(listA, listB, listC)
    .flatMap(List::stream)
    .collect(Collectors.toMap(
        Student::getName, // key: student's name
        s -> new Student(s.getName(), s.getMarks()), // value: new Student
        (s1, s2) -> { // merge students with same name: sum marks
            s1.setMarks(s1.getMarks() + s2.getMarks());
            return s1;
        }));

这里我使用了Collectors.toMap来创建地图(我还假设你有一个Student的构造函数,它接收一个名称和标记。)

此版本的Collectors.toMap需要三个参数:

  1. 返回每个元素的键的函数(此处为Student::getName
  2. 一个返回每个元素值的函数(我创建了一个新的Student实例,它是原始元素的副本,这是为了不修改原始流中的实例)
  3. 当存在具有相同键的元素时使用的合并功能,即对于具有相同名称的学生(我在此处总结了这些标记)。
  4. 如果您可以将以下复制构造函数和方法添加到Student类:

    public Student(Student another) {
        this.name = another.name;
        this.marks = another.marks;
    }
    
    public Student merge(Student another) {
        this.marks += another.marks;
        return this;
    }
    

    然后你可以用这种方式重写上面的代码:

    Map<String, Student> result = Stream.of(listA, listB, listC)
        .flatMap(List::stream)
        .collect(Collectors.toMap(
            Student::getName,
            Student::new,
            Student::merge));