day06b.py

from functools import reduce

f = open('input.txt', 'r')
items = f.read().split('\n')

rows = items[:-1]
ops = items[-1]

idxs = [i for i in range(len(ops)) if ops[i] != ' '] + [len(ops)+1]

res = 0
for i in range(len(idxs)-1):
    op = ops[idxs[i]]

    nums = [[] for _ in range(idxs[i+1]-idxs[i]-1)]
    
    for j in range(len(nums)):
        for row in rows:
            nums[j].append(row[idxs[i]+j])

    nums = [''.join(list(filter(lambda a: a != ' ', num))) for num in nums]
    nums = [int(num) for num in nums if num != '']

    if op == '+':
        res += sum(nums)
    if op == '*':
        res += reduce(lambda a,b: a*b, nums)

    # print(nums)

print(res)