使用 python 的列表理解的毕达哥拉斯三元组

Question

我可以使用 for 循环找出勾股三元组，如下所示：

def triplet(n): # Find all the Pythagorean triplets between 1 and n (inclusive)
  for a in range(n+1):
    for b in range(a):
      for c in range(b):
        if a*a == b*b + c*c:
          print(a, b, c)

我想使用列表推导将其替换为单行代码并尝试了以下部分：

[a, b, c in range(n+1), range(a), range(b) if a*a == b*b + c*c]

但是，我在右方括号中收到语法错误。我尝试使用简单的括号将列表更改为元组，但没有成功。我可以知道如何正确处理吗？

Answer 1

我想你是说

[(a,b,c) for a in range(n+1) for b in range(a) for c in range(b) if a*a == b*b + c*c]

这至少在语法上是有效的。

Answer 2

注意：此方案仅针对a + b + c <= N

时的问题

假设a<=b<=c，这个版本快一点：

triplet = [(a,b,c) for a in range(1,N//3+1) for b in range(a,N//2+1) for c in range(b,N-1) if a**2 + b**2 == c**2]

使用 python 的列表理解的毕达哥拉斯三元组

Pythagorean triplets using python's list comprehension

list-comprehension

pythagorean

python-3.x