java - 一起迭代两个 Java-8-Stream

Question

这个问题在这里已经有了答案:

Zipping streams using JDK8 with lambda (java.util.stream.Streams.zip) (14 个回答)

关闭 8年前。

我想一起迭代两个 Java-8-Stream，以便在每个迭代步骤中都有两个参数。类似的东西，somefunction产生类似 Stream<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 提供类似的东西，虽然没有 Pair在 Java 中:-(如果没有这样的 API-Function，是否有另一种同时迭代两个流的方法？

Answer 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 has pointed out如果 bs 是无限的而 as 不是，则它不起作用。对于适用于无限和有限流的所有可能组合的解决方案，在 hasNext 方法中检查两个源的中间 Iterator 似乎是不可避免的¹:

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);
}

如果你想要并行压缩，你应该考虑 Stream 的 source。例如。您可以压缩两个 ArrayList(或任何 RandomAccessList)，例如

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)