src/a4.py

   1 #!/usr/bin/env python3
   2 import sys
   3 import re
   4
   5 iterations = {}
   6
   7 for line in sys.stdin:
   8     match = re.match('\(= i(?P<i>\d+)a(?P<a>\d+) (?P<v>\d+)\)', line)
   9     if match:
  10         iterations[match.group('i')] = iterations.get(match.group('i'), {})
  11         iterations[match.group('i')][match.group('a')] = match.group('v')
  12
  13 print('$\\begin{array}{|l|lllllll|}')
  14 print('\t\\toprule')
  15 print('\t\\text{{n}} & a_1 & {} & a_7\\\\'.format(
  16     ' & '.join('a_{}'.format(a) for a in sorted(iterations['0']))))
  17 print('\t\\midrule')
  18 for i in sorted(iterations, key=int):
  19     print('\t{} & 1 & {} & 7\\\\'.format(i, ' & '.join(
  20           iterations[i][a] for a in sorted(iterations[i], key=int)
  21             )))
  22 print('\t\\bottomrule')
  23 print('\\end{array}$')