java - 如何将列表转换为 map ？

Question

最近，我与一位同事讨论了在 Java 中将 List 转换为 Map 的最佳方式，以及这样做是否有任何具体好处。

我想知道最佳的转化方法，如果有人能指导我，我将不胜感激。

这是个好方法吗:

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)