Project Euler #9

euler9.py

from itertools import count
from itertools import dropwhile
from math import sqrt
from operator import mul

def triple():
    for a in range(1, 1000):
        for b in range(a + 1, 1000):
            c = int(sqrt(a * a + b * b))
            if c * c == a * a + b * b:
                yield (a, b, c)

print reduce(mul,
                dropwhile(lambda (a, b, c): a + b + c != 1000,
                    triple()).next())