examples/trees.py

   1 #!/usr/bin/env python
   2 import iam_sympy_example
   3
   4 from sympy import *
   5
   6 def T(x):
   7     return x+x**2+2*x**3 + 4*x**4 + 9*x**5 + 20*x**6 + 48 * x**7 + \
   8             115* x**8 + 286*x**9+719*x**10
   9
  10 def A(x):
  11     return 1 + T(x) - T(x)**2/2 + T(x**2)/2
  12
  13
  14 x=Symbol("x")
  15 s = Poly(A(x), x)
  16 num = [s.coeff(n) for n in range(11)]
  17
  18 print s.as_basic()
  19 print num