input_file = "input.txt" banks: list[list[int]] = [] with open(input_file, 'r') as file: banks = [ [ int(n) for n in line.strip() ] for line in file.readlines() ] total_joltage = 0 for i, bank in enumerate(banks): n1 = max(bank[:-1]) n2 = max(bank[bank.index(n1) + 1:]) print(f'Largest Joltage for Bank {i + 1}: {n1}{n2}') total_joltage += 10 * n1 + n2 print(f'Total Joltage: {total_joltage}')