Only use the symbol '*' in part 2

2023/day03/solutions.py

@@ -8,7 +8,7 @@
 
 box = list(itertools.product((-1, 0, 1), (-1, 0, 1)))
 symbols = {
-    (i, j)
+    (i, j): x
     for i, l in enumerate(ls)
     for j, x in enumerate(l)
     if x != "." and not x.isdigit()
@@ -25,13 +25,19 @@
             for di, dj in box
             for j in range(match.start(), match.end())
         }
-        if symbols & boundary:
+        if symbols.keys() & boundary:
             part_sum += n
-        for symbol in symbols & boundary:
+        for symbol in symbols.keys() & boundary:
             parts_by_symbol[symbol].append(n)
 
 # Part 1
 print(part_sum)
 
 # Part 2
-print(sum(math.prod(v) for v in parts_by_symbol.values() if len(v) == 2))
+print(
+    sum(
+        math.prod(v)
+        for k, v in parts_by_symbol.items()
+        if len(v) == 2 and symbols[k] == "*"
+    )
+)