import itertools with open('data/sample.txt') as f: data = list(map(int,f.readlines())) for x, y in itertools.combinations(data, 2): if x+y == 2020: print(x*y) with open('data/data.txt') as f: data = list(map(int,f.readlines())) for x, y, z in itertools.combinations(data, 3): if x+y+z == 2020: print(x*y*z)