import os dir = os.path.dirname(os.path.abspath(__file__)) with open(os.path.join(dir, './input.txt'), 'r') as input_file: lines = input_file.readlines() ones = 0 fours = 0 sevens = 0 eights = 0 test = False for line in lines: input_segments = "" output_segments = "" if test == True: if '|' in line: input_segments = line.strip() else: output_segments = line.strip() else: parts = line.split('|') input_segments = parts[0].strip() output_segments = parts[1].strip() if output_segments == "": continue output_parts = output_segments.split() for output in output_parts: num_signals = len(output) if num_signals == 2: ones = ones + 1 elif num_signals == 3: sevens = sevens + 1 elif num_signals == 4: fours = fours + 1 elif num_signals == 7: eights = eights + 1 print(ones+fours+sevens+eights)