Java: как преобразовать список в карту

Question

Недавно я поговорил с коллегой о том, какой оптимальный способ конвертировать List в Map в Java и есть ли какие-то конкретные преимущества в этом.

Я хочу знать оптимальный подход к конверсии и был бы очень признателен, если бы кто-нибудь мог мне помочь.

Это хороший подход:

List<Object[]> results;
Map<Integer, String> resultsMap = new HashMap<Integer, String>();
for (Object[] o : results) {
    resultsMap.put((Integer) o[0], (String) o[1]);
}

Answer 1

С java-8 вы сможете сделать это в одну строку, используя streams и класс Collectors.

Map<String, Item> map = 
    list.stream().collect(Collectors.toMap(Item::getKey, item -> item));

Короткая демонстрация:

import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

public class Test{
    public static void main (String [] args){
        List<Item> list = IntStream.rangeClosed(1, 4)
                                   .mapToObj(Item::new)
                                   .collect(Collectors.toList()); //[Item [i=1], Item [i=2], Item [i=3], Item [i=4]]

        Map<String, Item> map = 
            list.stream().collect(Collectors.toMap(Item::getKey, item -> item));

        map.forEach((k, v) -> System.out.println(k + " => " + v));
    }
}
class Item {

    private final int i;

    public Item(int i){
        this.i = i;
    }

    public String getKey(){
        return "Key-"+i;
    }

    @Override
    public String toString() {
        return "Item [i=" + i + "]";
    }
}

Выход:

Key-1 => Item [i=1]
Key-2 => Item [i=2]
Key-3 => Item [i=3]
Key-4 => Item [i=4]

Как отмечено в комментариях, вы можете использовать Function.identity() вместо item -> item, хотя я нахожу i -> i довольно явным.

И для полноты обратите внимание, что вы можете использовать бинарный оператор, если ваша функция не является биективной. Например, давайте рассмотрим это List и функцию отображения, которая для значения int вычисляет результат по модулю 3:

List<Integer> intList = Arrays.asList(1, 2, 3, 4, 5, 6);
Map<String, Integer> map = 
    intList.stream().collect(toMap(i -> String.valueOf(i % 3), i -> i));

При запуске этого кода вы получите сообщение об ошибке java.lang.IllegalStateException: Duplicate key 1. Это связано с тем, что 1% 3 совпадает с 4% 3 и, следовательно, имеет то же значение ключа, что и функция сопоставления клавиш. В этом случае вы можете предоставить оператор слияния.

Вот тот, который суммирует значения; (i1, i2) -> i1 + i2;, который можно заменить ссылкой на метод Integer::sum.

Map<String, Integer> map = 
    intList.stream().collect(toMap(i -> String.valueOf(i % 3), 
                                   i -> i, 
                                   Integer::sum));

который теперь выводит:

0 => 9 (i.e 3 + 6)
1 => 5 (i.e 1 + 4)
2 => 7 (i.e 2 + 5)

Надеюсь, это поможет! :)

Answer 2

List<Item> list;
Map<Key,Item> map = new HashMap<Key,Item>();
for (Item i : list) map.put(i.getKey(),i);

Предположим, конечно, что у каждого элемента есть метод getKey(), который возвращает ключ правильного типа.

Answer 3

На всякий случай, если этот вопрос не закрыт как дубликат, правильный ответ - использовать Google Collections :

Map<String,Role> mappedRoles = Maps.uniqueIndex(yourList, new Function<Role,String>() {
  public String apply(Role from) {
    return from.getName(); // or something else
  }});

Answer 4

Начиная с Java 8, ответ @ ZouZou с использованием коллектора Collectors.toMap, безусловно, является идиоматическим способом решения этой проблемы.

И так как это такая общая задача, мыможет превратить его в статическую утилиту.

Таким образом, решение действительно становится однострочным.

/**
 * Returns a map where each entry is an item of {@code list} mapped by the
 * key produced by applying {@code mapper} to the item.
 *
 * @param list the list to map
 * @param mapper the function to produce the key from a list item
 * @return the resulting map
 * @throws IllegalStateException on duplicate key
 */
public static <K, T> Map<K, T> toMapBy(List<T> list,
        Function<? super T, ? extends K> mapper) {
    return list.stream().collect(Collectors.toMap(mapper, Function.identity()));
}

И вот как вы можете использовать его на List<Student>:

Map<Long, Student> studentsById = toMapBy(students, Student::getId);

Answer 5

Используя Java 8, вы можете делать следующее:

Map<Key, Value> result= results
                       .stream()
                       .collect(Collectors.toMap(Value::getName,Function.identity()));

Value может быть любым объектом, который вы используете.

Answer 6

A List и Map концептуально различны. List - это упорядоченная коллекция предметов. Элементы могут содержать дубликаты, и элемент может не иметь понятия уникального идентификатора (ключа). A Map имеет значения, сопоставленные с ключами. Каждая клавиша может указывать только на одно значение.

Поэтому, в зависимости от предметов вашего List, может быть или не быть возможным преобразовать его в Map. У ваших List предметов нет дубликатов? У каждого предмета есть уникальный ключ? Если так, то их можно поместить в Map.

Answer 7

Существует также простой способ сделать это, используя Maps.uniqueIndex (...) из Google гуава библиотеки

Answer 8

Алексис уже опубликовал ответ в Java 8 , используя метод toMap(keyMapper, valueMapper). Согласно документ для реализации этого метода:

Нет никаких гарантий относительно типа, изменчивости, сериализуемости или потокобезопасность карты вернулась.

Так что в случае, если нас интересует конкретная реализация интерфейса Map, например HashMap тогда мы можем использовать перегруженную форму как:

Map<String, Item> map2 =
                itemList.stream().collect(Collectors.toMap(Item::getKey, //key for map
                        Function.identity(),    // value for map
                        (o,n) -> o,             // merge function in case of conflict with keys
                        HashMap::new));         // map factory - we want HashMap and not any Map implementation

Хотя использование Function.identity() или i->i хорошо, но кажется, Function.identity() вместо i -> i может сэкономить некоторую память в соответствии с этим answer .

Answer 9

Универсальный метод

public static <K, V> Map<K, V> listAsMap(Collection<V> sourceList, ListToMapConverter<K, V> converter) {
    Map<K, V> newMap = new HashMap<K, V>();
    for (V item : sourceList) {
        newMap.put( converter.getKey(item), item );
    }
    return newMap;
}

public static interface ListToMapConverter<K, V> {
    public K getKey(V item);
}

Answer 10

Без Java-8 вы сможете сделать это в одной строке коллекций Commons и класса Closure

List<Item> list;
@SuppressWarnings("unchecked")
Map<Key, Item> map  = new HashMap<Key, Item>>(){{
    CollectionUtils.forAllDo(list, new Closure() {
        @Override
        public void execute(Object input) {
            Item item = (Item) input;
            put(i.getKey(), item);
        }
    });
}};