aoc2020/1/2.py

import sys

def find_sum(l, total):
    for i in l:
        m = list(filter(lambda x: i + x == total, l))
        if m:
            return (i, m[0])
    return (None, None)

ns = []
with open("input") as f:
    for n in f:
        n=int(n)
        n1, n2 = find_sum(ns, 2020-n)
        if n1:
            print(n1*n2*n)
            sys.exit(0)
        ns.append(n)