Как пройти несколько списков с помощью Java 8 stream?
у меня есть три списка,
List<Double> list1= new ArrayList(List.of(1, 1.5));
List<Double> list2= new ArrayList(List.of(30, 25));
List<Double> list3= new ArrayList(List.of(30, 25));
Я хочу пройти через каждый одновременно и распечатать
1 30 30
1.5 25 25
Как я могу это сделать с помощью Java-8 stream API?
4 ответов
это работает, когда размер списков одинаковый или другой:
List<Double> list1 = List.of(1D, 1.5D);
List<Double> list2 = List.of(30D, 25D);
List<Double> list3 = List.of(30D, 25D);
Stream<List<Double>> listStream = Stream.of(list1, list2, list3);
int maxSize = listStream.mapToInt(List::size).max().orElse(0);
IntStream.range(0, maxSize)
.forEach(index -> {
listStream
.filter(list -> list.size() > index)
.forEach(list -> System.out.print(list.get(index) + " "));
System.out.println();
});
вы могли бы сделать (при условии, что все списки имеют одинаковый размер):
IntStream.range(0, list1.size())
.forEach(x -> Syso(list1.get(x) + " " + list2.get(x) + " " + list3.get(x)))
но здесь нет никакой реальной выгоды, просто используйте простой цикл for.
и кстати не оберните их с new ArrayList(List.of(30, 25));
, этого было бы достаточно:
List<Double> list1 = List.of(1D, 1.5D)
List.of(..)
является статической фабрикой, возвращающей List
сам по себе, нет необходимости обертывать его в ArrayList::new
конструктор.
вот код, который вам, вероятно, нужен, однако, не стоит недооценивать мощность для for-loop
:
// Here you need to get the max of all the List sizes to traverse all Lists
int maxSize = Math.max(list1.size(), Math.max(list2.size(), list3.size()));
IntStream.range(0, maxSize) // Iterate 0 .. minSize-1
.mapToObj(i -> Stream.of(list1, list2, list3) // Map each index
.filter(list -> i < list.size()) // Secure the existing index
.map(list -> list.get(i)) // Get the item
.collect(Collectors.toList())) // Collect to List
.forEach(System.out::println); // And iterate (print)
выход:
[1, 30, 30]
[1.5, 25, 25]
в случае размера списка переменных итерация обхода безопасна, потому что я касаюсь только существующих индексов.
нет необходимости в потоках-просто транспонируйте списки:
private static class Transpose<T> extends AbstractList<List<T>> implements List<List<T>> {
private final List<T> [] lists;
public Transpose(List<T>... lists) {
this.lists = lists;
}
private class Column extends AbstractList<T> implements List<T> {
private final int column;
public Column(int column) {
this.column = column;
}
@Override
public T get(int row) {
return lists[row].get(column);
}
@Override
public int size() {
return lists.length;
}
}
@Override
public List<T> get(int column) {
return new Column(column);
}
@Override
public int size() {
return lists[0].size();
}
}
private void test() {
List<Double> list1= Arrays.asList(1.0, 1.5);
List<Double> list2= Arrays.asList(30.0, 25.0);
List<Double> list3= Arrays.asList(30.0, 25.0);
List<List<Double>> matrix = new Transpose(list1, list2, list3);
System.out.println(matrix);
}
гравюры:
[[1.0, 30.0, 30.0], [1.5, 25.0, 25.0]]
очевидно, это было бы сложнее, если бы списки не были одинаковой длины.