From f5448cb76e52586d1a0957ee6bd54bba2372b49e Mon Sep 17 00:00:00 2001 From: Tomasz Kramkowski Date: Mon, 6 Dec 2021 08:52:21 +0000 Subject: day 6 numpy: now with linalg.matrix_power --- 6np.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/6np.py b/6np.py index 6b97f9c..92273b1 100644 --- a/6np.py +++ b/6np.py @@ -3,7 +3,6 @@ from utils import open_day import numpy as np fish = Counter(map(int, open_day(6).read().split(','))) - fish = np.array([fish[i] for i in range(9)]) mat = np.array([ [0, 0, 0, 0, 0, 0, 1, 0, 1], @@ -17,7 +16,5 @@ mat = np.array([ [0, 0, 0, 0, 0, 0, 0, 1, 0] ]) -for i in range(256): - if i == 80: print(sum(fish)) - fish = np.matmul(fish, mat) -print(sum(fish)) +print(sum(np.matmul(fish, np.linalg.matrix_power(mat, 80)))) +print(sum(np.matmul(fish, np.linalg.matrix_power(mat, 256)))) -- cgit v1.2.3-54-g00ecf