Python中map对象的'in'操作为何非确定性？是Bug吗？

现象描述 #

在最新Python版本中运行以下代码，有时会多次输出0、1、2、3，但绝大多数时候会连续一千次输出1：

generated_names = set(["0"])

def generate_name()->str:
        global generated_names

        name_int = 0
        
        already_generated_names_int = map(int, generated_names)

        while name_int in already_generated_names_int:
            name_int +=1

        name = str(name_int)

        generated_names.add(name)
        return name

for i in range(1000):
    print(generate_name())

核心疑问 #

为什么in操作对map对象的行为和其他可迭代对象不同？将map转为set后函数可正常输出连续序列，但直接使用map却不行？

问题根源 #

1. map对象是一次性迭代器，而非可重复遍历的容器 #

map返回的是迭代器对象，核心特点是只能被完整遍历一次：遍历过程中会逐步消耗内部元素，遍历结束后迭代器就会变为空，无法再次复用。

而in操作对迭代器的逻辑是：从头开始遍历元素，直到找到匹配项或遍历完所有元素。一旦遍历过部分或全部元素，迭代器的指针就停留在当前位置，后续的in操作要么从该位置继续遍历（若有剩余元素），要么直接返回False（若已遍历完）。

2. 代码中的具体矛盾 #

拆解generate_name函数的执行逻辑：

• 第一次调用时，generated_names是{"0"}，already_generated_names_int是包含0的map迭代器：
  • 第一次判断0 in already_generated_names_int：遍历找到0，返回True，此时迭代器已被耗尽。
  • name_int自增为1，再次判断1 in already_generated_names_int：迭代器为空，返回False，循环终止，返回"1"并加入集合。
• 第二次调用时，generated_names是{"0", "1"}，map迭代器的元素顺序由set的无序性决定：
  • 若迭代器元素顺序是0在前：第一次判断找到0后，迭代器指针移到1的位置；name_int变为1，再次判断找到1后迭代器耗尽；name_int变为2，判断时迭代器为空，返回2——这种情况会输出连续序列。
  • 若迭代器元素顺序是1在前：第一次判断0 in 迭代器时，先遍历到1不匹配，继续遍历到0才匹配，此时迭代器已耗尽；name_int变为1，再次判断时迭代器为空，直接返回1——这就是大多数时候连续输出1的原因。

set的无序性导致map迭代器的元素顺序随机，最终造成了代码行为的非确定性。

3. 转为set后正常的原因 #

set是可重复遍历的容器，每次执行in操作都会完整遍历整个集合，不会出现迭代器耗尽的情况。无论判断多少次，都能准确检查name_int是否存在于集合中，因此可以稳定生成连续序列。

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