Javaでネストされたコレクションをフラット化する

投稿日: 2019-10-18 2022-10-12
タグ: Java, Java Collections, Java List

1. 概要

このクイック記事では、Javaでネストされたコレクションをフラット化する方法について説明します。

2. ネストされたコレクションの例

タイプStringのリストのリストがあるとします。

List<List<String>> nestedList = asList(
  asList("one:one"), 
  asList("two:one", "two:two", "two:three"), 
  asList("three:one", "three:two", "three:three", "three:four"));

**3. リストをforEachでフラット化**

このネストされたコレクションを文字列のリストにフラット化するために、forEachをJava8メソッドリファレンスと一緒に使用できます。

public <T> List<T> flattenListOfListsImperatively(
    List<List<T>> nestedList) {
    List<T> ls = new ArrayList<>();
    nestedList.forEach(ls::addAll);
    return ls;
}

そして、ここでメソッドの動作を確認できます。

@Test
public void givenNestedList_thenFlattenImperatively() {
    List<String> ls = flattenListOfListsImperatively(nestedList);
    
    assertNotNull(ls);
    assertTrue(ls.size() == 8);
    assertThat(ls, IsIterableContainingInOrder.contains(
      "one:one",
      "two:one", "two:two", "two:three", "three:one",
      "three:two", "three:three", "three:four"));
}

**4. リストをflatMapでフラット化**

StreamAPIのflatMapメソッドを利用して、ネストされたリストをフラット化することもできます。

これにより、ネストされた Stream 構造をフラット化し、最終的にすべての要素を特定のコレクションに収集できます。

public <T> List<T> flattenListOfListsStream(List<List<T>> list) {
    return list.stream()
      .flatMap(Collection::stream)
      .collect(Collectors.toList());    
}

そして、これが実際のロジックです：

@Test
public void givenNestedList_thenFlattenFunctionally() {
    List<String> ls = flattenListOfListsStream(nestedList);
    
    assertNotNull(ls);
    assertTrue(ls.size() == 8);
}

5. 結論

Java8の単純なforEachまたはflatMapメソッドをメソッド参照と組み合わせて使用すると、ネストされたコレクションをフラット化できます。

この記事で説明されているコードは、GitHubにあります。

getdocs

13036