day8.py

import numpy as np
import re


def rect(a, b):
    g[:b, :a] = 1

def rotate_row(a, b):
    g[a, :] = np.append(g[a, -b:], g[a, :-b])

def rotate_col(a, b):
    g[:, a] = np.append(g[-b:, a], g[:-b, a])

# test
g = np.zeros((3, 7))
rect(3, 2)
rotate_col(1, 1)
rotate_row(0, 4)
rotate_col(1, 1)

g = np.zeros((6, 50))
for line in open('data/day8.txt'):
    if not line.strip(): continue
    ns = list(map(int, re.findall('\d+', line)))
    if 'rect' in line:
        rect(*ns)
    elif 'row' in line:
        rotate_row(*ns)
    else:
        rotate_col(*ns)

print(np.sum(g))

print()
for i in range(6):
    for j in range(50):
        print('.' if g[i, j] else ' ', end='' if j < 49 else '\n')