Ява: як перетворити список на карту

Нещодавно я спілкувався з колегою про те, що було б оптимальним способом перетворення Listна MapJava та чи є якісь переваги від цього.

Я хочу знати оптимальний підхід до перетворення і дуже вдячний, якщо хтось може мене керувати.

Це хороший підхід:

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

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

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

З java-8, ви зможете зробити це в одному рядку, використовуючи потоки та 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)

Сподіваюся, це допомагає! :)

На випадок, якщо це запитання не буде закрито як дублікат, правильною відповіддю є використання колекцій Google :

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

За допомогою Java 8 ви можете зробити наступне:

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

Value може бути будь-яким об’єктом, який ви використовуєте.

Оскільки 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);

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

Тому, залежно від ваших Listелементів, перетворити його в а може бути, а може і не бути можливим Map. У ваших Listпредметів немає дублікатів? Чи кожен елемент має унікальний ключ? Якщо так, то можна поставити їх у Map.

Алексіс вже розмістив відповідь на 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може зберегти деяку пам’ять відповідно до цього відповіді .

Існує також простий спосіб зробити це за допомогою Maps.uniqueIndex (...) від Googleгуава бібліотеки

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

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);
}

Без java-8 ви зможете це зробити в одній лінії колекцій Commons та класу закриття

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);
        }
    });
}};

Багато рішень спадає на думку, залежно від того, що ви хочете досягти:

Кожен елемент списку є ключовим і цінним

for( Object o : list ) {
    map.put(o,o);
}

Елементи списку мають на що подивитися, можливо, ім'я:

for( MyObject o : list ) {
    map.put(o.name,o);
}

Елементи списку мають на що їх шукати, і немає жодної гарантії, що вони унікальні: Використовуйте Googles MultiMaps

for( MyObject o : list ) {
    multimap.put(o.name,o);
}

Визначення всіх елементів позиції як ключових:

for( int i=0; i<list.size; i++ ) {
    map.put(i,list.get(i));
}

...

Це дійсно залежить від того, що ви хочете досягти.

Як видно з прикладів, Карта - це зіставлення від ключа до значення, тоді як список - це лише ряд елементів, що мають позицію кожного. Тому вони просто не конвертуються автоматично.

Ось невеликий метод, який я написав саме для цієї мети. Він використовує Validate від Apache Commons.

Сміливо користуйтеся ним.

/**
 * Converts a <code>List</code> to a map. One of the methods of the list is called to retrive
 * the value of the key to be used and the object itself from the list entry is used as the
 * objct. An empty <code>Map</code> is returned upon null input.
 * Reflection is used to retrieve the key from the object instance and method name passed in.
 *
 * @param <K> The type of the key to be used in the map
 * @param <V> The type of value to be used in the map and the type of the elements in the
 *            collection
 * @param coll The collection to be converted.
 * @param keyType The class of key
 * @param valueType The class of the value
 * @param keyMethodName The method name to call on each instance in the collection to retrieve
 *            the key
 * @return A map of key to value instances
 * @throws IllegalArgumentException if any of the other paremeters are invalid.
 */
public static <K, V> Map<K, V> asMap(final java.util.Collection<V> coll,
        final Class<K> keyType,
        final Class<V> valueType,
        final String keyMethodName) {

    final HashMap<K, V> map = new HashMap<K, V>();
    Method method = null;

    if (isEmpty(coll)) return map;
    notNull(keyType, Messages.getString(KEY_TYPE_NOT_NULL));
    notNull(valueType, Messages.getString(VALUE_TYPE_NOT_NULL));
    notEmpty(keyMethodName, Messages.getString(KEY_METHOD_NAME_NOT_NULL));

    try {
        // return the Method to invoke to get the key for the map
        method = valueType.getMethod(keyMethodName);
    }
    catch (final NoSuchMethodException e) {
        final String message =
            String.format(
                    Messages.getString(METHOD_NOT_FOUND),
                    keyMethodName,
                    valueType);
        e.fillInStackTrace();
        logger.error(message, e);
        throw new IllegalArgumentException(message, e);
    }
    try {
        for (final V value : coll) {

            Object object;
            object = method.invoke(value);
            @SuppressWarnings("unchecked")
            final K key = (K) object;
            map.put(key, value);
        }
    }
    catch (final Exception e) {
        final String message =
            String.format(
                    Messages.getString(METHOD_CALL_FAILED),
                    method,
                    valueType);
        e.fillInStackTrace();
        logger.error(message, e);
        throw new IllegalArgumentException(message, e);
    }
    return map;
}

Ви можете використовувати API потоків Java 8.

public class ListToMap {

  public static void main(String[] args) {
    List<User> items = Arrays.asList(new User("One"), new User("Two"), new User("Three"));

    Map<String, User> map = createHashMap(items);
    for(String key : map.keySet()) {
      System.out.println(key +" : "+map.get(key));
    }
  }

  public static Map<String, User> createHashMap(List<User> items) {
    Map<String, User> map = items.stream().collect(Collectors.toMap(User::getId, Function.identity()));
    return map;
  }
}

Для отримання більш детальної інформації відвідайте: http://codecramp.com/java-8-streams-api-convert-list-map/

Приклад Java 8 для перетворення List<?>об'єктів у Map<k, v>:

List<Hosting> list = new ArrayList<>();
list.add(new Hosting(1, "liquidweb.com", new Date()));
list.add(new Hosting(2, "linode.com", new Date()));
list.add(new Hosting(3, "digitalocean.com", new Date()));

//example 1
Map<Integer, String> result1 = list.stream().collect(
    Collectors.toMap(Hosting::getId, Hosting::getName));

System.out.println("Result 1 : " + result1);

//example 2
Map<Integer, String> result2 = list.stream().collect(
    Collectors.toMap(x -> x.getId(), x -> x.getName()));

Код скопійовано з:
https://www.mkyong.com/java8/java-8-convert-list-to-map/

як вже було сказано, у java-8 ми маємо стисле рішення від Collectors:

  list.stream().collect(
         groupingBy(Item::getKey)
        )

а також ви можете вкласти кілька груп, передаючи інший метод groupingBy як другий параметр:

  list.stream().collect(
         groupingBy(Item::getKey, groupingBy(Item::getOtherKey))
        )

Таким чином, у нас вийде багаторівнева карта, як це: Map<key, Map<key, List<Item>>>

Використовуючи потоки java-8

results.stream().collect(Collectors.toMap(e->((Integer)e[0]), e->(String)e[1]));

Мені подобається відповідь Kango_V, але я думаю, що це занадто складно. Я думаю, що це простіше - можливо, занадто просто. Якщо це схильне, ви можете замінити рядок на загальний маркер і змусити його працювати для будь-якого типу ключа.

public static <E> Map<String, E> convertListToMap(Collection<E> sourceList, ListToMapConverterInterface<E> converterInterface) {
    Map<String, E> newMap = new HashMap<String, E>();
    for( E item : sourceList ) {
        newMap.put( converterInterface.getKeyForItem( item ), item );
    }
    return newMap;
}

public interface ListToMapConverterInterface<E> {
    public String getKeyForItem(E item);
}

Використовується так:

        Map<String, PricingPlanAttribute> pricingPlanAttributeMap = convertListToMap( pricingPlanAttributeList,
                new ListToMapConverterInterface<PricingPlanAttribute>() {

                    @Override
                    public String getKeyForItem(PricingPlanAttribute item) {
                        return item.getFullName();
                    }
                } );

Apache Commons MapUtils.populateMap

Якщо ви не використовуєте Java 8 і не хочете з якихось причин використовувати явний цикл, спробуйте MapUtils.populateMapз Apache Commons.

MapUtils.populateMap

Скажіть, у вас є список Pairs.

List<ImmutablePair<String, String>> pairs = ImmutableList.of(
    new ImmutablePair<>("A", "aaa"),
    new ImmutablePair<>("B", "bbb")
);

І тепер ви хочете карту Pairключа «s до Pairоб'єкту.

Map<String, Pair<String, String>> map = new HashMap<>();
MapUtils.populateMap(map, pairs, new Transformer<Pair<String, String>, String>() {

  @Override
  public String transform(Pair<String, String> input) {
    return input.getKey();
  }
});

System.out.println(map);

дає вихід:

{A=(A,aaa), B=(B,bbb)}

Це, forможливо, простіше зрозуміти. (Це нижче дає такий же вихід):

Map<String, Pair<String, String>> map = new HashMap<>();
for (Pair<String, String> pair : pairs) {
  map.put(pair.getKey(), pair);
}
System.out.println(map);