Rename inv_reduce to unfold to align with other implementations of the concept

This commit is contained in:
Luke Hubmayer-Werner 2021-12-21 00:46:22 +10:30
parent 52a57c0249
commit a84fdc6207
1 changed files with 2 additions and 2 deletions

View File

@ -1,6 +1,6 @@
import numpy as np import numpy as np
def inv_reduce(func, arg, n): def unfold(func, arg, n):
for _ in range(n): for _ in range(n):
arg = func(arg) arg = func(arg)
yield arg yield arg
@ -22,6 +22,6 @@ def visualize(array):
print('\n'.join(''.join(['.','#'][c] for c in row) for row in array)) print('\n'.join(''.join(['.','#'][c] for c in row) for row in array))
simulated_lights = [t.sum() for t in inv_reduce(sample, np.pad(input_arr, 51), 50)] simulated_lights = [t.sum() for t in unfold(sample, np.pad(input_arr, 51), 50)]
print('Part 1: Lights after 2 enhancements:', simulated_lights[2-1]) print('Part 1: Lights after 2 enhancements:', simulated_lights[2-1])
print('Part 2: Lights after 50 enhancements:', simulated_lights[50-1]) print('Part 2: Lights after 50 enhancements:', simulated_lights[50-1])