如何在列表中查找特定元素并将它们组合成一个新列表（Python3）

Question

首先，我是这个网站和编程的新手，所以如果我的代码编写水平不好或者我提出问题的方式有误，请指出我正确的方向。

我已经用 Python3 编写了一个程序，它使我能够获得一个包含小于我选择的给定数 N 的所有素数的列表。例如，如果我的输入是 10，我的程序将给出以下结果：

>>>primes(10)
[2, 3, 5, 7]

现在我正在尝试编写一个后续程序，它给出了给定数字 N 的孪生素数。例如：如果 N = 10，我知道有两个孪生素数 [3,5] 和 [5 ,7]。 我尝试了以下方法：

def twinprimes(n):
    x = primes(n)
    ps = [] 
    for i in range(len(x)):
        if x[i]+2==x[i]:
        ps.append((x[i])
    return ps

所以我正在尝试创建一个新列表 ps，其中添加了 twinprime 对的所有第一个值，以便我可以找到 len(ps) 以找到 twinprimes 的总值。

现在我知道我的代码有很多问题。我只是不知道如何让它工作。

Answer 1

你正在尝试测试一个素数是否等于它自己加 2：

if x[i]+2==x[i]:

没有一个数字等于它自己加上一个非零的其他数字，所以这个测试永远不会通过。

您想测试当前素数是否比列表中的下一个数小 2：

if x[i] == x[i + 1] - 2:  # two adjacent primes differ by 2

这将遇到i + 1 不是x 的有效索引的问题，因为您循环到（但不包括）len(x)； len(x) + 1 将提高 IndexError。但是您无论如何都不需要测试最后一个素数，因此以下方法可以工作：

for i in range(len(x) - 1):
    if x[i]  == x[i + 1] - 2:
        ps.append((x[i], x[i + 1]))

这会添加两个素数。

有更多的pythonic方法来生成这个列表；例如，将素数与zip() 配对，并使用列表推导：

def twinprimes(n):
    x = primes(n)
    return [(a, b) for a, b in zip(x, x[1:]) if a == b - 2]