python - 删除集合中包含特定字符的所有元素

Question

我有一组从生成器生成的几千个素数:

primes = set(primegen()) = set([..., 89, 97, 101, 103, ...])

其中一些素数中有一个零。我想摆脱他们。有没有办法一次完成所有这些？

目前，我正在删除元素，因为我使用正则表达式匹配遍历素数:

import re
zero = re.compile('.+0.+') 

while primes:
    p = str(primes.pop())
    if zero.match(p):
        continue
    # do other stuff

我认为这是最好的方法，但我很好奇我是否错了。

Answer 1

您可以使用集合理解来过滤现有的素数集合。

primes = {p for p in primes if '0' not in str(p)}