0

我只是StopIteration在一些嵌套生成器中滥用（使用 CPython 3.6.9），没有启用PEP 479（from __future__ import generator_stop），并且有一些糟糕的 hacky 代码使用next(iter(iterable))过早发出停止信号。

虽然 PEP 479 会StopIteration从生成器冒泡中捕获，但我认为我仍然会在嵌套的 for 循环中遇到这个问题。

现在，我将next(iter(...))用以下内容替换任何用法：

def take(iterable, *, n):
    """
    Robustly gets the first n items from an iterable and returns them as a
    list.

    You should always use this function in lieu of `next(iter(...))`! e.g.
    instead of:

        my_first = next(iter(container))

    you should instead do:

        my_first, = take(container, n=1)

    Throws RuntimeError if the iterable cannot yield n items.
    """
    iterator = iter(iterable)
    out = []
    for _ in range(n):
        try:
            out.append(next(iterator))
        except StopIteration:
            raise RuntimeError("Premature StopIteration encountered!")
    return out

我的问题是：这样的函数是否已经在 Python 的 stdlib 中？

我在 and 中查看了最新python.org的文档（对于 3.9），我能看到的最接近的东西是 我也可以转换为一个或任何其他可索引容器，但我想避免为了访问第一件事而需要遍历所有内容。itertoolsbuiltinstakewhilelist

4

1 回答 1

1

itertools.islice这样做（以及更多），如果没有生成足够的元素，则不会转换为列表或出错。

你可以用这个干净地编写你的函数：

def take(iterable, *, n):
    li = list(itertools.islice(iterable, n))
    if len(li) != n:
        raise RuntimeError("too short iterable for take")
    return li
于 2020-11-04T17:27:06.317 回答