从列表中删除非素数

Question

我有这样的东西：

palindromes=[1, 2, 3, 4, 5, 6, 7, 8, 9, 11, 101, 111, 121, ..., 99799, 99899, 99999]
# of course im generating it :)

def isPrime(number):
    for i in range(2,int(number/2)+1):
        if number%i == 0:
            return True
    return False

def removeNonPrimes(palindromes):
    for palindrom in palindromes:
        if isPrime(palindrom):
            palindromes.remove(palindrom)
    return palindromes

palindromes = removeNonPrimes(palindromes)

它不会删除所有非素数

我不知道为什么

Answer 1

在您的代码中：

def removeNonPrimes(palindromes):
    for palindrom in palindromes:
        if isPrime(palindrom):
            palindromes.remove(palindrom)
    return palindromes

您正在修改（使用.remove()）您正在迭代的同一个列表（使用in）。不建议这样做，您最终可能会删除您想要的不同项目。

相反，考虑一个列表推导：

def removeNonPrimes(palindromes):
    return [p for p in palindromes if isPrime(p)]

Answer 2

isPrime 的返回值与应有的相反。

Answer 3

您正在迭代一个可迭代对象并同时修改该可迭代对象。因此，当一个循环完成时，指针可能不一定指向您正在迭代的原始集合中的下一项。

如果 isPrime(palindrom) 返回 True 并且您删除了一项，则列表的长度会减一。

考虑回文 = [7,8,9,10] 且回文的值为 8 的情况。 palindrom 实际上指向回文[1]。循环结束后，回文将指向回文[2]。 同时由于 8 将被从回文中删除，因为它不是素数。 现在 palindromes = [7,9,10] 和 palindrom = palindromes[2] = 10。您可以看到值 9 从未被触及。

道德：永远不要操纵你正在迭代的可迭代对象。这就像砍掉你坐的那棵树的树枝。

（顺便说一句，函数名应该是 isNotPrime 而不是 isPrime 以反映它正在检查的内容。:)）

Answer 4

如果你想要所有回文素数，这会更好：

def eras(n):
    last = n + 1
    sieve = [0,0] + list(range(2,last))
    sqn = int(round(n ** 0.5))
    it = (i for i in xrange(2, sqn + 1) if sieve[i])
    for i in it:
        sieve[i*i:last:i] = [0] * (n//i - i + 1)
    return filter(None, sieve)

def is_palindrome(n):
    ns = str(n)
    ns_reversed = ''.join(reversed(ns))
    return ns == ns_reversed

if __name__ == '__main__':
    primes = eras(100 * 1000)
    print [p for p in primes if is_palindrome(p)]