#!/usr/bin/env python # -*- coding: utf-8 -*- import csv inputs = ["part2.txt", "input.txt"] def solve(filename): total = 0 with open(filename) as tsv: for line in csv.reader(tsv, dialect="excel-tab"): items = [] for item in line: i = int(item) items.append(i) for i in range(len(items)): for i2 in range(i + 1, len(items)): a1 = items[i] / items[i2] a2 = items[i2] / items[i] if float.is_integer(a1): total += a1 elif float.is_integer(a2): total += a2 return int(total) for puzzle in inputs: print(puzzle, solve(puzzle))