Solve KPP equation using py-pde

pypde_kpp.py

import numpy as np
import pde

# define PDE
eq = pde.PDE({'u': 'laplace(u) + u*(1-u)'})

# define grid
grid = pde.UnitGrid([32, 32])

# define initial condition
c = np.array([16, 16])
f = lambda x: np.heaviside(1 - np.linalg.norm(x - c), 1)

# hack
data = np.empty(grid.shape)
for cells in np.ndindex(*grid.shape):
    data[cells] = f(grid.cell_coords[cells])

state = pde.ScalarField(grid=grid, data=data)

result = eq.solve(state, t_range=10) # solve
result.plot()