import fileinput digit_map = { "one": 1, "two": 2, "three": 3, "four": 4, "five": 5, "six": 6, "seven": 7, "eight": 8, "nine": 9, } digits = list(digit_map) inp = list(fileinput.input()) # # Part 1 # part1 = sum( [ int( next(x for x in y if x.isdigit()) + next(x for x in reversed(y) if x.isdigit()) ) for y in inp ] ) print(part1) # # Part 2 # part2 = 0 for line in inp: newline = [] for i in range(len(line)): for digit in digits: if line[i:].startswith(digit): newline.append(str(digit_map[digit])) break else: if line[i].isdigit(): newline.append(line[i]) part2 += int(newline[0] + newline[-1]) print(part2)