import 2021

2021/day6/day6_2.py

@@ -0,0 +1,34 @@
+
+total = 0
+numbers = [0, 0, 0, 0, 0, 0, 0, 0, 0]
+
+with open("input.txt") as f:
+    data = [int(x) for x in f.readline().split(',')]
+
+for i in data:
+    if i == 0:
+        numbers[0] += 1
+    if i == 1:
+        numbers[1] += 1
+    if i == 2:
+        numbers[2] += 1
+    if i == 3:
+        numbers[3] += 1
+    if i == 4:
+        numbers[4] += 1
+    if i == 5:
+        numbers[5] += 1
+    if i == 6:
+        numbers[6] += 1
+    if i == 7:
+        numbers[7] += 1
+    if i == 8:
+        numbers[8] += 1
+
+def rotate(l):
+    return l[1:] + l[:1]
+
+for j in range(256):
+    numbers = rotate(numbers)
+    numbers[6] += numbers[8]
+    print(f'DAY {j+1} AMOUNT OF FISH: {sum(numbers)}')