Map to List after filtering on Map's key using Java8 stream

Question

I have a Map<String, List<String>>. I want to transform this map to a List after filtering on the map's key.

Example:

Map<String, List<String>> words = new HashMap<>();
List<String> aList = new ArrayList<>();
aList.add("Apple");
aList.add("Abacus");

List<String> bList = new ArrayList<>();
bList.add("Bus");
bList.add("Blue");
words.put("A", aList);
words.put("B", bList);

Given a key, say, "B"

Expected Output: ["Bus", "Blue"]

This is what I am trying:

 List<String> wordsForGivenAlphabet = words.entrySet().stream()
    .filter(x-> x.getKey().equalsIgnoreCase(inputAlphabet))
    .map(x->x.getValue())
    .collect(Collectors.toList());

I am getting an error. Can someone provide me with a way to do it in Java8?

Why don't you just use map.get(inputAlphabet.toUpperCase())? — fps
– fps, Commented Jul 19, 2017 at 3:37
@FedericoPeraltaSchaffner I assume because the Map might contains lower case letters also in general, this being just an example... — Eugene
– Eugene, Commented Jul 19, 2017 at 6:41
@Eugene Yes, I also think that this is a simplified example. I just commented so that OP has the chance to improve it. — fps
– fps, Commented Jul 19, 2017 at 14:35

Beri · Accepted Answer · 2018-04-12 05:51:01Z

17

Your sniplet wil produce a List<List<String>> not List<String>.

You are missing flatMap , that will convert stream of lists into a single stream, so basically flattens your stream:

List<String> wordsForGivenAlphabet = words.entrySet().stream()
    .filter(x-> x.getKey().equalsIgnoreCase(inputAlphabet))
    .map(Map.Entry::getValue)
    .flatMap(List::stream) 
    .collect(Collectors.toList());

You can also add distinct(), if you don't want values to repeat.

edited Apr 12, 2018 at 5:51

answered Jul 18, 2017 at 21:17

Beri

11.6k4 gold badges38 silver badges59 bronze badges

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

6 Comments

RoyalTiger Over a year ago

I am getting "not-static method cannot be referenced from a static context" compilation error on .flatMap(List::stream) line

Beri Over a year ago

Is your map value of type List,Set? Here words is of Map<String, List<String>> type. What is your words collection type?

RoyalTiger Over a year ago

I have Map<String, Object>. I tried the same code as mentioned above and getting the compilation error.

RoyalTiger Over a year ago

This Map is coming from another class and I know the values will always be List<String> for a specific key.

RoyalTiger Over a year ago

I tried below and it worked. Is it because fo casting?

List<String> l =  map.entrySet().stream()             .map(Map.Entry::getValue)             .filter(c -> c instanceof Collection)             .map(c -> (Collection<String>)c)             .flatMap(Collection::stream)             .collect(Collectors.toList());

|

Eugene · Accepted Answer · 2017-07-19 06:55:51Z

4

Federico is right in his comment, if all you want is to get the values of a certain key (inside a List) why don't you simply do a get (assuming all your keys are uppercase letters already) ?

 List<String> values = words.get(inputAlphabet.toUpperCase());

If on the other hand this is just to understand how stream operations work, there is one more way to do it (via java-9 Collectors.flatMapping)

List<String> words2 = words.entrySet().stream()
            .collect(Collectors.filtering(x -> x.getKey().equalsIgnoreCase(inputAlphabet),
                    Collectors.flatMapping(x -> x.getValue().stream(), 
                          Collectors.toList())));

answered Jul 19, 2017 at 6:55

Eugene

122k17 gold badges219 silver badges335 bronze badges

1 Comment

fps Over a year ago

Very nice to use JDK 9 collector's new features

Vitalii Muzalevskyi · Accepted Answer · 2017-07-18 21:38:07Z

0

As was previously told after collect you will get List<List<String>> with only one or zero value in it. You can use findFirst instead of collect it will return you Optional<List<String>>.

edited Jul 18, 2017 at 21:38

answered Jul 18, 2017 at 21:30

Vitalii Muzalevskyi

6728 silver badges18 bronze badges

2 Comments

user4910279 Over a year ago

When the map has "B" and "b" as keys, you get only one list.

Vitalii Muzalevskyi Over a year ago

Yes, I did not pay attention to equalsIgnoreCase, if it is importand my solution won't work properly.

friend · Accepted Answer · 2023-12-07 22:19:17Z

0

List<String> items = words.entrySet().stream()
  .filter(entry -> entry.getKey().equals("B"))
  .flatMap(entry -> entry.getValue().stream())
  .collect(Collectors.toList());
  System.out.println(items);

answered Dec 7, 2023 at 22:19

friend

1

1 Comment

dan1st Over a year ago

While this code may answer the question, adding a bit of explanation would greatly benefit it, especially for future readers.

Collectives™ on Stack Overflow

Map to List after filtering on Map's key using Java8 stream

4 Answers 4

6 Comments

1 Comment

2 Comments

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

4 Answers 4

6 Comments

1 Comment

2 Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related