Как пройти несколько списков с помощью 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]]

очевидно, это было бы сложнее, если бы списки не были одинаковой длины.