test8a.py

test = False
file = 'data/task8test.txt' if test else 'data/task8.txt'

data = []
with open(file) as f:
    lines = f.read().splitlines()
    for line in lines:
        data.extend(line.split(' | ')[1].split(' '))

count = sum(list(map(lambda x: 1 if len(x) == 2 or len(x) == 3 or len(x) == 4 or len (x)==7 else 0, data)))
print(count)