Question

如何获取生成器的第n个值？

public List<Fruit> getFruits() {
    return allBananas().stream().collect(Collectors.toList());
}

Answer 1

您可以定义类似in python的枚举方法：

function *enumerate(it, start) {
   start = start || 0;
   for(let x of it)
     yield [start++, x];
}

然后：

for(let [n, x] of enumerate(index()))
  if(n == 6) {
    console.log(x);
    break;
  }

http://www.es6fiddle.net/ia0rkxut/

同样，人们也可以重新实现pythonic range和islice：

function *range(start, stop, step) {
  while(start < stop) {
    yield start;
    start += step;
  }
}

function *islice(it, start, stop, step) {
  let r = range(start || 0, stop || Number.MAX_SAFE_INTEGER, step || 1);
  let i = r.next().value;
  for(var [n, x] of enumerate(it)) {
    if(n === i) {
      yield x;
      i = r.next().value;
    }
  }
}

然后：

console.log(islice(index(), 6, 7).next().value);

http://www.es6fiddle.net/ia0s6amd/

现实世界的实施需要更多的工作，但你明白了。

Answer 2

As T.J. Crowder pointed out，无法直接转到n元素，因为值是按需生成的，只有next函数才能检索到立即值。因此，我们需要明确跟踪消耗的项目数量。

唯一的解决方案是使用循环，我更喜欢用for..of迭代它。

我们可以创建一个像这样的函数

function elementAt(generator, n) {
    "use strict";

    let i = 0;

    if (n < 0) {
        throw new Error("Invalid index");
    }

    for (let value of generator) {
        if (i++ == n) {
            return value;
        }
    }

    throw new Error("Generator has fewer than " + n + " elements");
}

然后像这样调用它

console.log(elementAt(index(), 10));
// 10

另一个有用的功能可能是take，它允许您从生成器中获取第一个n元素，例如

function take(generator, n) {
    "use strict";

    let i = 1,
        result = [];

    if (n <= 0) {
        throw new Error("Invalid index");
    }

    for (let value of generator) {
        result.push(value);
        if (i++ == n) {
            return result;
        }
    }

    throw new Error("Generator has fewer than " + n + " elements");
}

console.log(take(index(), 10))
// [ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 ]

Answer 3

一个简单的循环可以：

let n = 10,
    iter = index();
while (--n > 0) iter.next();
console.log(iter.next().value); // 9

Answer 4

您可以创建一个大小为 n 的数组，并使用Array.from及其第二个参数来获取所需的值。假设iter是生成器gen的迭代器：

var iter = gen();

然后可以按如下方式获取第一个 n 值：

var values = Array.from(Array(n), iter.next, iter).map(o => o.value)

...当您只对 n ^th 值感兴趣时，可以跳过map部分，然后执行：

var value = Array.from(Array(n), iter.next, iter).pop().value

或者：

var value = [...Array(n)].reduce(iter.next.bind(iter), 1).value

缺点是你仍然（暂时）分配一个大小 n 的数组。

Answer 5

我想避免不必要地创建数组或其他中间值。这就是我对 nth 的实现的结果 -

function nth (iter, n)
{ for (const v of iter)
    if (--n < 0)
      return v
}

按照原始问题中的示例 -

// the 1st value
console.log(nth(index(), 0))

// the 3rd value
console.log(nth(index(), 2))

// the 10th value
console.log(nth(index(), 9))

0
2
9

对于有限生成器，如果索引越界，结果将是 undefined -

function* foo ()
{ yield 1
  yield 2
  yield 3
}

console.log(nth(foo(), 99))

undefined

展开下面的代码段以验证浏览器中的结果 -

function *index ()
{ let x = 0
  while (true)
    yield x++
}

function* foo ()
{ yield 1
  yield 2
  yield 3
}

function nth (iter, n) {
  for (const v of iter)
    if (--n < 0)
      return v
}

// the 1st value
console.log(nth(index(), 0))

// the 3rd value
console.log(nth(index(), 2))

// the 10th value?
console.log(nth(index(), 9))

// out-of-bounds?
console.log(nth(foo(), 99))

如何获取JavaScript生成器的第n个值？

5 个答案: