如何不从生成器返回重复值？

Question

我有如下功能：

def fun(content):
    for i in content:
        id = i.split('\"')[0]
        yield id    
    return id

问题是内容中有一些重复的值。 有没有办法知道值“id”是否已经在生成器“id”中？而不是得到最终的生成器然后使用 set()？

Answer 1

您可以在fun 中使用set 来跟踪已经看到的ID：

def fun(content):
    observed = set()
    for i in content:
        id = i.split('\"')[0]
        if id not in observed:
            observed.add(id)
            yield id

此外，由于您正在生成 id，因此您不需要在最后返回。