Java - List를 Map으로 변환

By JS | Last updated: May 23, 2021

List를 Map으로 변환하는 방법을 소개합니다.

단순한 방법

먼저 다음과 같이 Id와 Value를 저장하는 Item 클래스가 있습니다.

public class Item {
    private int id;
    private String value;
    public Item(int id, String value) {
        this.id = id;
        this.value = value;
    }

    public int getId() {
        return id;
    }

    public String getValue() {
        return value;
    }
}

아래 코드는 for-loop를 이용하여 List를 Map으로 변환하는 예제입니다.

List<Item> list = new ArrayList<>();
list.add(new Item(1, "first"));
list.add(new Item(2, "second"));
list.add(new Item(3, "third"));

Map<Integer, String> map = new HashMap<>();
for (Item item : list) {
    map.put(item.getId(), item.getValue());
}

System.out.println(map);

Output:

{1=first, 2=second, 3=third}

Stream을 이용한 방법

Stream과 Collectors.toMap()을 이용하여 다음과 같이 List를 Map으로 변환할 수 있습니다.

List<Item> list = new ArrayList<>();
list.add(new Item(1, "first"));
list.add(new Item(2, "second"));
list.add(new Item(3, "third"));

Map<Integer, String> map = list.stream().collect(
        Collectors.toMap(Item::getId, Item::getValue));

System.out.println(map);

Output:

{1=first, 2=second, 3=third}

중복 문제 해결

위의 코드는 Stream을 통해 중복된 데이터가 Map에 저장될 때 문제가 발생합니다.

만약 다음과 같이 중복된 Id를 가진 데이터가 Map에 입력될 때, Runtime error가 발생합니다.

List<Item> list = new ArrayList<>();
list.add(new Item(1, "first"));
list.add(new Item(2, "second"));
list.add(new Item(1, "third"));

Map<Integer, String> map = list.stream().collect(
        Collectors.toMap(Item::getId, Item::getValue));

toMap()(oldId, newId) -> oldId를 세번째 인자로 전달하여 문제를 해결할 수 있습니다. 이 코드는, 동일한 key의 데이터가 입력될 때, 이전에 저장된 oldId의 데이터를 유지하겠다는 의미입니다.

List<Item> list = new ArrayList<>();
list.add(new Item(1, "first"));
list.add(new Item(2, "second"));
list.add(new Item(1, "third"));

Map<Integer, String> map = list.stream().collect(
        Collectors.toMap(Item::getId,
                Item::getValue,
                (oldId, newId) -> oldId
        ));

System.out.println(map);

Output:

{1=first, 2=second}

만약 (oldId, newId) -> newId로 인자를 전달하면, 이전 데이터는 삭제되고, 새로운 데이터로 덮어써 집니다.

Map<Integer, String> map = list.stream().collect(
        Collectors.toMap(Item::getId,
                Item::getValue,
                (oldId, newId) -> newId
        ));

Output:

{1=third, 2=second}

원하는 Map으로 변환

Collectors.toMap()은 기본적으로 HashMap을 생성합니다. 만약 ConcurrentHashMap으로 Map을 만들고 싶다면 아래와 같이 ConcurrentHashMap::new를 인자로 전달하면 됩니다. 다른 Map 객체들도 방법은 동일합니다.

Map<Integer, String> map = list.stream().collect(
        Collectors.toMap(Item::getId,
                Item::getValue,
                (oldId, newId) -> oldId,
                ConcurrentHashMap::new
        ));

Map 정렬

LinkedHashMap은 입력된 순서를 보장하는 Map입니다. 만약 sorted()로 Stream 데이터를 정렬하고 Map에 저장하면 정렬된 순서가 유지됩니다.

아래 코드는 Id의 역순으로 정렬된 데이터를 Map에 저장하는 코드입니다.

List<Item> list = new ArrayList<>();
list.add(new Item(1, "first"));
list.add(new Item(2, "second"));
list.add(new Item(3, "third"));

Map<Integer, String> map = list.stream()
        .sorted(Comparator.comparingLong(Item::getId).reversed())
        .collect(Collectors.toMap(Item::getId,
                    Item::getValue,
                    (oldId, newId) -> oldId,
                    LinkedHashMap::new
                ));

System.out.println(map);

Output:

{3=third, 2=second, 1=first}

Related Posts

댓글을 보거나 쓰려면 이 버튼을 눌러주세요.
codechachaCopyright ©2019 codechacha