Day 17 (Part 2)

This commit is contained in:
Sebastian Seedorf
2020-12-17 11:52:45 +01:00
parent ce4e8322c9
commit ad8a6b1a67
3 changed files with 13 additions and 6 deletions

View File

@@ -20,6 +20,6 @@ for iter in range(6):
x = np.flatnonzero(grid.sum(axis=(1, 2)))
y = np.flatnonzero(grid.sum(axis=(0, 2)))
z = np.flatnonzero(grid.sum(axis=(0, 1)))
grid = grid[x, :, :][:, y, :][:, :, z]
grid = grid[min(x):max(x)+1, :, :][:, min(y):max(y)+1, :][:, :, min(z):max(z)+1]
print(np.sum(grid))