X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=mjo%2Forthogonal_polynomials.py;h=589aa80b63def36bc6503fbbea7dc18ab1f7c104;hb=8bed8a3c21a0d19d01a7625b3849ac07c9a9f9e6;hp=7e094fc201076938120972f268b55fd628470a47;hpb=930b77721b3dd6d00cc4f53f43c78cc6c16e2ed5;p=sage.d.git diff --git a/mjo/orthogonal_polynomials.py b/mjo/orthogonal_polynomials.py index 7e094fc..589aa80 100644 --- a/mjo/orthogonal_polynomials.py +++ b/mjo/orthogonal_polynomials.py @@ -78,11 +78,11 @@ def legendre_p(n, x, a = -1, b = 1): ... return sqrt(inner_product(v,v)) ... sage: def project(basis, v): - ... return sum([ inner_product(v, b)*b/norm(b)**2 - ... for b in basis]) + ... return sum( inner_product(v, b)*b/norm(b)**2 + ... for b in basis) ... sage: f = sin(x) - sage: legendre_basis = [ legendre_p(k, x, a, b) for k in range(0,4) ] + sage: legendre_basis = [ legendre_p(k, x, a, b) for k in xrange(4) ] sage: proj = project(legendre_basis, f) sage: proj.simplify_trig() 5/2*(7*(pi^2 - 15)*x^3 - 3*(pi^4 - 21*pi^2)*x)/pi^6 @@ -92,7 +92,7 @@ def legendre_p(n, x, a = -1, b = 1): We should agree with Maxima for all `n`:: sage: eq = lambda k: bool(legendre_p(k,x) == legendre_P(k,x)) - sage: all([eq(k) for k in range(0,20) ]) # long time + sage: all( eq(k) for k in xrange(20) ) # long time True We can evaluate the result of the zeroth polynomial:: @@ -196,6 +196,6 @@ def legendre_p(n, x, a = -1, b = 1): # From Abramowitz & Stegun, (22.3.2) with alpha = beta = 0. # Also massaged to support finite field elements. - P = sum([ c(m)*g(m) for m in range(0,n+1) ])/(2**n) + P = sum( c(m)*g(m) for m in xrange(n+1) )/(2**n) return P