Files
Learn-Python-Programming-Th…/ch05/pythagorean.triple.comprehension.py
T
adii1823 ef37ce0c4e ch05
2021-10-28 17:38:16 +05:30

11 lines
372 B
Python

# pythagorean.triple.comprehension.py
from math import sqrt
# this step is the same as before
mx = 10
triples = [(a, b, sqrt(a**2 + b**2))
for a in range(1, mx) for b in range(a, mx)]
# here we combine filter and map in one CLEAN list comprehension
triples = [(a, b, int(c)) for a, b, c in triples if c.is_integer()]
print(triples) # prints: [(3, 4, 5), (6, 8, 10)]