day25.py

#!/usr/bin/env python3
import numpy as np
import regex as re
from collections import defaultdict
from functools import lru_cache as cache

f = open("ressources/day25.txt", "r") #Open File
lines = f.readlines() #Separate in lines

area = []
keys = list()
locks = list()

separations = [-1]+[i for i, line in enumerate(lines) if line == "\n"]
for sep in separations:
    start = sep+1
    end = sep+8
    area = np.array([list(line.replace("\n", "").replace("#", "1").replace(".", "0")) for line in lines[start:end]])
    area = area.astype(int)
    isKey = np.sum(area[0]) == 0
    
    value = np.sum(area, axis=0)-1
    if isKey:
        keys.append(value)
    else:
        locks.append(value)
        
pairs = 0
for key in keys:
    for lock in locks:
        total = key+lock
        if np.all(total < 6):
            pairs += 1
            
print("Total of unique key/locks pairs is {}".format(pairs))