diff options
Diffstat (limited to 'doc/numpybook/comparison/weave')
| -rw-r--r-- | doc/numpybook/comparison/weave/filter.py | 11 | ||||
| -rw-r--r-- | doc/numpybook/comparison/weave/inline.py | 10 |
2 files changed, 10 insertions, 11 deletions
diff --git a/doc/numpybook/comparison/weave/filter.py b/doc/numpybook/comparison/weave/filter.py index b2fdb277e..6fc16c79f 100644 --- a/doc/numpybook/comparison/weave/filter.py +++ b/doc/numpybook/comparison/weave/filter.py @@ -8,11 +8,11 @@ def filter(a): for(i=1;i<Na[0]-1;i++) { for(j=1;j<Na[1]-1;j++) { B2(i,j) = A2(i,j) + (A2(i-1,j) + - A2(i+1,j) + A2(i,j-1) - + A2(i,j+1))*0.5 - + (A2(i-1,j-1) - + A2(i-1,j+1) - + A2(i+1,j-1) + A2(i+1,j) + A2(i,j-1) + + A2(i,j+1))*0.5 + + (A2(i-1,j-1) + + A2(i-1,j+1) + + A2(i+1,j-1) + A2(i+1,j+1))*0.25; } } @@ -20,4 +20,3 @@ def filter(a): b = zeros_like(a) weave.inline(code,['a','b']) return b - diff --git a/doc/numpybook/comparison/weave/inline.py b/doc/numpybook/comparison/weave/inline.py index 31499213e..bfac588aa 100644 --- a/doc/numpybook/comparison/weave/inline.py +++ b/doc/numpybook/comparison/weave/inline.py @@ -24,11 +24,11 @@ def arr(a): for(i=1;i<Na[0]-1;i++) { for(j=1;j<Na[1]-1;j++) { B2(i,j) = A2(i,j) + A2(i-1,j)*0.5 + - A2(i+1,j)*0.5 + A2(i,j-1)*0.5 - + A2(i,j+1)*0.5 - + A2(i-1,j-1)*0.25 - + A2(i-1,j+1)*0.25 - + A2(i+1,j-1)*0.25 + A2(i+1,j)*0.5 + A2(i,j-1)*0.5 + + A2(i,j+1)*0.5 + + A2(i-1,j-1)*0.25 + + A2(i-1,j+1)*0.25 + + A2(i+1,j-1)*0.25 + A2(i+1,j+1)*0.25; } } |