Sympy符号计算库

http://blog.csdn.net/pipisorry/article/details/39123247

SymPy是Python的一个数学符号计算库,目的在于成为一个富有特色的计算机代数系统。

Sympy基本概念

安装

pip install sympy

 SymPy的下载(Downloads)网页

内建的数字类型

SymPy有三个内建的数值类型:实数,有理数和整数。

有理数:Rational(1,2)

有理数类用两个整数来表示分子与分母,Rational(1,2)代表1/2等等。

>>>from sympy import *
>>>a=Rational(1,2)
>>>a
1/2
>>>a*2
1
>>>Rational(2)**50/Rational(10)**50
1/88817841970012523233890533447265625

特殊的常数:e pi

有一些特殊的常数像e、pi和无穷大oo,它们会被当作符号去对待。(1+pi不会求得值,反而它会保持为1+pi)。调用求表达式的浮点数-evalf()函数才会求值。

示例:

>>>pi**2
pi**2
>>>pi.evalf()
3.14159265358979
>>>(pi+exp(1)).evalf()
5.85987448204884

>>>oo > 99999

True

>>>oo + 1

oo

求表达式的浮点数函数evalf()

最好将x替换操作传到evalf参数里。If the substitution will be followed by numerical evaluation, it is better to pass the substitution to evalf.

>>> (1/x).evalf(subs={x: 3.0}, n=21)
0.333333333333333333333
rather than
>>> (1/x).subs({x: 3.0}).evalf(21)
0.333333333333333314830

皮皮blog

Sympy基本使用

定义变量函数-Symbols

对比其他的计算机代数系统,在SymPy中要明确声明符号变量:
from sympy import *

x=Symbol("x")
x = symbols('x')
x,y,z=symbols('x y z')
crazy = symbols('unrelated')

k, m, n = symbols(‘kmn’, integer=True)

from sympy.abc import x, y    #应该是定义好的

f = Function(“f”)

示例:

>>> x = symbols('x')
>>> x + 1
x + 1
>>>x,y,z=symbols('x y z')
>>> crazy = symbols('unrelated')
>>> crazy + 1
unrelated + 1

>>>x+y+x-y
2*x
>>> x = symbols('x')
>>> expr = x + 1
>>> x = 2 #Changing x to 2 had no effect on expr. This is because x = 2 changes the Python variable x to 2, but has no effect on the SymPy Symbol x, which was what we used in creating expr.
>>> print(expr)
x + 1

变量展开expand()

((x+y)**2).expand()

2*x*y + x**2 + y**2

变量替换函数subs

示例:

>>> x = symbols('x')
>>> expr = x + 1
>>> expr.subs(x, 2)
3
>>> from sympy import pi, exp, limit, oo
>>> from sympy.abc import x, y
>>> (1 + x*y).subs(x, pi)
pi*y + 1
>>> (1 + x*y).subs({x:pi, y:2})
1 + 2*pi
>>> (1 + x*y).subs([(x, pi), (y, 2)])
1 + 2*pi
>>> reps = [(y, x**2), (x, 2)]
>>> (x + y).subs(reps)
6
>>> (x + y).subs(reversed(reps))
x**2 + 2
>>> (x**2 + x**4).subs(x**2, y)
y**2 + y
>>> (x**2 + x**4).xreplace({x**2: y})
x**4 + y

simultaneous参数用法
>>> (x/y).subs([(x, 0), (y, 0)])
0
>>> (x/y).subs([(x, 0), (y, 0)], simultaneous=True)
nan
>>> ((x + y)/y).subs({x + y: y, y: x + y})
1
>>> ((x + y)/y).subs({x + y: y, y: x + y}, simultaneous=True)
y/(x + y)

[subs(*args, **kwargs)]

示例

from sympy import symbols
cnt1turn = 3000 / 100
xy1 = (1 * cnt1turn, 100)
xy2 = (30 * cnt1turn, 10)

x = symbols('x')

k1000 = round((xy1[1] - xy2[1]) / (xy1[0] - xy2[0]), 4)
y1000 = k1000 * (x - xy1[0]) + xy1[1]
print(y1000)

103.102 - 0.1034*x

代数

局部的代数式展开apart(expr, x)

示例:

In [1]: 1/( (x+2)*(x+1) )
       1
───────────────
(2 + x)*(1 + x)


In [2]: apart(1/( (x+2)*(x+1) ), x)
  1       1
───── - ─────
1 + x   2 + x


In [3]: (x+1)/(x-1)
-(1 + x)
────────
 1 - x


In [4]: apart((x+1)/(x-1), x)
      2
1 - ─────
    1 - x

代数式的合并together(expr, x)

(相当于展开的逆运算)

In [7]: together(1/x + 1/y + 1/z)
x*y + x*z + y*z
───────────────
     x*y*z


In [8]: together(apart((x+1)/(x-1), x), x)
-1 - x
──────
1 - x


In [9]: together(apart(1/( (x+2)*(x+1) ), x), x)
       1
───────────────
(2 + x)*(1 + x)

皮皮blog

微积分

极限limit

在sympy中极限遵循语法limit(function, variable, point) ,如计算x->0时f(x)的极限limit(f, x, 0)。

>>>from sympy import *
>>>x=Symbol("x")
>>>limit(sin(x)/x, x, 0)
1
>>>limit(x, x, oo)
oo
>>>limit(1/x, x, oo)
0
>>>limit(x**x, x, 0)
1

>>> limit(x**3 - 3*x, x, oo)
oo

有一些特殊的极限的例子,你可以阅读文件test_demidovich.py

微分diff

可以对任意SymPy表达式微分diff(func, var)。

>>>from sympy import *
>>>x = Symbol('x')
>>>diff(sin(x), x)
cos(x)
>>>diff(sin(2*x), x)
2*cos(2*x)
>>>diff(tan(x), x)
1 + tan(x)**2
验证
>>>limit((tan(x+y)-tan(x))/y, y, 0)
1 + tan(x)**2
计算高阶微分 diff(func, var, n) 
>>>diff(sin(2*x), x, 1)
2*cos(2*x)
>>>diff(sin(2*x), x, 2)
-4*sin(2*x)
>>>diff(sin(2*x), x, 3)
-8*cos(2*x)

积分integrate

SymPy支持不定积分,超越函数与特殊函数的定积分。SymPy有力的扩展Risch-Norman 算法和模型匹配算法。

>>>from sympy import *
 >>>x, y = symbols('xy')
 初等函数:
 >>>integrate(6*x**5, x)
 x**6
 >>>integrate(sin(x), x)
 -cos(x)
 >>>integrate(log(x), x)
 -x + x*log(x)
 >>>integrate(2*x + sinh(x), x)
 cosh(x) + x**2
 特殊函数:
 >>>integrate(exp(-x**2)*erf(x), x)
 pi**(1/2)*erf(x)**2/4
 定积分:
 >>>integrate(x**3, (x, -1, 1))
 0
 >>>integrate(sin(x), (x, 0, pi/2))
 1
 >>>integrate(cos(x), (x, -pi/2, pi/2))
 2
 一些广义积分也可以被支持:
 >>>integrate(exp(-x), (x, 0, oo))
 1
 >>>integrate(log(x), (x, 0, 1))
 -1

级数展开series

函数series(var, point, order)

示例1:
 >>>from sympy import *
 >>>x = Symbol('x')
 >>>cos(x).series(x, 0, 10)
 1 - x**2/2 + x**4/24 - x**6/720 + x**8/40320 + O(x**10)
 >>>(1/cos(x)).series(x, 0, 10)
 1 + x**2/2 + 5*x**4/24 + 61*x**6/720 + 277*x**8/8064 + O(x**10)
示例2:
 from sympy import Integral, Symbol, pprint
 x = Symbol("x")
 y = Symbol("y")
 e = 1/(x + y)
 s = e.series(x, 0, 5)
 print(s)
 pprint(s)
 1/y + x**2*y**(-3) + x**4*y**(-5) - x*y**(-2) - x**3*y**(-4) + O(x**5)
      2    4         3
 1   x    x    x    x
 ─ + ── + ── - ── - ── + O(x**5)
 y    3    5    2    4
     y    y    y    y

复数I

>>>from sympy import Symbol, exp, I
>>>x = Symbol("x")
>>>exp(I*x).expand()
exp(I*x)
>>>exp(I*x).expand(complex=True)
I*exp(-im(x))*sin(re(x)) + cos(re(x))*exp(-im(x))
>>>x = Symbol("x", real=True)
>>>exp(I*x).expand(complex=True)    #表达式展开(通过复数形式)
I*sin(x) + cos(x)

皮皮blog

函数

三角函数:

In [1]: sin(x+y).expand(trig=True)    #表达式展开(通过三角函数形式)
    cos(x)*sin(y) + cos(y)*sin(x)
In [2]: cos(x+y).expand(trig=True)
    cos(x)*cos(y) - sin(x)*sin(y)
In [3]: sin(I*x)
    I*sinh(x)
In [4]: sinh(I*x)
    I*sin(x)
In [5]: asinh(I)
π*I
───
 2
In [6]: asinh(I*x)
    I*asin(x)
In [15]: sin(x).series(x, 0, 10)
     3     5     7       9
    x     x     x       x
x - ── + ─── - ──── + ────── + O(x**10)
    6    120   5040   362880
In [16]: sinh(x).series(x, 0, 10)
     3     5     7       9
    x     x     x       x
x + ── + ─── + ──── + ────── + O(x**10)
    6    120   5040   362880
In [17]: asin(x).series(x, 0, 10)
     3      5      7       9
    x    3*x    5*x    35*x
x + ── + ──── + ──── + ───── + O(x**10)
    6     40    112     1152
In [18]: asinh(x).series(x, 0, 10)
     3      5      7       9
    x    3*x    5*x    35*x
x - ── + ──── - ──── + ───── + O(x**10)
    6     40    112     1152

球谐函数Ylm

from sympy.abc import theta, phi
In [2]: Ylm(1, 0, theta, phi)
     ————
╲╱ 3 *cos(θ)
────────────
        ——
  2*╲╱ π
In [3]: Ylm(1, 1, theta, phi)
    ——            I*φ
-╲╱ 6   *│sin(θ)│*ℯ
────────────────────
           ——
      4*╲╱ π
In [4]: Ylm(2, 1, theta, phi)
   ———                  I*φ
-╲╱ 30  *│sin(θ)│*cos(θ)*ℯ
────────────────────────────
                ——
          4*╲╱ π

阶乘和伽玛函数factorial

In [1]: x = Symbol("x")
In [2]: y = Symbol("y", integer=True)
In [3]: factorial(x)
    Γ(1 + x)
In [4]: factorial(y)
    y!
In [5]: factorial(x).series(x, 0, 3)
                    2           2    2  2
                   x *EulerGamma    π *x
1 - x*EulerGamma + ────────────── + ───── + O(x**3)
                         2            12

Zeta函数zeta

In [18]: zeta(4, x)
Out[18]: ζ(4, x)
In [19]: zeta(4, 1)
Out[19]:
 4
π
──
90
In [20]: zeta(4, 2)
Out[20]:
      4
     π
-1 + ──
     90
In [21]: zeta(4, 3)
Out[21]:
        4
  17   π
- ── + ──
  16   90

多项式

In [1]: chebyshevt(2, x)
Out[1]:
        2
-1 + 2*x
In [2]: chebyshevt(4, x)
Out[2]:
       2      4
1 - 8*x  + 8*x
In [3]: legendre(2, x)
Out[3]:
          2
       3*x
-1/2 + ────
        2
In [4]: legendre(8, x)
Out[4]:
           2         4         6         8
 35   315*x    3465*x    3003*x    6435*x
─── - ────── + ─────── - ─────── + ───────
128     32        64        32       128
In [5]: assoc_legendre(2, 1, x)
Out[5]:
             —————
          ╱     2
-3*x*╲╱  1 - x
In [6]: assoc_legendre(2, 2, x)
Out[6]:
       2
3 - 3*x
In [7]: hermite(3, x)
Out[7]:
           3
-12*x + 8*x

皮皮blog

微分方程

In [4]: f(x).diff(x, x) + f(x)     #注意在使用输入该命令之前,一定要声明f=Function('f')
Out[4]:
   2
  d
─────(f(x)) + f(x)
dx dx
In [5]: dsolve(f(x).diff(x, x) + f(x), f(x))
Out[5]: C₁*sin(x) + C₂*cos(x)

代数方程

In [7]: solve(x**4 - 1, x)
Out[7]: [i, 1, -1, -i]
In [8]: solve([x + 5*y - 2, -3*x + 6*y - 15], [x, y])
Out[8]: {y: 1, x: -3}

皮皮blog

线性代数

矩阵

矩阵由矩阵类创立:
>>>from sympy import Matrix
>>>Matrix([[1,0], [0,1]])
[1, 0]
[0, 1]
不只是数值矩阵,亦可为代数矩阵,即矩阵中存在符号:
>>>x = Symbol('x')
>>>y = Symbol('y')
>>>A = Matrix([[1,x], [y,1]])
>>>A
[1, x]
[y, 1]
>>>A**2
[1 + x*y,     2*x]
[    2*y, 1 + x*y]
 

系数匹配

使用 .match()方法,引用Wild类,来执行表达式的匹配。该方法会返回一个字典。

>>>from sympy import *
>>>x = Symbol('x')
>>>p = Wild('p')
>>>(5*x**2).match(p*x**2)
{p_: 5}
>>>q = Wild('q')
>>>(x**2).match(p*x**q)
{p_: 1, q_: 2}
如果匹配不成功,则返回None:
>>>print (x+1).match(p**x)
None
可以使用Wild类的‘exclude’参数(排除参数),排除不需要和无意义的匹配结果,来保证结论中的显示是唯一的:
>>>x = Symbol('x')
>>>p = Wild('p', exclude=[1,x])
>>>print (x+1).match(x+p) # 1 is excluded
None
>>>print (x+1).match(p+1) # x is excluded
None
>>>print (x+1).match(x+2+p) # -1 is not excluded
{p_: -1}

皮皮blog

打印输出

标准print

str(expression)返回如下:
>>>from sympy import Integral
>>>from sympy.abc import x
>>>print x**2
x**2
>>>print 1/x
1/x
>>>print Integral(x**2, x)
Integral(x**2, x)

Pretty Printing

用pprint函数可以输出不错的ascii艺术:
>>>from sympy import Integral, pprint
>>>from sympy.abc import x
>>>pprint(x**2) #doctest: +NORMALIZE_WHITESPACE
 2
x
>>>pprint(1/x)
1
-
x
>>>pprint(Integral(x**2, x))
  /
 |
 |  2
 | x  dx
 |
/

[Pretty PrintingWiki]

在python解释器中,为使pretty printing为默认输出,使用:

>>> from sympy import *

>>> import sys

>>> sys.displayhook = pprint

Python printing

>>>from sympy.printing.python import python
>>>from sympy import Integral
>>>from sympy.abc import x
>>>print python(x**2)
x = Symbol('x')
e = x**2
>>>print python(1/x)
x = Symbol('x')
e = 1/x
>>>print python(Integral(x**2, x))
x = Symbol('x')
e = Integral(x**2, x)

LaTeX printing

>>>from sympy import Integral, latex
>>>from sympy.abc import x
>>>latex(x**2)
$x^{2}$
>>>latex(1/x)
$\frac{1}{x}$
>>>latex(Integral(x**2, x))
$\int x^{2}\,dx$

MathML

>>>from sympy.printing.mathml import mathml
>>>from sympy import Integral, latex
>>>from sympy.abc import x
>>>print mathml(x**2)
<apply><power/><ci>x</ci><cn>2</cn></apply>
>>>print mathml(1/x)
<apply><power/><ci>x</ci><cn>-1</cn></apply>

Pyglet

>>>from sympy import Integral, preview
>>>from sympy.abc import x
>>>preview(Integral(x**2, x)) #doctest:+SKIP
用a pyglet window以LaTeX 提供表达式

如果安装了pyglet,一个呈现LaTeX表达式的窗口会被打开:

注解

Isympy默认调用pprint,所以这就是为什么看到pretty printing为默认的。

有一个打印的有效模块,sympy.printing。用这个模块实现其他的打印:

· pretty(expr), pretty_print(expr), pprint(expr): 分别返回或者输出,,表达式的漂亮描述。这是相同

·latex(expr), print_latex(expr):分别返回或者输出,LaTex描写的表达式

·mathml(expr), print_mathml(expr):分别返回或者输出,MathML描写的表达式

·print_gtk(expr): 表达式打印到Gtkmathview , 这是一个GTK小配件显示MathML代码。Gtkmathview程序是必须的。

from:Sympy符号计算库_皮皮blog-CSDN博客_sympy库

ref: 学习更多的SymPy参考SymPy User’s Guide 和 SymPy Modules Reference.

公共WIKI( wiki.sympy.org)有用的例子,教程和代码。

SymPy简易教程(一) - [Python]

SymPy简易教程(二) - [Python]

SymPy简易教程(三) - [Python]

SymPy简易教程(四) - [Python]

sympy

SymPy’s documentation!

教程symPy 0.7.2-git documentation

SymPy-符号运算好帮手

SymPy Tutorial(译)

 SymPy User’s GuideSymPy Modules Reference.

  • 9
    点赞
  • 28
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值