first

Day3_1/main.py

@@ -0,0 +1,60 @@
+from aocd import get_data
+from aocd import submit
+input_data = get_data(day=3, year=2021)
+
+
+
+test_data = """00100
+11110
+10110
+10111
+10101
+01111
+00111
+11100
+10000
+11001
+00010
+01010"""
+
+lines = input_data.split("\n")
+# lines = test_data.split("\n")
+
+line_len = len(lines[0])
+
+counts_ones = [0] * line_len
+counts_zeroes = [0] * line_len
+
+for line in lines:
+    power = list(line)
+    power = [int(i) for i in power]
+    for index, bit in enumerate(power):
+      if bit == 1:
+          counts_ones[index] += 1
+      elif bit == 0:
+          counts_zeroes[index] += 1
+
+print(counts_ones)
+print(counts_zeroes)
+
+#loop through bits and find highest lowest
+gamma = [0] * line_len
+epsilon = [0] * line_len
+for i in range(line_len):
+    print(i)
+    if counts_ones[i] > counts_zeroes[i]:
+       gamma[i] = 1
+       epsilon[i] = 0
+    elif counts_ones[i] < counts_zeroes[i]:
+       gamma[i] = 0
+       epsilon[i] = 1
+
+# print(gamma)
+# print(epsilon)
+
+gamma = ''.join(str(i) for i in gamma)
+epsilon = ''.join(str(i) for i in epsilon)
+
+powerconsumption = int(gamma, 2) * int(epsilon, 2) 
+print(powerconsumption)
+# submit(depth*horizontal)