Day 4! Needed numpy help for 04a but 04b was MUCH easier

Author: m4xic

04/04a.py

@@ -0,0 +1,51 @@
+import numpy as np
+import re
+
+with open('input', 'r') as f: source = f.read().split('\n')
+
+matrix = []
+
+for y in source:
+    row = []
+    for x in range(len(y)):
+        row.append(y[x])
+    matrix.append(row)
+
+print(matrix)
+
+npm = np.matrix(matrix)
+print(npm)
+
+# No way I could've done this without numpy and this answer: https://stackoverflow.com/a/6313414
+npdiags = [npm[::-1,:].diagonal(i) for i in range(-npm.shape[0]+1,npm.shape[1])]
+npdiags.extend(npm.diagonal(i) for i in range(npm.shape[1]-1,-npm.shape[0],-1))
+diags = [ i.tolist() for i in npdiags ]
+
+print(diags)
+
+total = 0
+
+# Search rows
+for y in matrix:
+    row = ''.join(y)
+    # This total adding thing shoulda been a function
+    total += len(re.findall(r'XMAS', row))
+    total += len(re.findall(r'XMAS', row[::-1]))
+
+print(f'{total=}')
+
+# Search columns
+for x in range(len(matrix[0])):
+    column = ''.join(npm[:, x].flatten().tolist()[0])
+    total += len(re.findall(r'XMAS', column))
+    total += len(re.findall(r'XMAS', column[::-1]))
+
+print(f'{total=}')
+
+# Search diagonals
+for d in diags:
+    diag = ''.join(d[0])
+    total += len(re.findall(r'XMAS', diag))
+    total += len(re.findall(r'XMAS', diag[::-1]))
+
+print(f'{total=}')

04/04b.py

@@ -0,0 +1,31 @@
+with open('input', 'r') as f: source = f.read().split('\n')
+
+matrix = []
+
+for y in source:
+    row = []
+    for x in range(len(y)):
+        row.append(y[x])
+    matrix.append(row)
+
+print(matrix)
+
+count = 0
+
+# M A S is 3 long so
+for y in range(len(matrix) - 2):
+    for x in range(len(matrix[x]) - 2):
+        print(f'{x},{y} = {matrix[y][x]}')
+
+        tl = matrix[y][x]
+        tr = matrix[y][x+2]
+        mi = matrix[y+1][x+1]
+        bl = matrix[y+2][x]
+        br = matrix[y+2][x+2]
+
+        if (tl == "M" and br == "S") or (tl == "S" and br == "M"):
+            if (tr == "S" and bl == "M") or (tr == "M" and bl == "S"):
+                if (mi == "A"):
+                    count += 1
+
+print(f'{count=}')