(N, Niter=100, func=num_update, args=())
| 32 | out=u[1:(-1), 1:(-1)]) |
| 33 | |
| 34 | def laplace(N, Niter=100, func=num_update, args=()): |
| 35 | u = np.zeros([N, N], order='C') |
| 36 | u[0] = 1 |
| 37 | for i in range(Niter): |
| 38 | func(u, *args) |
| 39 | return u |
| 40 | |
| 41 | func = {'inplace': num_inplace, 'normal': num_update}[update] |
| 42 |