AoC 2023

day1.py

import re

digitMap = {'one': '1', 'two':'2', 'three': '3', 'four': '4', 'five': '5', 'six': '6', 'seven': '7', 'eight': '8', 'nine': '9', '1': '1', '2':'2', '3': '3', '4': '4', '5': '5', '6': '6', '7': '7', '8': '8', '9': '9'}

def getNum1(s):
    print(s)
    digits = re.findall(r'\d', s)
    return int(digits[0]+digits[-1])

def getNum2(s):
    digits = re.findall(r'\d|one|two|three|four|five|six|seven|eight|nine', s)
    one = digitMap[digits[0]]
    digits = re.findall(r'\d|enin|thgie|neves|xis|evif|ruof|eerht|owt|eno', s[::-1])
    two = digitMap[digits[0][::-1]]
    return int(one+two)

print(sum([getNum1(s) for s in open('data1.txt', 'r', encoding="utf-8")]))
print(sum([getNum2(s) for s in open('data1.txt', 'r', encoding="utf-8")]))