aoc/2020/01_expense.py

#!/usr/bin/env python3
import sys
from functools import reduce
from itertools import combinations, product
from operator import mul

def group(expenses, n):
    for group in combinations(expenses, n):
        if sum(group) == 2020:
            return reduce(mul, group)

expenses = [int(line) for line in sys.stdin]
print(group(expenses, 2))
print(group(expenses, 3))