Как отобразить элементы на их индекс с помощью потоков?
Я получил поток некоторых пользовательских объектов, и я хотел бы создать карту Map<Integer, MyObject>
с индексом каждого объекта в качестве ключа. Чтобы дать вам простой пример:
Stream<String> myStream = Arrays.asList("one","two","three").stream();
Integer i = 0;
Map<Integer, String> result3 = myStream.collect(Collectors.toMap(x -> i++, x -> x));
Очевидно, это не компилируется, потому что:
локальные переменные, на которые ссылается лямбда-выражение, должны быть окончательными или эффективно окончательный
Есть ли простой способ сопоставить элементы потока с их индексами, чтобы ожидаемый результат для приведенного выше примера был примерно таким:
{1=one, 2=two, 3=three}
Ответы
Ответ 1
Ваша переменная i
не является окончательной.
Вы можете использовать AtomicInteger
в качестве оболочки Integer
:
Stream<String> myStream = Arrays.asList("one","two","three").stream();
AtomicInteger atomicInteger = new AtomicInteger(0);
Map<Integer, String> result3 = myStream.collect(Collectors.toMap(x -> atomicInteger.getAndIncrement(), Function.identity()));
Я считаю это немного хакерским, потому что это решает только проблему окончательной переменной. Поскольку это специальная версия ThreadSafe, это может привести к некоторым накладным расходам. Чистое решение stream
в ответе Самуэля Филиппа может лучше соответствовать вашим потребностям.
Ответ 2
Вы можете использовать IntStream
для решения этой проблемы:
List<String> list = Arrays.asList("one","two","three");
Map<Integer, String> map = IntStream.range(0, list.size()).boxed()
.collect(Collectors.toMap(Function.identity(), list::get));
Вы создаете IntStream
из 0
в list.size() - 1
(IntStream.range()
исключает последнее значение из потока) и сопоставляете каждый индекс со значением в вашем списке. Преимущество этого решения в том, что оно также будет работать с параллельными потоками, что невозможно при использовании AtomicInteger
.
Таким образом, результатом в этом случае будет:
{0=one, 1=two, 2=three}
Чтобы начать первый индекс в 1
, вы можете просто добавить 1
во время сбора:
List<String> list = Arrays.asList("one", "two", "three");
Map<Integer, String> map = IntStream.range(0, list.size()).boxed()
.collect(Collectors.toMap(i -> i + 1, list::get));
Это приведет к такому:
{1=one, 2=two, 3=three}
Ответ 3
Чистое решение, не требующее источника данных произвольного доступа, это
Map<Integer,String> result = Stream.of("one", "two", "three")
.collect(HashMap::new, (m,s) -> m.put(m.size() + 1, s),
(m1,m2) -> {
int offset = m1.size();
m2.forEach((i,s) -> m1.put(i + offset, s));
});
Это также работает с параллельными потоками.
В маловероятном случае, когда это повторяющаяся задача, стоит поместить логику в сборщик многократного использования, включая некоторые оптимизации:
public static <T> Collector<T,?,Map<Integer,T>> toIndexMap() {
return Collector.of(
HashMap::new,
(m,s) -> m.put(m.size() + 1, s),
(m1,m2) -> {
if(m1.isEmpty()) return m2;
if(!m2.isEmpty()) {
int offset = m1.size();
m2.forEach((i,s) -> m1.put(i + offset, s));
}
return m1;
});
}
Который затем можно использовать как
Map<Integer,String> result = Stream.of("one", "two", "three")
.collect(MyCollectors.toIndexMap());
или
Map<Integer,Integer> result = IntStream.rangeClosed(1, 1000)
.boxed().parallel()
.collect(MyCollectors.toIndexMap());
Ответ 4
Попробуйте это:
Скажем тогда String[] array = { "V","I","N","A","Y" };
,
Arrays.stream(array)
.map(ele-> index.getAndIncrement() + " -> " + ele)
.forEach(System.out::println);
Вывод:
0 -> V
1 -> I
2 -> N
3 -> A
4 -> Y
Ответ 5
У гуавы есть статический метод Streams#mapWithIndex
Stream<String> myStream = Stream.of("one","two","three");
Map<Long, String> result3 = Streams.mapWithIndex(myStream, (s, i) -> Maps.immutableEntry(i + 1, s))
.collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));
// {1=one, 2=two, 3=three}
System.out.println(result3);
Ответ 6
Мы можем использовать метод List.indexOf(Object o)
, чтобы получить индекс элемента в списке, создавая Map
:
List<String> list = Arrays.asList("one","two","three");
Map<Integer, String> result = list.stream()
.collect(Collectors.toMap(
k -> list.indexOf(k) + 1,
Function.identity(),
(v1, v2) -> v2));
System.out.println(result);
Если в списке есть дубликаты, индекс первого вхождения элемента будет добавлен в итоговую карту. Также, чтобы устранить ошибку слияния в карте во время столкновения клавиш, нам нужно убедиться, что есть функция слияния, поставляемая в toMap(keyMapper, valueMapper, mergeFunction)
Выход:
{1=one, 2=two, 3=three}