[SciPy-user] 2d trapz

Pauli Virtanen pav at iki.fi
Tue Mar 10 16:30:28 EDT 2009


Tue, 10 Mar 2009 20:54:40 +0100, Christian K. wrote:
[clip]
> I notice that I made not clear that by 'numerical' I meant integration
> of fixed sample data. So dblquad is not what I was looking for.

Use trapz twice?

import scipy as sp
import numpy as np
x = np.linspace(-10, 10, 200)
y = np.linspace(-10, 10, 80)

f = np.exp(-x[:,np.newaxis]**2 - y[np.newaxis,:]**2/7)/(np.sqrt(7)*np.pi)
print sp.trapz(sp.trapz(f, y[np.newaxis,:], axis=1), x, axis=0)
# -> 0.99999990566

    ***

There's a bug in Numpy <= 1.2.1 that requires that `f` and `y` to have 
same number of dimensions, so that's why the newaxis.

-- 
Pauli Virtanen




More information about the SciPy-User mailing list