# Vestigium - https://codingcompetitions.withgoogle.com/codejam/round/000000000019fd27/000000000020993c#problem ## Probleme Vestigium means "trace" in Latin # Codejam 2020, Qualification Round: Vestigium import numpy as np # I/O Code num_cases = int(input()) for case in range(1, num_cases + 1): N = int(input()) M = np.array([list(map(int, input().split())) for _ in range(N)]) K = np.trace(M) R = sum(len(set(r)) < N for r in M) C = sum(len(set(c)) < N for c in M.T) print('Case #{}: {} {} {}'.format(case, K, R, C)) from itertools import imap, izip def vestigium(): ... for case in xrange(input()): print 'Case #%d: %s' % (case+1, vestigium())