sebse/python-aoc-2021
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Day 09 (minified)

Browse Source
This commit is contained in:
Sebastian Seedorf
2021-12-10 10:13:34 +01:00
parent ff22d259e8
commit 8efc7412a7
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
day09/part1.py
View File

@@ -9,8 +9,8 @@ arr = np.array([[int(n) for n in line.strip()] for line in open("input.txt")])
# https://stackoverflow.com/questions/3986345/how-to-find-the-local-minima-of-a-smooth-multidimensional-array-in-numpy-efficie
neighborhood = morphology.generate_binary_structure(len(arr.shape), 1)
local_min = (filters.minimum_filter(arr, footprint=neighborhood) == arr)
local_max = (filters.maximum_filter(arr, footprint=neighborhood) == arr)
local_min = filters.minimum_filter(arr, footprint=neighborhood) == arr
local_max = filters.maximum_filter(arr, footprint=neighborhood) == arr
local_min_without_plateau = np.logical_and(local_min, np.logical_not(local_max))
local_min_locations = np.where(local_min_without_plateau)
local_min_values = arr[local_min_locations]