在 Python 中获取生成器的第 N 个元素

获取生成器的第 N 个元素:

  1. 使用 islice() 函数获取从索引 N 开始的迭代器。
  2. 将迭代器传递给 next() 函数。
  3. next() 函数将返回生成器的第 N 个元素。
from itertools import islice


def g():
    yield 1
    yield 2
    yield 3


gen = g()

second = next(
    islice(gen, 1, None),
    None
)
print(second)  # 👉️ 2

itertools.islice() 方法采用迭代器和可选的开始和停止索引。

我们使用了 0 的开始索引和 None 的停止索引。

当 stop 为 None 时,迭代将继续,直到迭代器耗尽。

对于无限长的生成器,迭代会在指定位置停止。

next() 函数从提供的迭代器中返回下一项。

该函数可以传递一个默认值作为第二个参数。

如果迭代器耗尽或为空,则返回默认值。

如果迭代器已用尽或为空且未提供默认值,则会引发 StopIteration 异常。

from itertools import islice


def g():
    yield 1
    yield 2
    yield 3


gen = g()

item = next(
    islice(gen, 100, None),
    'default value'
)
print(item)  # 👉️ default value

如果我们的生成器对象不是无限长的,我们可以将其转换为列表并访问索引 N 处的列表元素。

def g():
    yield 1
    yield 2
    yield 3


gen = g()

my_list = list(gen)

print(my_list[0])  # 👉️ 1
print(my_list[1])  # 👉️ 2
print(my_list[2])  # 👉️ 3

print(list(gen)) # 👉️ []

列表类接受一个可迭代对象并返回一个列表对象。

我们还可以使用 enumerate() 函数来访问当前迭代的索引。

def g():
    yield 1
    yield 2
    yield 3


gen = g()

nthElement = next(
    (item for index, item in enumerate(gen) if index == 1),
    None
)

print(nthElement)  # 👉️ 2

在每次迭代中,我们检查当前索引是否等于特定索引。

如果满足条件,我们返回相应的项目。

我们选择哪种方法是个人喜好问题。 我会将 list() 类用于非无限生成器对象,因为转换为列表非常直观且易读。