"""Decode stupid 7 segment displays""" count = 0 with open("8_input.txt") as f: for line in f: digits = line.split("|")[1].strip().split(" ") for digit in digits: if len(digit) in [2, 3, 4, 7]: # Corresponds to 1, 7, 4, 8 count += 1 print(count)