我可以使用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)
我想使用list comprehension用一行代替它,并尝试了以下部分:
[a, b, c in range(n+1), range(a), range(b) if a*a == b*b + c*c]
但是,我在结束方括号上遇到语法错误。我试图使用简单的括号将列表更改为元组,但没有成功。我可以知道如何做对吗?
答案 0 :(得分:5)
我认为你的意思是
[(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]
至少在语法上是有效的。
答案 1 :(得分:1)
假设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]