day 6: numpy

1 files changed, 23 insertions, 0 deletions

diff --git a/6np.py b/6np.py
new file mode 100644
index 0000000..6b97f9c
--- /dev/null
+++ b/6np.py
@@ -0,0 +1,23 @@
+from collections import Counter
+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],
+    [1, 0, 0, 0, 0, 0, 0, 0, 0],
+    [0, 1, 0, 0, 0, 0, 0, 0, 0],
+    [0, 0, 1, 0, 0, 0, 0, 0, 0],
+    [0, 0, 0, 1, 0, 0, 0, 0, 0],
+    [0, 0, 0, 0, 1, 0, 0, 0, 0],
+    [0, 0, 0, 0, 0, 1, 0, 0, 0],
+    [0, 0, 0, 0, 0, 0, 1, 0, 0],
+    [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))