我们使用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类型。
任何输入都会非常有用
答案 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
需要三个参数:
Student::getName
)Student
实例,它是原始元素的副本,这是为了不修改原始流中的实例)如果您可以将以下复制构造函数和方法添加到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));