关于字典:用流映射的Java 8列表

Java 8 list to map with stream

我有一个List集合。
我需要将其转换为Map
地图的关键字必须是集合中项目的索引。
我无法弄清楚如何使用流来做到这一点。
就像是:

1
items.stream().collect(Collectors.toMap(...));

有帮助吗?

由于这个问题被确定为可能重复,我需要补充一点,我的具体问题是 - 如何获取列表中项目的位置并将其作为键值


您可以使用IntStream创建Stream索引,然后将它们转换为Map

1
2
3
4
Map<Integer,Item> map =
    IntStream.range(0,items.size())
             .boxed()
             .collect(Collectors.toMap (i -> i, i -> items.get(i)));


另一个完整性的解决方案是使用自定义收集器:

1
2
3
4
5
6
7
8
public static < T > Collector<T, ?, Map<Integer, T>> toMap() {
    return Collector.of(HashMap::new, (map, t) -> map.put(map.size(), t),
            (m1, m2) -> {
                int s = m1.size();
                m2.forEach((k, v) -> m1.put(k+s, v));
                return m1;
            });
}

用法:

1
Map<Integer, Item> map = items.stream().collect(toMap());

此解决方案是并行友好的,不依赖于源(您可以使用列表而无需随机访问或Files.lines()或其他)。


不要觉得你必须在流中做任何事情。我会这样做:

1
2
AtomicInteger index = new AtomicInteger();
items.stream().collect(Collectors.toMap(i -> index.getAndIncrement(), i -> i));

只要您不对流进行并行化,这将起作用,并且它避免了潜在的昂贵和/或有问题(在重复的情况下)get()indexOf()操作。

(你不能使用常规的int变量来代替AtomicInteger,因为从lambda表达式外部使用的变量必须是有效的。注意,当无争议时(如本例所示),AtomicInteger非常快并且赢得了'这会造成性能问题。但如果它让您担心,您可以使用非线程安全的计数器。)


这是更新的答案,没有评论中提到的问题。

1
Map<Integer,Item> outputMap = IntStream.range(0,inputList.size()).boxed().collect(Collectors.toMap(Function.identity(), i->inputList.get(i)));


Eran的答案通常是随机访问列表的最佳方法。

如果您的List不是随机访问,或者您有Stream而不是List,则可以使用forEachOrdered

1
2
3
4
Stream<Item> stream = ... ;
Map<Integer, Item> map = new HashMap<>();
AtomicInteger index = new AtomicInteger();
stream.forEachOrdered(item -> map.put(index.getAndIncrement(), item));

如果流是并行的,这是安全的,即使目标地图是线程不安全的并且作为副作用进行操作。 forEachOrdered保证按顺序一次一个地处理项目。因此,并行运行不会产生任何加速。 (如果在forEachOrdered之前管道中存在昂贵的操作,可能会有一些加速。)


使用第三方库(例如,质子包,但还有其他库),您可以zip使用其索引和瞧的值:

1
2
StreamUtils.zipWithIndex(items.stream())
    .collect(Collectors.toMap(Indexed::getIndex, Indexed::getValue));

虽然getIndex返回long,但您可能需要使用类似于:

1
i -> Integer.valueOf((int) i.getIndex())