同时迭代两个 Java-8 流

java java-8 java-stream

2022-09-01 11:02:40

我想将两个Java-8-Streams迭代在一起，这样我在每个迭代步骤中都有两个参数。类似的东西，产生类似.somefunctionStream<Pair<A,B>>

Stream<A> as;
Stream<B> bs;
somefunction (as, bs)
  .forEach ((a, b) -> foo (a, b));
// or something like
somefunction (as, bs)
  .forEach ((Pair<A, B> abs) -> foo (abs.left (), abs.right ()));

我想知道，如果Java提供了这样的东西，尽管Java中没有：-（如果没有这样的API函数，是否有另一种方法可以同时迭代两个流？Pair

答案 1

static <A, B> Stream<Pair<A, B>> zip(Stream<A> as, Stream<B> bs)
{
    Iterator<A> i=as.iterator();
    return bs.filter(x->i.hasNext()).map(b->new Pair<>(i.next(), b));
}

这不提供并行执行，但原始实现也没有。zip

正如F. Böller所指出的，如果它是无限的，那么它就不起作用。对于适用于无限流和有限流的所有可能组合的解决方案，在方法中检查两个源的中间体似乎是不可避免的¹：bsasIteratorhasNext

static <A, B> Stream<Pair<A,B>> zip(Stream<A> as, Stream<B> bs) {
    Iterator<A> i1 = as.iterator();
    Iterator<B> i2 = bs.iterator();
    Iterable<Pair<A,B>> i=()->new Iterator<Pair<A,B>>() {
        public boolean hasNext() {
            return i1.hasNext() && i2.hasNext();
        }
        public Pair<A,B> next() {
            return new Pair<A,B>(i1.next(), i2.next());
        }
    };
    return StreamSupport.stream(i.spliterator(), false);
}

如果您想要并行压缩功能，则应考虑 .例如，您可以压缩两个（或任何），如StreamArrayListRandomAccessList

ArrayList<Foo> l1=new ArrayList<>();
ArrayList<Bar> l2=new ArrayList<>();
IntStream.range(0, Math.min(l1.size(), l2.size()))
         .mapToObj(i->new Pair(l1.get(i), l2.get(i)))
         . …

¹（除非您直接实现）Spliterator

答案 2