How to get a Set from a list of objects using Java Streams

Question

This might be a simple Java streams question. Say, I have a List<Student> object.

public class Student {
    public String name;
    public Set<String> subjects;

    public Set<String> getSubjects() {
        return subjects;
    }
}

How can I get all the subjects taken by the list of students?

I can do this using a for each loop. How can I convert the below code to use Streams?

for (Student student : students) {
    subjectsTaken.addAll(student.getSubjects());
}

Here is my attempt at using Java 8 streams. This gives me an Incompatible types error.

Set<String> subjectsTaken = students.stream()
        .map(student -> student.getSubjects())
        .collect(Collectors.toSet());

Eran · Accepted Answer · 2018-12-17 13:28:04Z

6

Your current code produces a Set<Set<String>>, not a Set<String>.

You should use flatMap, not map:

Set<String> subjectsTaken = 
    students.stream() // Stream<Student>
           .flatMap(student -> student.getSubjects().stream()) // Stream<String>
           .collect(Collectors.toSet()); // Set<String>

answered Dec 17, 2018 at 13:28

Eran

395k57 gold badges726 silver badges793 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

ETO · Accepted Answer · 2018-12-17 13:50:21Z

5

Try this:

Set<String> subjectsTaken = 
                   students.stream()
                           .map(Student::getSubjects)
                           .flatMap(Set::stream) 
                           .collect(Collectors.toSet());

The idea is to map the students to their subjects first, then flatten the Stream<Set<String>> to Stream<String> and finally collect the stream to a Set.

I would suggest you to use method references instead of lambda expressions where it's possible (if it doesn't degrade readability).

edited Dec 17, 2018 at 13:50

answered Dec 17, 2018 at 13:34

ETO

7,3471 gold badge23 silver badges40 bronze badges

Comments

HPH · Accepted Answer · 2018-12-17 14:18:30Z

1

A different alternative using Stream<T>#<R>collect :

students.stream()
    .map(Student::getSubjects)
    .<Set<String>>collect(HashSet::new, Collection::addAll, Collection::addAll)

answered Dec 17, 2018 at 14:18

HPH

3982 silver badges11 bronze badges

Collectives™ on Stack Overflow

How to get a Set from a list of objects using Java Streams

3 Answers 3

Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related