bert/adventofcode
1
0
Fork 0
mirror of https://github.com/bertptrs/adventofcode.git synced 2025-12-25 21:00:31 +01:00
Code Issues Projects Releases Wiki Activity

More numpy for speed

Browse Source
This commit is contained in:
Bert Peters
2024-12-25 10:34:40 +01:00
parent 4a479c1646
commit f3a3e1fca3
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
2024/src/aoc/days/day25.py
View File

@@ -20,11 +20,10 @@ class DayRunner(CombinedRunner):
else: else:
keys.append(heights) keys.append(heights)
fitting = 0 locks = numpy.stack(locks, axis=0)
for key in keys: fitting = sum(
for lock in locks: numpy.count_nonzero(numpy.all((key + locks) <= 7, axis=1)) for key in keys
if numpy.all((key + lock) <= 7): )
fitting += 1
return fitting, None return fitting, None