如何判断Python生成器迭代到最后一个元素？解决输出格式问题

先假设你的素数生成器代码如下（核心逻辑一致即可，若实际代码有出入可按需调整）：

def prime_generator():
    for num in range(2, 101):
        is_prime = True
        for i in range(2, int(num**0.5) + 1):
            if num % i == 0:
                is_prime = False
                break
        if is_prime:
            yield num

一、调整输出格式：最后一个元素后加句号 #

方法1：转列表处理（简单高效，适合小数据集） #

把生成器的所有元素一次性转成列表，用join拼接元素（自动用逗号分隔），最后追加句号即可：

primes = list(prime_generator())
print(', '.join(map(str, primes)) + '.')

输出结果示例：2, 3, 5, 7, 11, ..., 97.

方法2：直接迭代生成器（适合大数据/无限生成器，低内存占用） #

如果不想一次性加载所有元素到内存，可以通过预取元素+跟踪后续迭代的方式判断最后一个元素：

gen = prime_generator()
# 先取出第一个元素
try:
    current_num = next(gen)
except StopIteration:
    # 生成器为空时的处理逻辑
    pass
else:
    # 打印第一个元素（不换行）
    print(current_num, end='')
    # 迭代剩余元素，每个元素前加逗号
    for num in gen:
        print(f", {num}", end='')
    # 最后追加句号
    print('.')

也可以用itertools.tee创建两个迭代器，一个提前一步探测后续元素：

import itertools

gen = prime_generator()
# 复制两个相同的迭代器
iter1, iter2 = itertools.tee(gen)
# 让第二个迭代器先走一步
next(iter2, None)

for num in iter1:
    # 检查第二个迭代器是否还有下一个元素
    if next(iter2, None) is None:
        print(f"{num}.")
    else:
        print(f"{num}, ", end='')

二、如何判断生成器迭代到最后一个元素 #

生成器是惰性迭代的，没有内置的长度属性，无法直接通过索引或长度判断当前元素是否为最后一个，常用思路有两种：

• 转列表预存：将生成器所有元素转成列表后，可通过len(primes)和索引primes[-1]定位最后一个元素。但这种方法会把所有元素加载到内存，不适合元素极多或无限的生成器。
• 双迭代器/预取探测：通过两个迭代器（或手动预取下一个元素），在迭代当前元素时，提前检查是否还有下一个元素。如果没有，则当前元素就是最后一个。这种方法不需要预存所有元素，内存占用低，适合大数据场景。

内容的提问来源于stack exchange，提问作者Lauriell