From d4f482804b9777768fa4f9fedde0cd549578502b Mon Sep 17 00:00:00 2001 From: Brian Chu Date: Sun, 19 Dec 2021 22:57:37 -0800 Subject: catch up on solutions up to day 20 --- day20.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) create mode 100644 day20.py (limited to 'day20.py') diff --git a/day20.py b/day20.py new file mode 100644 index 0000000..74ca7fd --- /dev/null +++ b/day20.py @@ -0,0 +1,19 @@ +#!/usr/bin/env python +import numpy as np +from scipy.ndimage import convolve + +with open('day20.txt') as data: + algo, image = data.read().split('\n\n') + algo = np.array(list(algo)) + algo = (algo == '#').astype(int) + image = np.stack(list(map(list, image.strip().split('\n')))) + image = (image == '#').astype(int) + image = np.pad(image, (100, 100)) + +weights = 2 ** np.arange(9).reshape(3,3) +# part 1 +for i in range(50): + image = algo[convolve(image, weights)] + if i == 1: + print(image.sum()) +print(image.sum()) -- cgit 1.4.1-2-gfad0