Abelian sandpile model: Difference between revisions

Content added Content deleted
(Added Python solution)
Line 542: Line 542:
main()</lang>
main()</lang>

=={{header|Python}}==
<lang Python>
import numpy as np
import matplotlib.pyplot as plt


def iterate(grid):
changed = False
for ii, arr in enumerate(grid):
for jj, val in enumerate(arr):
if val > 3:
grid[ii, jj] -= 4
if ii > 0:
grid[ii - 1, jj] += 1
if ii < len(grid)-1:
grid[ii + 1, jj] += 1
if jj > 0:
grid[ii, jj - 1] += 1
if jj < len(grid)-1:
grid[ii, jj + 1] += 1
changed = True
return grid, changed


def simulate(grid):
while True:
grid, changed = iterate(grid)
if not changed:
return grid


if __name__ == '__main__':
start_grid = np.zeros((10, 10))
start_grid[4:5, 4:5] = 64
final_grid = simulate(start_grid.copy())
plt.figure()
plt.gray()
plt.imshow(start_grid)
plt.figure()
plt.gray()
plt.imshow(final_grid)
</lang>
<b>Output:</b> </n>
Before:
<lang Python>
[[0. 0. 0. 0. 0. 0. 0. 0. 0. 0.]
[0. 0. 0. 0. 0. 0. 0. 0. 0. 0.]
[0. 0. 0. 0. 0. 0. 0. 0. 0. 0.]
[0. 0. 0. 0. 0. 0. 0. 0. 0. 0.]
[0. 0. 0. 0.64. 0. 0. 0. 0. 0.]
[0. 0. 0. 0. 0. 0. 0. 0. 0. 0.]
[0. 0. 0. 0. 0. 0. 0. 0. 0. 0.]
[0. 0. 0. 0. 0. 0. 0. 0. 0. 0.]
[0. 0. 0. 0. 0. 0. 0. 0. 0. 0.]
[0. 0. 0. 0. 0. 0. 0. 0. 0. 0.]]
</lang>
After:
<lang Python>
[[0. 0. 0. 0. 0. 0. 0. 0. 0. 0.]
[0. 0. 0. 1. 2. 1. 0. 0. 0. 0.]
[0. 0. 2. 2. 2. 2. 2. 0. 0. 0.]
[0. 1. 2. 2. 2. 2. 2. 1. 0. 0.]
[0. 2. 2. 2. 0. 2. 2. 2. 0. 0.]
[0. 1. 2. 2. 2. 2. 2. 1. 0. 0.]
[0. 0. 2. 2. 2. 2. 2. 0. 0. 0.]
[0. 0. 0. 1. 2. 1. 0. 0. 0. 0.]
[0. 0. 0. 0. 0. 0. 0. 0. 0. 0.]
[0. 0. 0. 0. 0. 0. 0. 0. 0. 0.]]
</lang>


=={{header|Rust}}==
=={{header|Rust}}==