batteries = [] with open("day3/input.txt") as file: lines = file.readlines() for line in lines: line = line.strip("\n") data = [] for joltage in line: data.append(int(joltage)) batteries.append(data) joltageSum = 0 for bank in batteries: tenthPlace = max(bank) onethPlace = 0 indexOfMax = bank.index(tenthPlace) if(indexOfMax == len(bank)-1): onethPlace = tenthPlace tenthPlace = max(bank[:-1]) else: onethPlace = max(bank[indexOfMax+1::]) joltageBank = int(f"{tenthPlace}{onethPlace}") print(f"Bank {batteries.index(bank)+1} joltage is {joltageBank}") joltageSum+=joltageBank print(joltageSum)