some more solutions

2025/aoc/helpers.py

@@ -28,9 +28,34 @@ def read_ints(day: int) -> list[int]:
     return read_lines(day, int)
 
 
-def read_grid(day: int, dtype=str) -> np.ndarray:
-    """Read input as a 2D numpy grid of characters or digits."""
+def read_grid(day: int, dtype=str, split=False, keep_spaces=False, width=None):
+    """Read input as a 2D grid.
+
+    split=False: each character is a cell (default)
+    split=True:  split on whitespace
+    split=',':   split on specific delimiter
+    keep_spaces: preserve all whitespace (don't strip)
+    width=5:     fixed-width columns (5 chars each)
+    """
+    if width:
+        lines = get_input_path(day).read_text().splitlines()
+        rows = []
+        for line in lines:
+            row = [line[i:i+width].strip() for i in range(0, len(line), width)]
+            rows.append(row)
+        return np.array(rows)
+    if keep_spaces:
+        lines = get_input_path(day).read_text().splitlines()
+        max_len = max(len(line) for line in lines)
+        rows = [list(line.ljust(max_len)) for line in lines]
+        return rows
     lines = read_input(day).splitlines()
+    if split:
+        delim = None if split is True else split
+        rows = [line.split(delim) for line in lines]
+        if dtype == int:
+            return np.array([[int(x) for x in row] for row in rows])
+        return np.array(rows)
     if dtype == int:
         return np.array([[int(c) for c in line] for line in lines])
     return np.array([list(line) for line in lines])
@@ -41,6 +66,53 @@ def read_blocks(day: int) -> list[str]:
     return read_input(day).split("\n\n")
 
 
+def to_digit_grids(grid):
+    """Convert whitespace-split grid to list of digit grids.
+
+    Input:  [['123', '328'], ['45', '64'], ['6', '98']]
+    Output: [
+        [['1','2','3'], ['4','5',''], ['6','','']],  # kolumn 1
+        [['3','2','8'], ['6','4',''], ['9','8','']]  # kolumn 2
+    ]
+    """
+    cols = list(zip(*grid))
+    grids = []
+    for col in cols:
+        max_len = max(len(s) for s in col)
+        digit_grid = [[c for c in s] + [''] * (max_len - len(s)) for s in col]
+        grids.append(np.array(digit_grid))
+    return grids
+
+
+def split_grid_on_cols(grid):
+    """Split grid on whitespace columns.
+
+    Input: grid from read_grid(day, keep_spaces=True)
+    Returns: list of numpy arrays, one per column group
+    """
+    arr = np.array(grid)
+    empty_cols = np.all(arr == ' ', axis=0)
+
+    # Hitta start/slut för varje grupp (icke-tomma kolumner)
+    result = []
+    in_group = False
+    start = 0
+
+    for i, is_empty in enumerate(empty_cols):
+        if not is_empty and not in_group:
+            start = i
+            in_group = True
+        elif is_empty and in_group:
+            result.append(arr[:, start:i])
+            in_group = False
+
+    # Sista gruppen
+    if in_group:
+        result.append(arr[:, start:])
+
+    return result
+
+
 # Common grid directions
 DIRS_4 = [(0, 1), (1, 0), (0, -1), (-1, 0)]  # right, down, left, up
 DIRS_8 = DIRS_4 + [(1, 1), (1, -1), (-1, 1), (-1, -1)]  # + diagonals