sebse/python-aoc-2020
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Day 17 (Part 1 / Part 2 not working)

Browse Source
This commit is contained in:
Sebastian Seedorf
2020-12-17 11:31:04 +01:00
parent 0c36586b21
commit ce4e8322c9
3 changed files with 58 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
day17/input.txt Normal file
View File

@@ -0,0 +1,3 @@
.#.
..#
###

25
day17/part1.py Normal file
View File

@@ -0,0 +1,25 @@
#!/usr/bin/env python3
import numpy as np
from scipy import signal
lines = [[(1 if x == '#' else 0) for x in x.strip()] for x in open("input.txt")]
grid = np.array(lines, dtype=np.byte)
grid = np.expand_dims(grid, axis=0)
kernel = np.ones((3, 3, 3), dtype=np.byte)
kernel[1, 1, 1] = 0
for iter in range(6):
grid = np.pad(grid, pad_width=1, mode='constant', constant_values=0)
neighbors = signal.convolve(grid, kernel, mode='same')
set_inactive = np.logical_and(grid == 1, np.floor_divide(neighbors, 2) != 1)
set_active = np.logical_and(grid == 0, neighbors == 3)
grid[set_inactive] = 0
grid[set_active] = 1
x = np.flatnonzero(grid.sum(axis=(1, 2)))
y = np.flatnonzero(grid.sum(axis=(0, 2)))
z = np.flatnonzero(grid.sum(axis=(0, 1)))
grid = grid[x, :, :][:, y, :][:, :, z]
print(np.sum(grid))

30
day17/part2.py Normal file
View File

@@ -0,0 +1,30 @@
#!/usr/bin/env python3
import numpy as np
from scipy import signal
lines = [[(1 if x == '#' else 0) for x in x.strip()] for x in open("input.txt")]
grid = np.array(lines, dtype=np.byte)
grid = np.expand_dims(grid, axis=0)
grid = np.expand_dims(grid, axis=0)
print(grid.shape)
kernel = np.ones((3, 3, 3, 3), dtype=np.byte)
kernel[1, 1, 1, 1] = 0
for iter in range(2):
grid = np.pad(grid, pad_width=1, mode='constant', constant_values=0)
neighbors = signal.convolve(grid, kernel, mode='same')
set_inactive = np.logical_and(grid == 1, np.floor_divide(neighbors, 2) != 1)
set_active = np.logical_and(grid == 0, neighbors == 3)
grid[set_inactive] = 0
grid[set_active] = 1
x = np.flatnonzero(grid.sum(axis=(1, 2, 3)))
y = np.flatnonzero(grid.sum(axis=(0, 2, 3)))
z = np.flatnonzero(grid.sum(axis=(0, 1, 3)))
w = np.flatnonzero(grid.sum(axis=(0, 1, 2)))
grid = grid[x, :, :, :][:, y, :, :][:, :, z, :][:, :, :, w]
print(iter+1, grid.shape)
print(np.sum(grid))