symbolic.py

   1 #!/usr/bin/python
   2
   3 from sympy import *
   4
   5 import sys
   6
   7 #import sys, codecs
   8 #if sys.stdout.encoding is None:
   9 #  sys.stdout = codecs.open("/dev/stdout", "w", 'utf-8')
  10
  11 init_printing(use_unicode=False)
  12
  13 r, theta, s, la, mu = symbols('r theta s la mu')
  14
  15 pprint(theta)
  16
  17 # start      original formulation
  18 # rightvars  replaces
  19
  20 p_start = Matrix([
  21   r * (1 - cos(theta)),
  22   r * sin(theta),
  23   mu * s,
  24 ])
  25
  26 p_rightvars = p_start.subs(theta, s/r).subs(r, 1/la)
  27
  28 pprint(p_rightvars)