Sympy Docs PDF 1.3
Sympy Docs PDF 1.3
Release 1.3
1 Installation 1
1.1 Anaconda . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1
1.2 Git . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1
1.3 Other Methods . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 2
1.4 Run SymPy . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 2
1.5 Mpmath . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 2
1.6 Questions . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 2
2 SymPy Tutorial 5
2.1 Preliminaries . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 5
2.2 Introduction . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 7
2.3 Gotchas . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 10
2.4 Basic Operations . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 15
2.5 Printing . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 19
2.6 Simplification . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 25
2.7 Calculus . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 40
2.8 Solvers . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 46
2.9 Matrices . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 50
2.10 Advanced Expression Manipulation . . . . . . . . . . . . . . . . . . . . . . . . . . . . 60
i
5.6 Discrete Module . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 382
5.7 Numerical evaluation . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 393
5.8 Structural Details of Code Generation with SymPy . . . . . . . . . . . . . . . . . . 400
5.9 Numeric Computation . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 443
5.10 Functions Module . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 446
5.11 Geometry Module . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 580
5.12 Holonomic Functions . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 680
5.13 Symbolic Integrals . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 689
5.14 Numeric Integrals . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 719
5.15 Integration over Polytopes . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 725
5.16 Lie Algebra Module . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 727
5.17 Logic Module . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 742
5.18 Matrices . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 754
5.19 Polynomials Manipulation Module . . . . . . . . . . . . . . . . . . . . . . . . . . . . 810
5.20 Printing System . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1046
5.21 Plotting Module . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1087
5.22 Pyglet Plotting Module . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1101
5.23 Assumptions Module . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1105
5.24 Term rewriting . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1132
5.25 Series Module . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1135
5.26 Sets . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1162
5.27 Simplify . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1185
5.28 Details on the Hypergeometric Function Expansion Module . . . . . . . . . . . . . 1211
5.29 Stats . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1220
5.30 ODE . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1262
5.31 PDE . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1319
5.32 Solvers . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1328
5.33 Diophantine . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1347
5.34 Inequality Solvers . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1372
5.35 Solveset . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1375
5.36 Tensor Module . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1399
5.37 Utilities . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1430
5.38 Parsing input . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1496
5.39 Calculus . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1501
5.40 Physics Module . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1510
5.41 Category Theory Module . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1814
5.42 Differential Geometry Module . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1832
5.43 Vector Module . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1847
5.44 Contributions to docs . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1887
8 Wiki 1903
8.1 FAQ . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1903
ii
10 Planet SymPy 1907
13 About 1913
13.1 SymPy Development Team . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1913
13.2 Financial and Infrastructure Support . . . . . . . . . . . . . . . . . . . . . . . . . . 1926
13.3 License . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 1927
Bibliography 1933
Index 1961
iii
iv
CHAPTER
ONE
INSTALLATION
The SymPy CAS can be installed on virtually any computer with Python 2.7 or above. SymPy
does require mpmath Python library to be installed first. The recommended method of instal-
lation is through Anaconda, which includes mpmath, as well as several other useful libraries.
Alternatively, some Linux distributions have SymPy packages available.
SymPy officially supports Python 2.7, 3.4, 3.5, 3.6 and PyPy.
1.1 Anaconda
Anaconda is a free Python distribution from Continuum Analytics that includes SymPy, Mat-
plotlib, IPython, NumPy, and many more useful packages for scientific computing. This
is recommended because many nice features of SymPy are only enabled when certain li-
braries are installed. For example, without Matplotlib, only simple text-based plotting is
enabled. With the IPython notebook or qtconsole, you can get nicer LATEX printing by running
init_printing().
If you already have Anaconda and want to update SymPy to the latest version, use:
1.2 Git
If you wish to contribute to SymPy or like to get the latest updates as they come, install SymPy
from git. To download the repository, execute the following from the command line:
If you want to install SymPy, but still want to use the git version, you can run from your
repository:
This will cause the installed version to always point to the version in the git directory.
1
SymPy Documentation, Release 1.3
Various Linux distributions have SymPy available as a package. You may also install SymPy
from source or using pip.
After installation, it is best to verify that your freshly-installed SymPy works. To do this, start
up Python and import the SymPy libraries:
$ python
>>> from sympy import *
From here, execute some simple SymPy statements like the ones below:
>>> x = Symbol('x')
>>> limit(sin(x)/x, x, 0)
1
>>> integrate(1/x, x)
log(x)
For a starter guide on using SymPy effectively, refer to the SymPy Tutorial (page 5).
1.5 Mpmath
Versions of SymPy prior to 1.0 included mpmath, but it now depends on it as an external
dependency. If you installed SymPy with Anaconda, it will already include mpmath. Use:
import sys
import mpmath
sys.modules['sympy.mpmath'] = mpmath
before the code that imports sympy.mpmath. It is recommended to change code that uses
sympy.mpmath to use mpmath directly wherever possible.
1.6 Questions
If you have a question about installation or SymPy in general, feel free to visit our chat on
Gitter. In addition, our mailing list is an excellent source of community support.
2 Chapter 1. Installation
SymPy Documentation, Release 1.3
If you think there’s a bug or you would like to request a feature, please open an issue ticket.
1.6. Questions 3
SymPy Documentation, Release 1.3
4 Chapter 1. Installation
CHAPTER
TWO
SYMPY TUTORIAL
2.1 Preliminaries
This tutorial assumes that the reader already knows the basics of the Python programming
language. If you do not, the official Python tutorial is excellent.
This tutorial assumes a decent mathematical background. Most examples require knowledge
lower than a calculus level, and some require knowledge at a calculus level. Some of the
advanced features require more than this. If you come across a section that uses some math-
ematical function you are not familiar with, you can probably skip over it, or replace it with a
similar one that you are more familiar with. Or look up the function on Wikipedia and learn
something new. Some important mathematical concepts that are not common knowledge will
be introduced as necessary.
2.1.1 Installation
Quick Tip
You do not need to install SymPy to try it. You can use the online shell at http://live.sympy.
org, or the shell at the bottom right of this documentation page.
You will need to install SymPy first. See the installation guide (page 1).
Alternately, you can just use the SymPy Live Sphinx extension to run the code blocks in the
browser. For example, click on the green “Run code block in SymPy Live” button below
The SymPy Live shell in the bottom corner will pop up and evaluate the code block. You can
also click any individual line to evaluate it one at a time.
The SymPy Live shell is a fully interactive Python shell. You can type any expression in the
input box to evaluate it. Feel free to use it throughout the tutorial to experiment.
To show or hide the SymPy Live shell at any time, just click the green button on the bottom
right of the screen.
5
SymPy Documentation, Release 1.3
By default, the SymPy Live shell uses LATEX for output. If you want the output to look more like
the output in the documentation, change the output format to Str or Unicode in the settings.
If you wish to modify an example before evaluating it, change the evaluation mode to “copy”
in the SymPy Live settings. This will cause clicking on an example to copy the example to
the SymPy Live shell, but not evaluate it, allowing you to change it before execution. You can
also use the up/down arrow keys on your keyboard in the input box to move through the shell
history.
The SymPy Live shell is also available at http://live.sympy.org, with extra features, like a
mobile phone enhanced version and saved history.
2.1.2 Exercises
This tutorial was the basis for a tutorial given at the 2013 SciPy conference in Austin, TX.
The website for that tutorial is here. It has links to videos, materials, and IPython notebook
exercises. The IPython notebook exercises in particular are recommended to anyone going
through this tutorial.
This tutorial aims to give an introduction to SymPy for someone who has not used the library
before. Many features of SymPy will be introduced in this tutorial, but they will not be ex-
haustive. In fact, virtually every functionality shown in this tutorial will have more options
or capabilities than what will be shown. The rest of the SymPy documentation serves as API
documentation, which extensively lists every feature and option of each function.
These are the goals of this tutorial:
• To give a guide, suitable for someone who has never used SymPy (but who has used
Python and knows the necessary mathematics).
• To be written in a narrative format, which is both easy and fun to follow. It should read
like a book.
• To give insightful examples and exercises, to help the reader learn and to make it enter-
taining to work through.
• To introduce concepts in a logical order.
• To use good practices and idioms, and avoid antipatterns. Functions or methodologies
that tend to lead to antipatterns are avoided. Features that are only useful to advanced
users are not shown.
• To be consistent. If there are multiple ways to do it, only the best way is shown.
• To avoid unnecessary duplication, it is assumed that previous sections of the tutorial
have already been read.
Feedback on this tutorial, or on SymPy in general is always welcome. Just write to our mailing
list.
2.2 Introduction
Symbolic computation deals with the computation of mathematical objects symbolically. This
means that the mathematical objects are represented exactly, not approximately, and math-
ematical expressions with unevaluated variables are left in symbolic form.
Let’s take an example. Say we wanted to use the built-in Python functions to compute square
roots. We might do something like this
9 is a perfect square, so we got the exact answer, 3. But suppose we computed the square
root of a number that isn’t a perfect square
>>> math.sqrt(8)
2.82842712475
Here we got an approximate result. 2.82842712475 is not the exact square root of 8 (indeed,
the actual square root of 8 cannot be represented by a finite decimal, since it is an irrational
number). If all we cared about was the decimal form of the square root of 8, we would be
done.
√ √ √
But suppose we want to go further. Recall that 8 = 4 · 2 = 2 2. We would have a hard
time deducing this from the above result. This is where symbolic computation comes in. With
a symbolic computation system like SymPy, square roots of numbers that are not perfect
squares are left unevaluated by default
Furthermore—and this is where we start to see the real power of symbolic computation—
symbolic results can be symbolically simplified.
>>> sympy.sqrt(8)
2*sqrt(2)
The above example starts to show how we can manipulate irrational numbers exactly using
SymPy. But it is much more powerful than that. Symbolic computation systems (which by
the way, are also often called computer algebra systems, or just CASs) such as SymPy are
capable of computing symbolic expressions with variables.
As we will see later, in SymPy, variables are defined using symbols. Unlike many symbolic
manipulation systems, variables in SymPy must be defined before they are used (the reason
for this will be discussed in the next section (page 10)).
Let us define a symbolic expression, representing the mathematical expression x + 2y.
2.2. Introduction 7
SymPy Documentation, Release 1.3
Note that we wrote x + 2*y just as we would if x and y were ordinary Python variables. But
in this case, instead of evaluating to something, the expression remains as just x + 2*y. Now
let us play around with it:
>>> expr + 1
x + 2*y + 1
>>> expr - x
2*y
Notice something in the above example. When we typed expr - x, we did not get x + 2*y -
x, but rather just 2*y. The x and the -x automatically canceled one another. This is similar to
how sqrt(8) automatically turned into 2*sqrt(2) above. This isn’t always the case in SymPy,
however:
>>> x*expr
x*(x + 2*y)
Here, we might have expected x(x + 2y) to transform into x2 + 2xy, but instead we see that the
expression was left alone.
√ This√ is a common theme in SymPy. Aside from obvious simplifica-
tions like x − x = 0 and 8 = 2 2, most simplifications are not performed automatically. This
is because we might prefer the factored form x(x + 2y), or we might prefer the expanded form
x2 + 2xy. Both forms are useful in different circumstances. In SymPy, there are functions to
go from one form to the other
The real power of a symbolic computation system such as SymPy is the ability to do all sorts
of computations symbolically. SymPy can simplify expressions, compute derivatives, inte-
grals, and limits, solve equations, work with matrices, and much, much more, and do it all
symbolically. It includes modules for plotting, printing (like 2D pretty printed output of math
formulas, or LATEX), code generation, physics, statistics, combinatorics, number theory, ge-
ometry, logic, and more. Here is a small sampling of the sort of symbolic power SymPy is
capable of, to whet your appetite.
This will make all further examples pretty print with unicode characters.
>>> init_printing(use_unicode=True)
>>> diff(sin(x)*exp(x), x)
x x
e ⋅sin(x) + e ⋅cos(x)
∫
Compute (ex sin (x) + ex cos (x)) dx.
sin (x)
Find limx→0 x .
>>> limit(sin(x)/x, x, 0)
1
Solve x2 − 2 = 0.
>>> solve(x**2 - 2, x)
[-√2, √2]
>>> y = Function('y')
>>> dsolve(Eq(y(t).diff(t, t) - y(t), exp(t)), y(t))
-t ⎛ t⎞ t
y(t) = C2 ⋅e + ⎜C1 + ─⎟⋅e
⎝ 2⎠
Rewrite the Bessel function Jν (z) in terms of the spherical Bessel function jν (z).
2.2. Introduction 9
SymPy Documentation, Release 1.3
There are many computer algebra systems out there. This Wikipedia article lists many of
them. What makes SymPy a better choice than the alternatives?
First off, SymPy is completely free. It is open source, and licensed under the liberal BSD
license, so you can modify the source code and even sell it if you want to. This contrasts
with popular commercial systems like Maple or Mathematica that cost hundreds of dollars in
licenses.
Second, SymPy uses Python. Most computer algebra systems invent their own language.
Not SymPy. SymPy is written entirely in Python, and is executed entirely in Python. This
means that if you already know Python, it is much easier to get started with SymPy, because
you already know the syntax (and if you don’t know Python, it is really easy to learn). We
already know that Python is a well-designed, battle-tested language. The SymPy developers
are confident in their abilities in writing mathematical software, but programming language
design is a completely different thing. By reusing an existing language, we are able to focus
on those things that matter: the mathematics.
Another computer algebra system, Sage also uses Python as its language. But Sage is large,
with a download of over a gigabyte. An advantage of SymPy is that it is lightweight. In
addition to being relatively small, it has no dependencies other than Python, so it can be used
almost anywhere easily. Furthermore, the goals of Sage and the goals of SymPy are different.
Sage aims to be a full featured system for mathematics, and aims to do so by compiling all the
major open source mathematical systems together into one. When you call some function in
Sage, such as integrate, it calls out to one of the open source packages that it includes. In
fact, SymPy is included in Sage. SymPy on the other hand aims to be an independent system,
with all the features implemented in SymPy itself.
A final important feature of SymPy is that it can be used as a library. Many computer algebra
systems focus on being usable in interactive environments, but if you wish to automate or
extend them, it is difficult to do. With SymPy, you can just as easily use it in an interactive
Python environment or import it in your own Python application. SymPy also provides APIs
to make it easy to extend it with your own custom functions.
2.3 Gotchas
To begin, we should make something about SymPy clear. SymPy is nothing more than a Python
library, like NumPy, Django, or even modules in the Python standard library sys or re. What
this means is that SymPy does not add anything to the Python language. Limitations that are
inherent in the Python language are also inherent in SymPy. It also means that SymPy tries
to use Python idioms whenever possible, making programming with SymPy easy for those
already familiar with programming with Python. As a simple example, SymPy uses Python
syntax to build expressions. Implicit multiplication (like 3x or 3 x) is not allowed in Python,
and thus not allowed in SymPy. To multiply 3 and x, you must type 3*x with the *.
2.3.1 Symbols
One consequence of this fact is that SymPy can be used in any environment where Python is
available. We just import it, like we would any other library:
This imports all the functions and classes from SymPy into our interactive Python session.
Now, suppose we start to do a computation.
>>> x + 1
Traceback (most recent call last):
...
NameError: name 'x' is not defined
Oops! What happened here? We tried to use the variable x, but it tells us that x is not defined.
In Python, variables have no meaning until they are defined. SymPy is no different. Unlike
many symbolic manipulation systems you may have used, in SymPy, variables are not defined
automatically. To define variables, we must use symbols.
>>> x = symbols('x')
>>> x + 1
x + 1
symbols takes a string of variable names separated by spaces or commas, and creates Sym-
bols out of them. We can then assign these to variable names. Later, we will investigate some
convenient ways we can work around this issue. For now, let us just define the most common
variable names, x, y, and z, for use through the rest of this section
As a final note, we note that the name of a Symbol and the name of the variable it is assigned
to need not have anything to do with one another.
Here we have done the very confusing thing of assigning a Symbol with the name a to the
variable b, and a Symbol of the name b to the variable a. Now the Python variable named a
points to the SymPy Symbol named b, and visa versa. How confusing. We could have also
done something like
This also shows that Symbols can have names longer than one character if we want.
Usually, the best practice is to assign Symbols to Python variables of the same name, although
there are exceptions: Symbol names can contain characters that are not allowed in Python
variable names, or may just want to avoid typing long names by assigning Symbols with long
names to single letter Python variables.
To avoid confusion, throughout this tutorial, Symbol names and Python variable names will
always coincide. Furthermore, the word “Symbol” will refer to a SymPy Symbol and the word
“variable” will refer to a Python variable.
Finally, let us be sure we understand the difference between SymPy Symbols and Python
variables. Consider the following:
2.3. Gotchas 11
SymPy Documentation, Release 1.3
x = symbols('x')
expr = x + 1
x = 2
print(expr)
What do you think the output of this code will be? If you thought 3, you’re wrong. Let’s see
what really happens
>>> x = symbols('x')
>>> expr = x + 1
>>> x = 2
>>> print(expr)
x + 1
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.
When we created expr, the Python variable x was a Symbol. After we created, it, we changed
the Python variable x to 2. But expr remains the same. This behavior is not unique to SymPy.
All Python programs work this way: if a variable is changed, expressions that were already
created with that variable do not change automatically. For example
>>> x = 'abc'
>>> expr = x + 'def'
>>> expr
'abcdef'
>>> x = 'ABC'
>>> expr
'abcdef'
Quick Tip
In this example, if we want to know what expr is with the new value of x, we need to reevaluate
the code that created expr, namely, expr = x + 1. This can be complicated if several lines
created expr. One advantage of using a symbolic computation system like SymPy is that we
can build a symbolic representation for expr, and then substitute x with values. The correct
way to do this in SymPy is to use subs, which will be discussed in more detail later.
>>> x = symbols('x')
>>> expr = x + 1
>>> expr.subs(x, 2)
3
Another very important consequence of the fact that SymPy does not extend Python syntax is
that = does not represent equality in SymPy. Rather it is Python variable assignment. This is
hard-coded into the Python language, and SymPy makes no attempts to change that.
You may think, however, that ==, which is used for equality testing in Python, is used for
SymPy as equality. This is not quite correct either. Let us see what happens when we use ==.
>>> x + 1 == 4
False
>>> Eq(x + 1, 4)
Eq(x + 1, 4)
There is one additional caveat about == as well. Suppose we want to know if (x+1)2 = x2 +2x+1.
We might try something like this:
We got False again. However, (x + 1)2 does equal x2 + 2x + 1. What is going on here? Did we
find a bug in SymPy, or is it just not powerful enough to recognize this basic algebraic fact?
Recall from above that == represents exact structural equality testing. “Exact” here means
that two expressions will compare equal with == only if they are exactly equal structurally.
Here, (x + 1)2 and x2 + 2x + 1 are not the same symbolically. One is the power of an addition of
two terms, and the other is the addition of three terms.
It turns out that when using SymPy as a library, having == test for exact structural equality is
far more useful than having it represent symbolic equality, or having it test for mathematical
equality. However, as a new user, you will probably care more about the latter two. We have
already seen an alternative to representing equalities symbolically, Eq. To test if two things
are equal, it is best to recall the basic fact that if a = b, then a − b = 0. Thus, the best way to
check if a = b is to take a − b and simplify it, and see if it goes to 0. We will learn later (page 25)
that the function to do this is called simplify. This method is not infallible—in fact, it can
be theoretically proven that it is impossible to determine if two symbolic expressions are
identically equal in general—but for most common expressions, it works quite well.
>>> a = (x + 1)**2
>>> b = x**2 + 2*x + 1
>>> simplify(a - b)
0
>>> c = x**2 - 2*x + 1
>>> simplify(a - c)
4*x
There is also a method called equals that tests if two expressions are equal by evaluating
them numerically at random points.
2.3. Gotchas 13
SymPy Documentation, Release 1.3
You may have noticed that we have been using ** for exponentiation instead of the standard ^.
That’s because SymPy follows Python’s conventions. In Python, ^ represents logical exclusive
or. SymPy follows this convention:
Finally, a small technical discussion on how SymPy works is in order. When you type some-
thing like x + 1, the SymPy Symbol x is added to the Python int 1. Python’s operator rules
then allow SymPy to tell Python that SymPy objects know how to be added to Python ints, and
so 1 is automatically converted to the SymPy Integer object.
This sort of operator magic happens automatically behind the scenes, and you rarely need to
even know that it is happening. However, there is one exception. Whenever you combine a
SymPy object and a SymPy object, or a SymPy object and a Python object, you get a SymPy
object, but whenever you combine two Python objects, SymPy never comes into play, and so
you get a Python object.
>>> type(Integer(1) + 1)
<class 'sympy.core.numbers.Integer'>
>>> type(1 + 1)
<... 'int'>
Note: On running the example above in SymPy Live, (1+1) is wrapped by Integer, so it does
not show the correct output.
This is usually not a big deal. Python ints work much the same as SymPy Integers, but there
is one important exception: division. In SymPy, the division of two Integers gives a Rational:
>>> Integer(1)/Integer(3)
1/3
>>> type(Integer(1)/Integer(3))
<class 'sympy.core.numbers.Rational'>
But in Python / represents either integer division or floating point division, depending on
whether you are in Python 2 or Python 3, and depending on whether or not you have run from
__future__ import division:
Note: On running the example above in SymPy Live, (1/2) is wrapped by Integer, so it does
not show the correct output.
>>> Rational(1, 2)
1/2
This problem also comes up whenever we have a larger symbolic expression with int/int in
it. For example:
Note: On running the example above in SymPy Live, (1/2) is wrapped by Integer, so it does
not show the correct output.
This happens because Python first evaluates 1/2 into 0.5, and then that is cast into a SymPy
type when it is added to x. Again, we can get around this by explicitly creating a Rational:
>>> x + Rational(1, 2)
x + 1/2
There are several tips on avoiding this situation in the Gotchas and Pitfalls (page 69) docu-
ment.
For more discussion on the topics covered in this section, see Gotchas and Pitfalls (page 69).
Here we discuss some of the most basic operations needed for expression manipulation in
SymPy. Some more advanced operations will be discussed later in the advanced expression
manipulation (page 60) section.
2.4.1 Substitution
One of the most common things you might want to do with a mathematical expression is
substitution. Substitution replaces all instances of something in an expression with something
else. It is done using the subs method. For example
>>> expr.subs(x, 0)
2
2. Replacing a subexpression with another subexpression. There are two reasons we might
want to do this. The first is if we are trying to build an expression that has some symme-
xx
try, such as xx . To build this, we might start with x**y, and replace y with x**y. We
would then get x**(x**y). If we replaced y in this new expression with x**x, we would
get x**(x**(x**x)), the desired expression.
There are two important things to note about subs. First, it returns a new expression. SymPy
objects are immutable. That means that subs does not modify it in-place. For example
Quick Tip
Here, we see that performing expr.subs(x, 0) leaves expr unchanged. In fact, since SymPy
expressions are immutable, no function will change them in-place. All functions will return
new expressions.
To perform multiple substitutions at once, pass a list of (old, new) pairs to subs.
It is often useful to combine this with a list comprehension to do a large set of similar replace-
ments all at once. For example, say we had x4 − 4x3 + 4x2 − 2x + 3 and we wanted to replace
all instances of x that have an even power with y, to get y 4 − 4x3 + 4y 2 − 2x + 3.
>>> expr = x**4 - 4*x**3 + 4*x**2 - 2*x + 3
>>> replacements = [(x**i, y**i) for i in range(5) if i % 2 == 0]
>>> expr.subs(replacements)
-4*x**3 - 2*x + y**4 + 4*y**2 + 3
The sympify function (that’s sympify, not to be confused with simplify) can be used to
convert strings into SymPy expressions.
For example
>>> str_expr = ”x**2 + 3*x - 1/2”
>>> expr = sympify(str_expr)
>>> expr
x**2 + 3*x - 1/2
>>> expr.subs(x, 2)
19/2
2.4.3 evalf
SymPy can evaluate floating point expressions to arbitrary precision. By default, 15 digits of
precision are used, but you can pass any number as the argument to evalf. Let’s compute
the first 100 digits of π.
>>> pi.evalf(100)
3.
,→141592653589793238462643383279502884197169399375105820974944592307816406286208998628034825342117068
To numerically evaluate an expression with a Symbol at a point, we might use subs followed
by evalf, but it is more efficient and numerically stable to pass the substitution to evalf
using the subs flag, which takes a dictionary of Symbol: point pairs.
>>> expr = cos(2*x)
>>> expr.evalf(subs={x: 2.4})
0.0874989834394464
Sometimes there are roundoff errors smaller than the desired precision that remain after an
expression is evaluated. Such numbers can be removed at the user’s discretion by setting the
chop flag to True.
2.4.4 lambdify
subs and evalf are good if you want to do simple evaluation, but if you intend to evaluate
an expression at many points, there are more efficient ways. For example, if you wanted to
evaluate an expression at a thousand points, using SymPy would be far slower than it needs
to be, especially if you only care about machine precision. Instead, you should use libraries
like NumPy and SciPy.
The easiest way to convert a SymPy expression to an expression that can be numerically
evaluated is to use the lambdify function. lambdify acts like a lambda function, except it
converts the SymPy names to the names of the given numerical library, usually NumPy. For
example
You can use other libraries than NumPy. For example, to use the standard library math mod-
ule, use ”math”.
To use lambdify with numerical libraries that it does not know about, pass a dictionary of
sympy_name:numerical_function pairs. For example
2.5 Printing
As we have already seen, SymPy can pretty print its output using Unicode characters. This
is a short introduction to the most common printing options available in SymPy.
2.5.1 Printers
There are several printers available in SymPy. The most common ones are
• str
• srepr
• ASCII pretty printer
• Unicode pretty printer
• LaTeX
• MathML
• Dot
In addition to these, there are also “printers” that can output SymPy objects to code, such as
C, Fortran, Javascript, Theano, and Python. These are not discussed in this tutorial.
If all you want is the best pretty printing, use the init_printing() function. This will auto-
matically enable the best printer available in your environment.
Quick Tip
You can also change the printer used in SymPy Live. Just change the “Output Format” in
the settings.
If you plan to work in an interactive calculator-type session, the init_session() function will
automatically import everything in SymPy, create some common Symbols, setup plotting, and
run init_printing().
Python console for SymPy 0.7.3 (Python 2.7.5-64-bit) (ground types: gmpy)
2.5. Printing 19
SymPy Documentation, Release 1.3
>>>
If LATEX is not installed, but Matplotlib is installed, it will use the Matplotlib rendering
engine. If Matplotlib is not installed, it uses the Unicode pretty printer.
• In the IPython notebook, it will use MathJax to render LATEX.
• In an IPython console session, or a regular Python session, it will use the Unicode pretty
printer if the terminal supports Unicode.
• In a terminal that does not support Unicode, the ASCII pretty printer is used.
2.5. Printing 21
SymPy Documentation, Release 1.3
In addition to automatic printing, you can explicitly use any one of the printers by calling the
appropriate function.
str
To get a string form of an expression, use str(expr). This is also the form that is produced
by print(expr). String forms are designed to be easy to read, but in a form that is correct
Python syntax so that it can be copied and pasted. The str() form of an expression will
usually look exactly the same as the expression as you would enter it.
srepr
The srepr form of an expression is designed to show the exact form of an expression. It will
be discussed more in the Advanced Expression Manipulation (page 60) section. To get it, use
srepr()1 .
The srepr form is mostly useful for understanding how an expression is built internally.
The ASCII pretty printer is accessed from pprint(). If the terminal does not support Unicode,
the ASCII printer is used by default. Otherwise, you must pass use_unicode=False.
pprint() prints the output to the screen. If you want the string form, use pretty().
The Unicode pretty printer is also accessed from pprint() and pretty(). If the terminal
supports Unicode, it is used automatically. If pprint() is not able to detect that the terminal
supports unicode, you can pass use_unicode=True to force it to use Unicode.
repr() on the elements of the list, and some SymPy functions return lists (such as solve()). Since srepr() is so
verbose, it is unlikely that anyone would want it called by default on the output of solve().
2.5. Printing 23
SymPy Documentation, Release 1.3
LATEX
The latex() function has many options to change the formatting of different things. See its
documentation (page 1072) for more details.
MathML
There is also a printer to MathML, called print_mathml(). It must be imported from sympy.
printing.mathml.
>>> from sympy.printing.mathml import print_mathml
>>> print_mathml(Integral(sqrt(1/x), x))
<apply>
<int/>
<bvar>
<ci>x</ci>
</bvar>
<apply>
<root/>
<apply>
<power/>
<ci>x</ci>
<cn>-1</cn>
</apply>
</apply>
</apply>
print_mathml() prints the output. If you want the string, use the function mathml().
Dot
The dotprint() function in sympy.printing.dot prints output to dot format, which can be
rendered with Graphviz. See the Advanced Expression Manipulation (page 60) section for
some examples of the output of this printer.
>>> from sympy.printing.dot import dotprint
>>> from sympy.abc import x
>>> print(dotprint(x+2))
digraph{
# Graph style
”ordering”=”out”
(continues on next page)
2.6 Simplification
To make this document easier to read, we are going to enable pretty printing.
2.6.1 simplify
Now let’s jump in and do some interesting mathematics. One of the most useful features of
a symbolic manipulation system is the ability to simplify mathematical expressions. SymPy
has dozens of functions to perform various kinds of simplification. There is also one general
function called simplify() that attempts to apply all of these functions in an intelligent way
to arrive at the simplest form of an expression. Here are some examples
Here, gamma(x) is Γ(x), the gamma function. We see that simplify() is capable of handling
a large class of expressions.
But simplify() has a pitfall. It just applies all the major simplification operations in SymPy,
and uses heuristics to determine the simplest result. But “simplest” is not a well-defined
term. For example, say we wanted to “simplify” x2 + 2x + 1 into (x + 1)2 :
We did not get what we want. There is a function to perform this simplification, called fac-
tor(), which will be discussed below.
2.6. Simplification 25
SymPy Documentation, Release 1.3
Another pitfall to simplify() is that it can be unnecessarily slow, since it tries many kinds
of simplifications before picking the best one. If you already know exactly what kind of sim-
plification you are after, it is better to apply the specific simplification function(s) that apply
those simplifications.
Applying specific simplification functions instead of simplify() also has the advantage that
specific functions have certain guarantees about the form of their output. These will be dis-
cussed with each function below. For example, factor(), when called on a polynomial with
rational coefficients, is guaranteed to factor the polynomial into irreducible factors. sim-
plify() has no guarantees. It is entirely heuristical, and, as we saw above, it may even miss
a possible type of simplification that SymPy is capable of doing.
simplify() is best when used interactively, when you just want to whittle down an expres-
sion to a simpler form. You may then choose to apply specific functions once you see what
simplify() returns, to get a more precise result. It is also useful when you have no idea what
form an expression will take, and you need a catchall function to simplify it.
expand
expand() is one of the most common simplification functions in SymPy. Although it has a lot
of scopes, for now, we will consider its function in expanding polynomial expressions. For
example:
Given a polynomial, expand() will put it into a canonical form of a sum of monomials.
expand() may not sound like a simplification function. After all, by its very name, it makes
expressions bigger, not smaller. Usually this is the case, but often an expression will become
smaller upon calling expand() on it due to cancellation.
factor
factor() takes a polynomial and factors it into irreducible factors over the rational numbers.
For example:
For polynomials, factor() is the opposite of expand(). factor() uses a complete multivari-
ate factorization algorithm over the rational numbers, which means that each of the factors
returned by factor() is guaranteed to be irreducible.
If you are interested in the factors themselves, factor_list returns a more structured output.
Note that the input to factor and expand need not be polynomials in the strict sense. They
will intelligently factor or expand any kind of expression (though note that the factors may
not be irreducible if the input is no longer a polynomial over the rationals).
collect
>>> collected_expr.coeff(x, 2)
-z + 2
cancel
cancel() will take any rational function and put it into the standard canonical form, pq , where
p and q are expanded polynomials with no common factors, and the leading coefficients of p
and q do not have denominators (i.e., are integers).
2.6. Simplification 27
SymPy Documentation, Release 1.3
Note that since factor() will completely factorize both the numerator and the denominator
of an expression, it can also be used to do the same thing:
>>> factor(expr)
2
(y - z)
────────
x - 1
However, if you are only interested in making sure that the expression is in canceled form,
cancel() is more efficient than factor().
apart
Note: SymPy follows Python’s naming conventions for inverse trigonometric functions,
which is to append an a to the front of the function’s name. For example, the inverse co-
sine, or arc cosine, is called acos().
>>> acos(x)
acos(x)
>>> cos(acos(x))
x
>>> asin(1)
π
─
2
trigsimp
Much like simplify(), trigsimp() applies various trigonometric identities to the input ex-
pression, and then uses a heuristic to return the “best” one.
expand_trig
To expand trigonometric functions, that is, apply the sum or double angle identities, use
expand_trig().
2.6. Simplification 29
SymPy Documentation, Release 1.3
2.6.4 Powers
To summarize
This is important to remember, because by default, SymPy will not perform simplifications if
they are not true in general.
In order to make SymPy perform simplifications involving identities that are only true under
certain assumptions, we need to put assumptions on our Symbols. We will undertake a full
discussion of the assumptions system later, but for now, all we need to know are the following.
• By default, SymPy Symbols are assumed to be complex (elements of C). That is, a sim-
plification will not be applied to an expression with a given Symbol unless it holds for all
complex numbers.
• Symbols can be given different assumptions by passing the assumption to symbols().
For the rest of this section, we will be assuming that x and y are positive, and that a and
b are real. We will leave z, t, and c as arbitrary complex Symbols to demonstrate what
happens in that case.
Note: In SymPy, sqrt(x) is just a shortcut to x**Rational(1, 2). They are exactly the
same object.
powsimp
>>> powsimp(x**a*x**b)
a + b
x
>>> powsimp(x**a*y**a)
a
(x⋅y)
>>> powsimp(t**c*z**c)
c c
t ⋅z
If you know that you want to apply this simplification, but you don’t want to mess with as-
sumptions, you can pass the force=True flag. This will force the simplification to take place,
regardless of assumptions.
Note that in some instances, in particular, when the exponents are integers or rational num-
bers, and identity 2 holds, it will be applied automatically.
>>> (z*t)**2
2 2
t ⋅z
>>> sqrt(x*y)
√x⋅√y
This means that it will be impossible to undo this identity with powsimp(), because even if
powsimp() were to put the bases together, they would be automatically split apart again.
>>> powsimp(z**2*t**2)
2 2
t ⋅z
(continues on next page)
2.6. Simplification 31
SymPy Documentation, Release 1.3
expand_power_exp / expand_power_base
>>> expand_power_base((x*y)**a)
a a
x ⋅y
>>> expand_power_base((z*t)**c)
c
(t⋅z)
And as with powsimp(), you can force the expansion to happen without fiddling with assump-
tions by using force=True.
As with identity 2, identity 1 is applied automatically if the power is a number, and hence
cannot be undone with expand_power_exp().
>>> x**2*x**3
5
x
>>> expand_power_exp(x**5)
5
x
powdenest
>>> powdenest((x**a)**b)
a⋅b
x
As before, the identity is not applied if it is not true under the given assumptions.
>>> powdenest((z**a)**b)
b
(continues on next page)
Note: In SymPy, as in Python and most programming languages, log is the natural logarithm,
also known as ln. SymPy automatically provides an alias ln = log in case you forget this.
>>> ln(x)
log(x)
Logarithms have similar issues as powers. There are two main identities
1. log (xy) = log (x) + log (y)
2. log (xn ) = n log (x)
Neither identity is true for arbitrary complex x and y, due to the branch cut in the complex
plane for the complex logarithm. However, sufficient conditions for the identities to hold are
if x and y are positive and n is real.
expand_log
To apply identities 1 and 2 from left to right, use expand_log(). As always, the identities will
not be applied unless they are valid.
>>> expand_log(log(x*y))
log(x) + log(y)
>>> expand_log(log(x/y))
log(x) - log(y)
>>> expand_log(log(x**2))
(continues on next page)
2.6. Simplification 33
SymPy Documentation, Release 1.3
As with powsimp() and powdenest(), expand_log() has a force option that can be used to
ignore assumptions.
>>> expand_log(log(z**2))
⎛ 2⎞
log⎝z ⎠
>>> expand_log(log(z**2), force=True)
2⋅log(z)
logcombine
logcombine() also has a force option that can be used to ignore assumptions.
>>> factorial(n)
n!
( )
The binomial coefficient function is binomial. binomial(n, k) represents nk , the number
of ways to choose k items from a set of n distinct items. It is also often written as nCk, and is
pronounced “n choose k”.
>>> binomial(n, k)
⎛n⎞
⎜ ⎟
⎝k⎠
The factorial
∫∞ function is closely related to the gamma function, gamma. gamma(z) represents
Γ(z) = 0 tz−1 e−t dt, which for positive integer z is the same as (z − 1)!.
>>> gamma(z)
Γ(z)
rewrite
A common way to deal with special functions is to rewrite them in terms of one another. This
works for any function in SymPy, not just special functions. To rewrite an expression in terms
of a function, use expr.rewrite(function). For example,
>>> tan(x).rewrite(sin)
2
2⋅sin (x)
─────────
sin(2⋅x)
>>> factorial(x).rewrite(gamma)
Γ(x + 1)
For some tips on applying more targeted rewriting, see the Advanced Expression Manipula-
tion (page 60) section.
expand_func
To expand special functions in terms of some identities, use expand_func(). For example
2.6. Simplification 35
SymPy Documentation, Release 1.3
hyperexpand
hyperexpand() also works on the more general Meijer G-function (see its documentation
(page 555) for more information).
combsimp
gammasimp
Let’s use SymPy to explore continued fractions. A continued fraction is an expression of the
form
1
a0 +
1
a1 +
1
a2 +
.. 1
.+
an
where a0 , . . . , an are integers, and a1 , . . . , an are positive. A continued fraction can also be infi-
nite, but infinite objects are more difficult to represent in computers, so we will only examine
the finite case here.
A continued fraction of the above form is often represented as a list [a0 ; a1 , . . . , an ]. Let’s write
a simple function that converts such a list to its continued fraction form. The easiest way
to construct a continued fraction from a list is to work backwards. Note that despite the
apparent symmetry of the definition, the first element, a0 , must usually be handled differently
from the rest.
We use Integer(0) in list_to_frac so that the result will always be a SymPy object, even if
we only pass in Python ints.
Every finite continued fraction is a rational number, but we are interested in symbolics here,
so let’s create a symbolic continued fraction. The symbols() function that we have been using
has a shortcut to create numbered symbols. symbols('a0:5') will create the symbols a0, a1,
…, a4.
2.6. Simplification 37
SymPy Documentation, Release 1.3
This form is useful for understanding continued fractions, but lets put it into standard rational
function form using cancel().
>>> frac = cancel(frac)
>>> frac
a0 ⋅a1 ⋅a2 ⋅a3 ⋅a4 + a0 ⋅a1 ⋅a2 + a0 ⋅a1 ⋅a4 + a0 ⋅a3 ⋅a4 + a0 + a2 ⋅a3 ⋅a4 + a2 + a4
─────────────────────────────────────────────────────────────────────────
a1 ⋅a2 ⋅a3 ⋅a4 + a1 ⋅a2 + a1 ⋅a4 + a3 ⋅a4 + 1
Now suppose we were given frac in the above canceled form. In fact, we might be given the
fraction in any form, but we can always put it into the above canonical form with cancel().
Suppose that we knew that it could be rewritten as a continued fraction. How could we do
this with SymPy? A continued fraction is recursively c + f1 , where c is an integer and f is
a (smaller) continued fraction. If we could write the expression in this form, we could pull
out each c recursively and add it to a list. We could then get a continued fraction with our
list_to_frac() function.
The key observation here is that we can convert an expression to the form c + f1 by doing a
partial fraction decomposition with respect to c. This is because f does not contain c. This
means we need to use the apart() function. We use apart() to pull the term out, then
subtract it from the expression, and take the reciprocal to get the f part.
>>> l = []
>>> frac = apart(frac, a0)
>>> frac
a2 ⋅a3 ⋅a4 + a2 + a4
a0 + ───────────────────────────────────────
a1 ⋅a2 ⋅a3 ⋅a4 + a1 ⋅a2 + a1 ⋅a4 + a3 ⋅a4 + 1
>>> l.append(a0)
>>> frac = 1/(frac - a0)
>>> frac
a1 ⋅a2 ⋅a3 ⋅a4 + a1 ⋅a2 + a1 ⋅a4 + a3 ⋅a4 + 1
───────────────────────────────────────
a2 ⋅a3 ⋅a4 + a2 + a4
Quick Tip
You can execute multiple lines at once in SymPy Live. Typing Shift-Enter instead of Enter
will enter a newline instead of executing.
Of course, this exercise seems pointless, because we already know that our frac is
list_to_frac([a0, a1, a2, a3, a4]). So try the following exercise. Take a list of sym-
bols and randomize them, and create the canceled continued fraction, and see if you can
reproduce the original list. For example
Click on “Run code block in SymPy Live” on the definition of list_to_frac() above, and
then on the above example, and try to reproduce l from frac. I have deleted l at the
end to remove the temptation for peeking (you can check your answer at the end by call-
ing cancel(list_to_frac(l)) on the list that you generate at the end, and comparing it to
orig_frac.
See if you can think of a way to figure out what symbol to pass to apart() at each stage (hint:
1
think of what happens to a0 in the formula a0 + a1 +··· when it is canceled).
2.6. Simplification 39
SymPy Documentation, Release 1.3
2.7 Calculus
This section covers how to do basic calculus tasks such as derivatives, integrals, limits, and
series expansions in SymPy. If you are not familiar with the math of any part of this section,
you may safely skip it.
2.7.1 Derivatives
>>> diff(cos(x), x)
-sin(x)
>>> diff(exp(x**2), x)
⎛ 2⎞
⎝x ⎠
2⋅x⋅e
diff can take multiple derivatives at once. To take multiple derivatives, pass the variable as
many times as you wish to differentiate, or pass a number after the variable. For example,
both of the following find the third derivative of x4 .
>>> diff(x**4, x, x, x)
24⋅x
>>> diff(x**4, x, 3)
24⋅x
You can also take derivatives with respect to many variables at once. Just pass each derivative
in order, using the same syntax as for single variable derivatives. For example, each of the
7
following will compute ∂x∂y∂ 2 ∂z4 exyz .
diff can also be called as a method. The two ways of calling diff are exactly the same, and
are provided only for convenience.
>>> expr.diff(x, y, y, z, 4)
3 2 ⎛ 3 3 3 2 2 2 ⎞ x⋅y⋅z
x ⋅y ⋅⎝x ⋅y ⋅z + 14⋅x ⋅y ⋅z + 52⋅x⋅y⋅z + 48⎠⋅e
To create an unevaluated derivative, use the Derivative class. It has the same syntax as
diff.
These unevaluated objects are useful for delaying the evaluation of the derivative, or for print-
ing purposes. They are also used when SymPy does not know how to compute the derivative
of an expression (for example, if it contains an undefined function, which are described in the
Solving Differential Equations (page 49) section).
Derivatives of unspecified order can be created using tuple (x, n) where n is the order of
the derivative with respect to x.
>>> m, n, a, b = symbols('m n a b')
>>> expr = (a*x + b)**m
>>> expr.diff((x, n))
n
∂ ⎛ m⎞
───⎝(a⋅x + b) ⎠
n
∂x
2.7.2 Integrals
To compute an integral, use the integrate function. There are two kinds of integrals, definite
and indefinite. To compute an indefinite integral, that is, an antiderivative, or primitive, just
pass the variable after the expression.
>>> integrate(cos(x), x)
sin(x)
Note that SymPy does not include the constant of integration. If you want it, you can add
one yourself, or rephrase your problem as a differential equation and use dsolve to solve it,
which does add the constant (see Solving Differential Equations (page 49)).
Quick Tip
∞ in SymPy is oo (that’s the lowercase letter “oh” twice). This is because oo looks like ∞,
and is easy to type.
2.7. Calculus 41
SymPy Documentation, Release 1.3
we would do
As with indefinite integrals, you can pass multiple limit tuples to perform a multiple integral.
For example, to compute
∫ ∞∫ ∞
e−x −y dx dy,
2 2
−∞ −∞
do
As with Derivative, you can create an unevaluated integral using Integral. To later evaluate
this integral, call doit.
integrate uses powerful algorithms that are always improving to compute both definite and
indefinite integrals, including heuristic pattern matching type algorithms, a partial imple-
mentation of the Risch algorithm, and an algorithm using Meijer G-functions that is useful
for computing integrals in terms of special functions, especially definite integrals. Here is a
sampling of some of the power of integrate.
This last example returned a Piecewise expression because the integral does not converge
unless ℜ(y) > 1.
2.7.3 Limits
SymPy can compute symbolic limits with the limit function. The syntax to compute
lim f (x)
x→x0
is limit(f(x), x, x0).
>>> limit(sin(x)/x, x, 0)
1
2.7. Calculus 43
SymPy Documentation, Release 1.3
limit should be used instead of subs whenever the point of evaluation is a singularity. Even
though SymPy has objects to represent ∞, using them for evaluation is not reliable because
they do not keep track of things like rate of growth. Also, things like ∞ − ∞ and ∞
∞ return nan
(not-a-number). For example
>>> expr = x**2/exp(x)
>>> expr.subs(x, oo)
nan
>>> limit(expr, x, oo)
0
Like Derivative and Integral, limit has an unevaluated counterpart, Limit. To evaluate
it, use doit.
>>> expr = Limit((cos(x) - 1)/x, x, 0)
>>> expr
⎛cos(x) - 1⎞
lim ⎜──────────⎟
x─→0⁺⎝ x ⎠
>>> expr.doit()
0
To evaluate a limit at one side only, pass '+' or '-' as a third argument to limit. For example,
to compute
1
lim ,
x→0+ x
do
>>> limit(1/x, x, 0, '+')
∞
As opposed to
>>> limit(1/x, x, 0, '-')
-∞
SymPy can compute asymptotic series expansions of functions around a point. To compute
the expansion of f (x) around the point x = x0 terms of order xn , use f(x).series(x, x0, n).
x0 and n can be omitted, in which case the defaults x0=0 and n=6 will be used.
>>> expr = exp(sin(x))
>>> expr.series(x, 0, 4)
2
x ⎛ 4⎞
1 + x + ── + O⎝x ⎠
2
( )
The O x4 term at the end represents the Landau order term at x = 0 (not to be confused with
big O notation used in computer science, which generally represents the Landau order term
at x = ∞). It means that all x terms with power greater than or equal to x4 are omitted. Order
terms can be created and manipulated outside of series. They automatically absorb higher
order terms.
If you do not want the order term, use the removeO method.
So far we have looked at expressions with analytic derivatives and primitive functions respec-
tively. But what if we want to have an expression to estimate a derivative of a curve for which
we lack a closed form representation, or for which we don’t know the functional values for
yet. One approach would be to use a finite difference approach.
The simplest way the differentiate using finite differences is to use the differentiate_finite
function:
If we want to expand the intermediate derivative we may pass the flag evaluate=True:
>>> f = Function('f')
>>> dfdx = f(x).diff(x)
>>> dfdx.as_finite_difference()
-f(x - 1/2) + f(x + 1/2)
here the first order derivative was approximated around x using a minimum number of points
(2 for 1st order derivative) evaluated equidistantly using a step-size of 1. We can use arbitrary
steps (possibly containing symbolic expressions):
2.7. Calculus 45
SymPy Documentation, Release 1.3
>>> f = Function('f')
>>> d2fdx2 = f(x).diff(x, 2)
>>> h = Symbol('h')
>>> d2fdx2.as_finite_difference([-3*h,-h,2*h])
f(-3⋅h) f(-h) 2⋅f(2⋅h)
─────── - ───── + ────────
2 2 2
5⋅h 3⋅h 15⋅h
If you are just interested in evaluating the weights, you can do so manually:
note that we only need the last element in the last sublist returned from fi-
nite_diff_weights. The reason for this is that the function also generates weights for lower
derivatives and using fewer points (see the documentation of finite_diff_weights for more
details).
If using finite_diff_weights directly looks complicated, and the as_finite_difference
method of Derivative instances is not flexible enough, you can use apply_finite_diff
which takes order, x_list, y_list and x0 as parameters:
2.8 Solvers
Recall from the gotchas (page 12) section of this tutorial that symbolic equations in SymPy
are not represented by = or ==, but by Eq.
>>> Eq(x, y)
x = y
However, there is an even easier way. In SymPy, any expression not in an Eq is automatically
assumed to equal 0 by the solving functions. Since a = b if and only if a − b = 0, this means
that instead of using x == y, you can just use x - y. For example
This is particularly useful if the equation you wish to solve is already equal to 0. Instead of
typing solveset(Eq(expr, 0), x), you can just use solveset(expr, x).
The main function for solving algebraic equations is solveset. The syntax for solveset is
solveset(equation, variable=None, domain=S.Complexes) Where equations may be in
the form of Eq instances or expressions that are assumed to be equal to zero.
Please note that there is another function called solve which can also be used to solve equa-
tions. The syntax is solve(equations, variables) However, it is recommended to use
solveset instead.
When solving a single equation, the output of solveset is a FiniteSet or an Interval or
ImageSet of the solutions.
>>> solveset(x**2 - x, x)
{0, 1}
>>> solveset(x - x, x, domain=S.Reals)
ℝ
>>> solveset(sin(x) - 1, x, domain=S.Reals)
⎧ π ⎫
⎨2⋅n⋅π + ─ | n ∊ ℤ⎬
⎩ 2 ⎭
If there are no solutions, an EmptySet is returned and if it is not able to find solutions then a
ConditionSet is returned.
In the solveset module, the linear system of equations is solved using linsolve. In future we
would be able to use linsolve directly from solveset. Following is an example of the syntax
of linsolve.
• List of Equations Form:
• A*x = b Form
2.8. Solvers 47
SymPy Documentation, Release 1.3
In the solveset module, the non linear system of equations is solved using nonlinsolve.
Following are examples of nonlinsolve.
1. When only real solution is present:
>>> n = Dummy('n')
>>> system = [exp(x) - sin(y), 1/y - 3]
>>> real_soln = (log(sin(S(1)/3)), S(1)/3)
>>> img_lamda = Lambda(n, 2*n*I*pi + Mod(log(sin(S(1)/3)), 2*I*pi))
>>> complex_soln = (ImageSet(img_lamda, S.Integers), S(1)/3)
>>> soln = FiniteSet(real_soln, complex_soln)
>>> nonlinsolve(system, [x, y]) == soln
True
4. If non linear system of equations is Positive dimensional system (A system with infinitely
many solutions is said to be positive-dimensional):
Note:
1. The order of solution corresponds the order of given symbols.
2. Currently nonlinsolve doesn’t return solution in form of LambertW (if there is solution
present in the form of LambertW).
solve can be used for such cases:
3. Currently nonlinsolve is not properly capable of solving the system of equations having
trigonometric functions.
solve can be used for such cases(not all solution):
solveset reports each solution only once. To get the solutions of a polynomial including
multiplicity use roots.
The output {0: 1, 3: 2} of roots means that 0 is a root of multiplicity 1 and 3 is a root of
multiplicity 2.
Note: Currently solveset is not capable of solving the following types of equations:
• Equations solvable by LambertW (Transcendental equation solver).
solve can be used for such cases:
>>> solve(x*exp(x) - 1, x )
[LambertW(1)]
To solve differential equations, use dsolve. First, create an undefined function by passing
cls=Function to the symbols function.
f and g are now undefined functions. We can call f(x), and it will represent an unknown
function.
>>> f(x)
f(x)
2.8. Solvers 49
SymPy Documentation, Release 1.3
>>> f(x).diff(x)
d
──(f(x))
dx
To solve the ODE, pass it and the function to solve for to dsolve.
dsolve returns an instance of Eq. This is because in general, solutions to differential equa-
tions cannot be solved explicitly for the function.
The arbitrary constants in the solutions from dsolve are symbols of the form C1, C2, C3, and
so on.
2.9 Matrices
To make a matrix in SymPy, use the Matrix object. A matrix is constructed by providing a list
of row vectors that make up the matrix. For example, to construct the matrix
1 −1
3 4
0 2
use
To make it easy to make column vectors, a list of elements is considered to be a column vector.
Matrices are manipulated just like any other object in SymPy or Python.
One important thing to note about SymPy matrices is that, unlike every other object in SymPy,
they are mutable. This means that they can be modified in place, as we will see below. The
downside to this is that Matrix cannot be used in places that require immutability, such as
inside other SymPy expressions or as keys to dictionaries. If you need an immutable version
of Matrix, use ImmutableMatrix.
Shape
Here are some basic operations on Matrix. To get the shape of a matrix use shape
To get an individual row or column of a matrix, use row or col. For example, M.row(0) will
get the first row. M.col(-1) will get the last column.
>>> M.row(0)
[1 2 3]
>>> M.col(-1)
⎡3⎤
⎢ ⎥
⎣4⎦
2.9. Matrices 51
SymPy Documentation, Release 1.3
To delete a row or column, use row_del or col_del. These operations will modify the Matrix
in place.
>>> M.col_del(0)
>>> M
⎡2 3⎤
⎢ ⎥
⎣0 4⎦
>>> M.row_del(1)
>>> M
[2 3]
To insert rows or columns, use row_insert or col_insert. These operations do not operate
in place.
>>> M
[2 3]
>>> M = M.row_insert(1, Matrix([[0, 4]]))
>>> M
⎡2 3⎤
⎢ ⎥
⎣0 4⎦
>>> M = M.col_insert(0, Matrix([1, -2]))
>>> M
⎡1 2 3⎤
⎢ ⎥
⎣-2 0 4⎦
Unless explicitly stated, the methods mentioned below do not operate in place. In general,
a method that does not operate in place will return a new Matrix and a method that does
operate in place will return None.
As noted above, simple operations like addition and multiplication are done just by using +,
*, and **. To find the inverse of a matrix, just raise it to the -1 power.
Several constructors exist for creating common matrices. To create an identity matrix, use
eye. eye(n) will create an n × n identity matrix.
>>> eye(3)
⎡1 0 0⎤
⎢ ⎥
⎢0 1 0⎥
⎢ ⎥
⎣0 0 1⎦
>>> eye(4)
⎡1 0 0 0⎤
⎢ ⎥
⎢0 1 0 0⎥
⎢ ⎥
⎢0 0 1 0⎥
⎢ ⎥
⎣0 0 0 1⎦
To create a matrix of all zeros, use zeros. zeros(n, m) creates an n × m matrix of 0s.
>>> zeros(2, 3)
⎡0 0 0⎤
⎢ ⎥
⎣0 0 0⎦
2.9. Matrices 53
SymPy Documentation, Release 1.3
>>> ones(3, 2)
⎡1 1⎤
⎢ ⎥
⎢1 1⎥
⎢ ⎥
⎣1 1⎦
To create diagonal matrices, use diag. The arguments to diag can be either numbers or
matrices. A number is interpreted as a 1 × 1 matrix. The matrices are stacked diagonally. The
remaining elements are filled with 0s.
>>> diag(1, 2, 3)
⎡1 0 0⎤
⎢ ⎥
⎢0 2 0⎥
⎢ ⎥
⎣0 0 3⎦
>>> diag(-1, ones(2, 2), Matrix([5, 7, 5]))
⎡-1 0 0 0⎤
⎢ ⎥
⎢0 1 1 0⎥
⎢ ⎥
⎢0 1 1 0⎥
⎢ ⎥
⎢0 0 0 5⎥
⎢ ⎥
⎢0 0 0 7⎥
⎢ ⎥
⎣0 0 0 5⎦
Determinant
RREF
To put a matrix into reduced row echelon form, use rref. rref returns a tuple of two elements.
The first is the reduced row echelon form, and the second is a tuple of indices of the pivot
columns.
Note: The first element of the tuple returned by rref is of type Matrix. The second is of
type tuple.
Nullspace
To find the nullspace of a matrix, use nullspace. nullspace returns a list of column vectors
that span the nullspace of the matrix.
Columnspace
To find the columnspace of a matrix, use columnspace. columnspace returns a list of column
vectors that span the columnspace of the matrix.
2.9. Matrices 55
SymPy Documentation, Release 1.3
To find the eigenvalues of a matrix, use eigenvals. eigenvals returns a dictionary of eigen-
value:algebraic multiplicity pairs (similar to the output of roots (page 49)).
>>> M = Matrix([[3, -2, 4, -2], [5, 3, -3, -2], [5, -2, 2, -2], [5, -2, -3, 3]])
>>> M
⎡3 -2 4 -2⎤
⎢ ⎥
⎢5 3 -3 -2⎥
⎢ ⎥
⎢5 -2 2 -2⎥
⎢ ⎥
⎣5 -2 -3 3 ⎦
>>> M.eigenvals()
{-2: 1, 3: 1, 5: 2}
This means that M has eigenvalues -2, 3, and 5, and that the eigenvalues -2 and 3 have alge-
braic multiplicity 1 and that the eigenvalue 5 has algebraic multiplicity 2.
To find the eigenvectors of a matrix, use eigenvects. eigenvects returns a list of tuples of
the form (eigenvalue:algebraic multiplicity, [eigenvectors]).
>>> M.eigenvects()
⎡⎛ ⎡⎡0⎤⎤⎞ ⎛ ⎡⎡1⎤⎤⎞ ⎛ ⎡⎡1⎤ ⎡0 ⎤⎤⎞⎤
⎢⎜ ⎢⎢ ⎥⎥⎟ ⎜ ⎢⎢ ⎥⎥⎟ ⎜ ⎢⎢ ⎥ ⎢ ⎥⎥⎟⎥
⎢⎜ ⎢⎢1⎥⎥⎟ ⎜ ⎢⎢1⎥⎥⎟ ⎜ ⎢⎢1⎥ ⎢-1⎥⎥⎟⎥
⎢⎜-2, 1, ⎢⎢ ⎥⎥⎟, ⎜3, 1, ⎢⎢ ⎥⎥⎟, ⎜5, 2, ⎢⎢ ⎥, ⎢ ⎥⎥⎟⎥
⎢⎜ ⎢⎢1⎥⎥⎟ ⎜ ⎢⎢1⎥⎥⎟ ⎜ ⎢⎢1⎥ ⎢0 ⎥⎥⎟⎥
⎢⎜ ⎢⎢ ⎥⎥⎟ ⎜ ⎢⎢ ⎥⎥⎟ ⎜ ⎢⎢ ⎥ ⎢ ⎥⎥⎟⎥
⎣⎝ ⎣⎣1⎦⎦⎠ ⎝ ⎣⎣1⎦⎦⎠ ⎝ ⎣⎣0⎦ ⎣1 ⎦⎦⎠⎦
This shows us that, for example, the eigenvalue 5 also has geometric multiplicity 2, because
it has two eigenvectors. Because the algebraic and geometric multiplicities are the same for
all the eigenvalues, M is diagonalizable.
To diagonalize a matrix, use diagonalize. diagonalize returns a tuple (P, D), where D is
diagonal and M = P DP −1 .
>>> P, D = M.diagonalize()
>>> P
⎡0 1 1 0 ⎤
⎢ ⎥
⎢1 1 1 -1⎥
(continues on next page)
Quick Tip
lambda is a reserved keyword in Python, so to create a Symbol called λ, while using the
same names for SymPy Symbols and Python variables, use lamda (without the b). It will
still pretty print as λ.
Note that since eigenvects also includes the eigenvalues, you should use it instead of eigen-
vals if you also want the eigenvectors. However, as computing the eigenvectors may often
be costly, eigenvals should be preferred if you only wish to find the eigenvalues.
If all you want is the characteristic polynomial, use charpoly. This is more efficient than
eigenvals, because sometimes symbolic roots can be expensive to calculate.
Zero Testing
If your matrix operations are failing or returning wrong answers, the common reasons would
likely be from zero testing. If there is an expression not properly zero-tested, it can possibly
bring issues in finding pivots for gaussian elimination, or deciding whether the matrix is
inversible, or any high level functions which relies on the prior procedures.
2.9. Matrices 57
SymPy Documentation, Release 1.3
Currently, the SymPy’s default method of zero testing _iszero is only guaranteed to be ac-
curate in some limited domain of numerics and symbols, and any complicated expressions
beyond its decidability are treated as None, which behaves similarly to logical False.
The list of methods using zero testing procedures are as followings.
echelon_form , is_echelon , rank , rref , nullspace , eigenvects , inverse_ADJ , in-
verse_GE , inverse_LU , LUdecomposition , LUdecomposition_Simple , LUsolve
They have property iszerofunc opened up for user to specify zero testing method, which can
accept any function with single input and boolean output, while being defaulted with _iszero.
Here is an example of solving an issue caused by undertested zero.1 2
You can trace down which expression is being underevaluated, by injecting a custom zero test
with warnings enabled.
[]
1 Inspired by https://gitter.im/sympy/sympy?at=5b7c3e8ee5b40332abdb206c
2
Discovered from https://github.com/sympy/sympy/issues/15141
You can clearly see nullspace returning proper result, after injecting an alternative zero test.
Note that this approach is only valid for some limited cases of matrices containing only nu-
merics, hyperbolics, and exponentials. For other matrices, you should use different method
opted for their domains.
Possible suggestions would be either taking advantage of rewriting and simplifying, with
tradeoff of speed3 , or using random numeric testing, with tradeoff of accuracy4 .
If you wonder why there is no generic algorithm for zero testing that can work with any sym-
bolic entities, it’s because of the constant problem stating that zero testing is undecidable5
, and not only the SymPy, but also other computer algebra systems67 would face the same
fundamental issue.
However, discovery of any zero test failings can provide some good examples to improve
SymPy, so if you have encountered one, you can report the issue to SymPy issue tracker8 to
get detailed help from the community.
3 Suggested from https://github.com/sympy/sympy/issues/10120
4
Suggested from https://github.com/sympy/sympy/issues/10279
5
https://en.wikipedia.org/wiki/Constant_problem
6
How mathematica tests zero https://reference.wolfram.com/language/ref/PossibleZeroQ.html
7
How matlab tests zero https://www.mathworks.com/help/symbolic/mupad_ref/iszero.html
8
https://github.com/sympy/sympy/issues
2.9. Matrices 59
SymPy Documentation, Release 1.3
In this section, we discuss some ways that we can perform advanced manipulation of expres-
sions.
Quick Tip
To play with the srepr form of expressions in the SymPy Live shell, change the output
format to Repr in the settings.
Before we can do this, we need to understand how expressions are represented in SymPy.
A mathematical expression is represented as a tree. Let us take the expression 2x + xy, i.e.,
2**x + x*y. We can see what this expression looks like internally by using srepr
The easiest way to tear this apart is to look at a diagram of the expression tree:
Add
Mul Pow
Note: The above diagram was made using Graphviz and the dotprint (page 1086) function.
First, let’s look at the leaves of this tree. Symbols are instances of the class Symbol. While
we have been doing
>>> x = symbols('x')
>>> x = Symbol('x')
Either way, we get a Symbol with the name “x”1 . For the number in the expression, 2, we got
Integer(2). Integer is the SymPy class for integers. It is similar to the Python built-in type
int, except that Integer plays nicely with other SymPy types.
When we write 2**x, this creates a Pow object. Pow is short for “power”.
>>> srepr(2**x)
”Pow(Integer(2), Symbol('x'))”
>>> Pow(2, x)
2**x
Note that in the srepr output, we see Integer(2), the SymPy version of integers, even though
technically, we input 2, a Python int. In general, whenever you combine a SymPy object with
a non-SymPy object via some function or operation, the non-SymPy object will be converted
into a SymPy object. The function that does this is sympify2 .
>>> type(2)
<... 'int'>
>>> type(sympify(2))
<class 'sympy.core.numbers.Integer'>
We have seen that 2**x is represented as Pow(2, x). What about x*y? As we might expect,
this is the multiplication of x and y. The SymPy class for multiplication is Mul.
>>> srepr(x*y)
”Mul(Symbol('x'), Symbol('y'))”
Thus, we could have created the same object by writing Mul(x, y).
>>> Mul(x, y)
x*y
Now we get to our final expression, 2**x + x*y. This is the addition of our last two objects,
Pow(2, x), and Mul(x, y). The SymPy class for addition is Add, so, as you might expect, to
create this object, we use Add(Pow(2, x), Mul(x, y)).
SymPy expression trees can have many branches, and can be quite deep or quite broad. Here
is a more complicated example
1 We have been using symbols instead of Symbol because it automatically splits apart strings into multiple Symbols.
symbols('x y z') returns a tuple of three Symbols. Symbol('x y z') returns a single Symbol called x y z.
2 Technically, it is an internal function called _sympify, which differs from sympify in that it does not convert
Here is a diagram
Add
Symbol('y') Symbol('x')
This expression reveals some interesting things about SymPy expression trees. Let’s go
through them one by one.
Let’s first look at the term x**2. As we expected, we see Pow(x, 2). One level up, we see we
have Mul(-1, Pow(x, 2)). There is no subtraction class in SymPy. x - y is represented as
x + -y, or, more completely, x + -1*y, i.e., Add(x, Mul(-1, y)).
>>> srepr(x - y)
”Add(Symbol('x'), Mul(Integer(-1), Symbol('y')))”
Add
Symbol('x') Mul
Symbol('y') Integer(-1)
Next, look at 1/y. We might expect to see something like Div(1, y), but similar to subtrac-
tion, there is no class in SymPy for division. Rather, division is represented by a power of -1.
Hence, we have Pow(y, -1). What if we had divided something other than 1 by y, like x/y?
Let’s see.
Mul
Symbol('x') Pow
Symbol('y') Integer(-1)
Finally, let’s look at the sin(x*y)/2 term. Following the pattern of the previous exam-
ple, we might expect to see Mul(sin(x*y), Pow(Integer(2), -1)). But instead, we have
Mul(Rational(1, 2), sin(x*y)). Rational numbers are always combined into a single term
in a multiplication, so that when we divide by 2, it is represented as multiplying by 1/2.
Finally, one last note. You may have noticed that the order we entered our expression and the
order that it came out from srepr or in the graph were different. You may have also noticed
this phenomenon earlier in the tutorial. For example
>>> 1 + x
x + 1
This because in SymPy, the arguments of the commutative operations Add and Mul are stored
in an arbitrary (but consistent!) order, which is independent of the order inputted (if you’re
worried about noncommutative multiplication, don’t be. In SymPy, you can create noncom-
mutative Symbols using Symbol('A', commutative=False), and the order of multiplication
for noncommutative Symbols is kept the same as the input). Furthermore, as we shall see in
the next section, the printing order and the order in which things are stored internally need
not be the same either.
Quick Tip
The way an expression is represented internally and the way it is printed are often not the
same.
In general, an important thing to keep in mind when working with SymPy expression trees
is this: the internal representation of an expression and the way it is printed need not be the
same. The same is true for the input form. If some expression manipulation algorithm is not
working in the way you expected it to, chances are, the internal representation of the object
is different from what you thought it was.
Now that you know how expression trees work in SymPy, let’s look at how to dig our way
through an expression tree. Every object in SymPy has two very important attributes, func,
and args.
func
func is the head of the object. For example, (x*y).func is Mul. Usually it is the same as the
class of the object (though there are exceptions to this rule).
Two notes about func. First, the class of an object need not be the same as the one used to
create it. For example
We created Add(x, x), so we might expect expr.func to be Add, but instead we got Mul. Why
is that? Let’s take a closer look at expr.
>>> expr
2*x
Add(x, x), i.e., x + x, was automatically converted into Mul(2, x), i.e., 2*x, which is a Mul.
SymPy classes make heavy use of the __new__ class constructor, which, unlike __init__,
allows a different class to be returned from the constructor.
Second, some classes are special-cased, usually for efficiency reasons3 .
>>> Integer(2).func
<class 'sympy.core.numbers.Integer'>
>>> Integer(0).func
<class 'sympy.core.numbers.Zero'>
>>> Integer(-1).func
<class 'sympy.core.numbers.NegativeOne'>
For the most part, these issues will not bother us. The special classes Zero, One, NegativeOne,
and so on are subclasses of Integer, so as long as you use isinstance, it will not be an issue.
args
args are the top-level arguments of the object. (x*y).args would be (x, y). Let’s look at
some examples
From this, we can see that expr == Mul(3, y**2, x). In fact, we can see that we can com-
pletely reconstruct expr from its func and its args.
>>> expr.func(*expr.args)
3*x*y**2
>>> expr == expr.func(*expr.args)
True
Note that although we entered 3*y**2*x, the args are (3, x, y**2). In a Mul, the Rational
coefficient will come first in the args, but other than that, the order of everything else follows
no special pattern. To be sure, though, there is an order.
Mul’s args are sorted, so that the same Mul will have the same args. But the sorting is based
on some criteria designed to make the sorting unique and efficient that has no mathematical
significance.
3 Classes like One and Zero are singletonized, meaning that only one object is ever created, no matter how many
times the class is called. This is done for space efficiency, as these classes are very common. For example, Zero
might occur very often in a sparse matrix represented densely. As we have seen, NegativeOne occurs any time we
have -x or 1/x. It is also done for speed efficiency because singletonized objects can be compared by is. The unique
objects for each singletonized class can be accessed from the S object.
The srepr form of our expr is Mul(3, x, Pow(y, 2)). What if we want to get at the args of
Pow(y, 2). Notice that the y**2 is in the third slot of expr.args, i.e., expr.args[2].
>>> expr.args[2]
y**2
>>> expr.args[2].args
(y, 2)
Now what if we try to go deeper. What are the args of y. Or 2. Let’s see.
>>> y.args
()
>>> Integer(2).args
()
They both have empty args. In SymPy, empty args signal that we have hit a leaf of the
expression tree.
So there are two possibilities for a SymPy expression. Either it has empty args, in which case
it is a leaf node in any expression tree, or it has args, in which case, it is a branch node of any
expression tree. When it has args, it can be completely rebuilt from its func and its args.
This is expressed in the key invariant.
Key Invariant
Every well-formed SymPy expression must either have empty args or satisfy expr == expr.
func(*expr.args).
(Recall that in Python if a is a tuple, then f(*a) means to call f with arguments from the
elements of a, e.g., f(*(1, 2, 3)) is the same as f(1, 2, 3).)
This key invariant allows us to write simple algorithms that walk expression trees, change
them, and rebuild them into new expressions.
With this knowledge, let’s look at how we can recurse through an expression tree. The nested
nature of args is a perfect fit for recursive functions. The base case will be empty args. Let’s
write a simple function that goes through an expression and prints all the args at each level.
See how nice it is that () signals leaves in the expression tree. We don’t even have to write
a base case for our recursion; it is handled automatically by the for loop.
Let’s test our function.
Can you guess why we called our function pre? We just wrote a pre-order traversal function
for our expression tree. See if you can write a post-order traversal function.
Such traversals are so common in SymPy that the generator functions preorder_traversal
and postorder_traversal are provided to make such traversals easy. We could have also
written our algorithm as
There are generally two ways to prevent the evaluation, either pass an evaluate=False pa-
rameter while constructing the expression, or create an evaluation stopper by wrapping the
expression with UnevaluatedExpr.
For example:
If you don’t remember the class corresponding to the expression you want to build (operator
overloading usually assumes evaluate=True), just use sympify and pass a string:
Note that evaluate=False won’t prevent future evaluation in later usages of the expression:
That’s why the class UnevaluatedExpr comes handy. UnevaluatedExpr is a method provided
by SymPy which lets the user keep an expression unevaluated. By unevaluated it is meant
that the value inside of it will not interact with the expressions outside of it to give simplified
outputs. For example:
>>> (x + expr).doit()
3*x
Other examples:
Remember that expr2 will be evaluated if included into another expression. Combine both of
the methods to prevent both inside and outside evaluations:
UnevalutedExpr is supported by SymPy printers and can be used to print the result in differ-
ent output forms. For example
In order to release the expression and get the evaluated LaTeX form, just use .doit():
>>> print(latex(uexpr.doit()))
\frac{15}{28}
THREE
3.1 Introduction
SymPy runs under the Python Programming Language, so there are some things that may
behave differently than they do in other, independent computer algebra systems like Maple
or Mathematica. These are some of the gotchas and pitfalls that you may encounter when
using SymPy. See also the FAQ, the Tutorial (page 5), the remainder of the SymPy Docs, and
the official Python Tutorial.
If you are already familiar with C or Java, you might also want to look at this 4 minute Python
tutorial.
Ignore #doctest: +SKIP in the examples. That has to do with internal testing of the examples.
The equals sign (=) is the assignment operator, not equality. If you want to do x = y, use
Eq(x, y) for equality. Alternatively, all expressions are assumed to equal zero, so you can
just subtract one side and use x - y.
The proper use of the equals sign is to assign expressions to variables.
For example:
Double equals signs (==) are used to test equality. However, this tests expressions exactly,
not symbolically. For example:
69
SymPy Documentation, Release 1.3
If you want to test for symbolic equality, one way is to subtract one expression from the
other and run it through functions like expand(), simplify(), and trigsimp() and see if the
equation reduces to 0.
>>> from sympy import simplify, cos, sin, expand
>>> simplify((x + 1)**2 - (x**2 + 2*x + 1))
0
>>> eq = sin(2*x) - 2*sin(x)*cos(x)
>>> simplify(eq)
0
>>> expand(eq, trig=True)
0
Note: See also Why does SymPy say that two equal expressions are unequal? in the FAQ.
3.3 Variables
When you use = to do assignment, remember that in Python, as in most programming lan-
guages, the variable does not change if you change the value you assigned to it. The equa-
tions you are typing use the values present at the time of creation to “fill in” values, just like
regular Python definitions. They are not altered by changes made afterwards. Consider the
following:
>>> from sympy import Symbol
>>> a = Symbol('a') # Symbol, `a`, stored as variable ”a”
>>> b = a + 1 # an expression involving `a` stored as variable ”b”
>>> print(b)
a + 1
>>> a = 4 # ”a” now points to literal integer 4, not Symbol('a')
>>> print(a)
4
>>> print(b) # ”b” is still pointing at the expression involving `a`
a + 1
Changing quantity a does not change b; you are not working with a set of simultaneous equa-
tions. It might be helpful to remember that the string that gets printed when you print a
variable referring to a SymPy object is the string that was given to it when it was created;
that string does not have to be the same as the variable that you assign it to.
>>> from sympy import var
>>> r, t, d = var('rate time short_life')
>>> d = r*t
>>> print(d)
(continues on next page)
If you need variables that have dependence on each other, you can define functions. Use the
def operator. Indent the body of the function. See the Python docs for more information on
defining functions.
Note: See also Why doesn’t changing one variable change another that depends on it? in
3.3. Variables 71
SymPy Documentation, Release 1.3
the FAQ.
3.3.2 Symbols
Symbols are variables, and like all other variables, they need to be assigned before you can
use them. For example:
If you use isympy, it runs the following commands for you, giving you some default Symbols
and Functions.
If you want control over the assumptions of the variables, use Symbol() and symbols(). See
Keyword Arguments (page 80) below.
Lastly, it is recommended that you not use I, E, S, N, C, O, or Q for variable or symbol names, as
those are used for the imaginary unit (i), the base of the natural logarithm (e), the sympify()
function (see Symbolic Expressions (page 73) below), numeric evaluation (N() is equivalent to
evalf() (page 393) ), the big O order symbol (as in O(n log n)), and the assumptions object that
holds a list of supported ask keys (such as Q.real), respectively. You can use the mnemonic
OSINEQ to remember what Symbols are defined by default in SymPy. Or better yet, always
use lowercase letters for Symbol names. Python will not prevent you from overriding default
SymPy names or functions, so be careful.
If you have IPython installed and use isympy, you can also press the TAB key to get a list of all
built-in names and to autocomplete. Also, see this page for a trick for getting tab completion
in the regular Python console.
Note: See also What is the best way to create symbols? in the FAQ.
SymPy uses its own classes for integers, rational numbers, and floating point numbers instead
of the default Python int and float types because it allows for more control. But you have
to be careful. If you type an expression that just has numbers in it, it will default to a Python
expression. Use the sympify() function, or just S(), to ensure that something is a SymPy
expression.
>>> 6.2 # Python float. Notice the floating point accuracy problems.
6.2000000000000002
>>> type(6.2) # <type 'float'> in Python 2.x, <class 'float'> in Py3k
<... 'float'>
>>> S(6.2) # SymPy Float has no such problems because of arbitrary precision.
6.20000000000000
>>> type(S(6.2))
<class 'sympy.core.numbers.Float'>
If you include numbers in a SymPy expression, they will be sympified automatically, but there
is one gotcha you should be aware of. If you do <number>/<number> inside of a SymPy ex-
pression, Python will evaluate the two numbers before SymPy has a chance to get to them.
The solution is to sympify() one of the numbers, or use Rational.
If the two integers are not directly separated by a division sign then you don’t have to worry
about this problem:
>>> x**(2*x/3)
x**(2*x/3)
Note: A common mistake is copying an expression that is printed and reusing it. If the
expression has a Rational (i.e., <number>/<number>) in it, you will not get the same result,
obtaining the Python result for the division rather than a SymPy Rational.
>>> x = Symbol('x')
>>> print(solve(7*x -22, x))
[22/7]
>>> 22/7 # If we just copy and paste we get int 3 or a float
3.142857142857143
>>> # One solution is to just assign the expression to a variable
>>> # if we need to use it again.
>>> a = solve(7*x - 22, x)[0]
>>> a
22/7
The other solution is to put quotes around the expression and run it through S() (i.e., sympify
it):
>>> S(”22/7”)
22/7
Also, if you do not use isympy, you could use from __future__ import division to prevent
the / sign from performing integer division.
But be careful: you will now receive floats where you might have desired a Rational:
>>> x**(1/2)
x**0.5
Rational only works for number/number and is only meant for rational numbers. If you
want a fraction with symbols or expressions in it, just use /. If you do number/expression or
expression/number, then the number will automatically be converted into a SymPy Number.
You only need to be careful with number/number.
>>> Rational(2, x)
Traceback (most recent call last):
...
TypeError: invalid input: x
>>> 2/x
2/x
SymPy keeps track of the precision of Float objects. The default precision is 15 digits. When
an expression involving a Float is evaluated, the result will be expressed to 15 digits of
precision but those digits (depending on the numbers involved with the calculation) may not
all be significant.
The first issue to keep in mind is how the Float is created: it is created with a value and
a precision. The precision indicates how precise of a value to use when that Float (or an
expression it appears in) is evaluated.
The values can be given as strings, integers, floats, or rationals.
• strings and integers are interpreted as exact
>>> Float(100)
100.000000000000
>>> Float('100', 5)
100.00
• to have the precision match the number of digits, the null string can be used
for the precision
>>> Float(100, '')
100.
>>> Float('12.34')
12.3400000000000
>>> Float('12.34', '')
12.34
Next, notice that each of those values looks correct to 3 digits. But if we try to evaluate them
to 20 digits, a difference will become apparent:
The 0.25 (with precision of 3) represents a number that has a non-repeating binary
decimal; 1/7 is repeating in binary and decimal – it cannot be represented accurately
too far past those first 3 digits (the correct decimal is a repeating 142857):
>>> s.n(20)
0.25000000000000000000
>>> r.n(20)
0.14285278320312500000
Although the evalf engine tried to maintain 10 digits of precision (since that was the highest
precision represented) the 3-digit precision used limits the accuracy to about 4 digits – not all
the digits you see are significant. evalf doesn’t try to keep track of the number of significant
digits.
That very simple expression involving the addition of two numbers with different precisions
will hopefully be instructive in helping you understand why more complicated expressions
(like trig expressions that may not be simplified) will not evaluate to an exact zero even
though, with the right simplification, they should be zero. Consider this unsimplified trig
identity, multiplied by a big number:
When the cos and sin terms were evaluated to 15 digits of precision and multiplied by the
big number, they gave a large number that was only precise to 15 digits (approximately) and
when the 20 digit big number was subtracted the result was not zero.
There are three things that will help you obtain more precise numerical values for expressions:
1) Pass the desired substitutions with the call to evaluate. By doing the subs first,
the Float values cannot be updated as necessary. By passing the desired substitu-
tions with the call to evalf the ability to re-evaluate as necessary is gained and the
results are impressively better:
2) Use Rationals, not Floats. During the evaluation process, the Rational can be
computed to an arbitrary precision while the Float, once created – at a default of
15 digits – cannot. Compare the value of -1.4e+3 above with the nearly zero value
obtained when replacing x with a Rational representing 1/10 – before the call to
evaluate:
3) Try to simplify the expression. In this case, SymPy will recognize the trig identity
and simplify it to zero so you don’t even have to evaluate it numerically:
>>> big_trig_identity.simplify()
0
Expressions in SymPy are immutable, and cannot be modified by an in-place operation. This
means that a function will always return an object, and the original expression will not be
modified. The following example snippet demonstrates how this works:
def main():
var('x y a b')
expr = 3*x + 4*y
print('original =', expr)
expr_modified = expr.subs({x: a, y: b})
print('modified =', expr_modified)
if __name__ == ”__main__”:
main()
The output shows that the subs() function has replaced variable x with variable a, and vari-
able y with variable b:
The subs() function does not modify the original expression expr. Rather, a modified copy of
the expression is returned. This returned object is stored in the variable expr_modified. Note
that unlike C/C++ and other high-level languages, Python does not require you to declare a
variable before it is used.
SymPy uses the same default operators as Python. Most of these, like */+-, are standard.
Aside from integer division discussed in Python numbers vs. SymPy Numbers (page 73) above,
you should also be aware that implied multiplication is not allowed. You need to use * when-
ever you wish to multiply something. Also, to raise something to a power, use **, not ^ as
many computer algebra systems use. Parentheses () change operator precedence as you
would normally expect.
In isympy, with the ipython shell:
>>> 2x
Traceback (most recent call last):
...
SyntaxError: invalid syntax
>>> 2*x
2*x
>>> (x + 1)^2 # This is not power. Use ** instead.
Traceback (most recent call last):
...
TypeError: unsupported operand type(s) for ^: 'Add' and 'int'
>>> (x + 1)**2
(x + 1)**2
>>> pprint(3 - x**(2*x)/(x + 1))
2*x
x
- ----- + 3
x + 1
SymPy uses different names for some functions than most computer algebra systems. In par-
ticular, the inverse trig functions use the python names of asin(), acos() and so on instead
of the usual arcsin and arccos. Use the methods described in Symbols (page 72) above to
see the names of all SymPy functions.
The symbols [], {}, =, and () have special meanings in Python, and thus in SymPy. See the
Python docs linked to above for additional information.
3.5.1 Lists
Square brackets [] denote a list. A list is a container that holds any number of different
objects. A list can contain anything, including items of different types. Lists are mutable,
which means that you can change the elements of a list after it has been created. You access
the items of a list also using square brackets, placing them after the list or list variable. Items
are numbered using the space before the item.
Example:
>>> a = [x, 1] # A simple list of two items
>>> a
[x, 1]
>>> a[0] # This is the first item
x
>>> a[0] = 2 # You can change values of lists after they have been created
>>> print(a)
[2, 1]
>>> print(solve(x**2 + 2*x - 1, x)) # Some functions return lists
[-1 + sqrt(2), -sqrt(2) - 1]
Note: See the Python docs for more information on lists and the square bracket notation for
accessing elements of a list.
3.5.2 Dictionaries
Curly brackets {} denote a dictionary, or a dict for short. A dictionary is an unordered list of
non-duplicate keys and values. The syntax is {key: value}. You can access values of keys
using square bracket notation.
>>> d = {'a': 1, 'b': 2} # A dictionary.
>>> d
{'a': 1, 'b': 2}
(continues on next page)
3.5.3 Tuples
Parentheses (), aside from changing operator precedence and their use in function calls, (like
cos(x)), are also used for tuples. A tuple is identical to a list (page 78), except that it is not
mutable. That means that you cannot change their values after they have been created. In
general, you will not need tuples in SymPy, but sometimes it can be more convenient to type
parentheses instead of square brackets.
>>> (x,)
(x,)
>>> (x)
x
integrate takes a sequence as the second argument if you want to integrate with
limits (and a tuple or list will work):
Aside from the usage described above (page 69), equals signs (=) are also used to give named
arguments to functions. Any function that has key=value in its parameters list (see below
on how to find this out), then key is set to value by default. You can change the value of the
key by supplying your own value using the equals sign in the function call. Also, functions
that have ** followed by a name in the parameters list (usually **kwargs or **assumptions)
allow you to add any number of key=value pairs that you want, and they will all be evaluated
according to the function.
sqrt(x**2) doesn’t auto simplify to x because x is assumed to be complex by de-
fault, and, for example, sqrt((-1)**2) == sqrt(1) == 1 != -1:
>>> sqrt(x**2)
sqrt(x**2)
The square root will now simplify since it knows that x >= 0:
>>> sqrt(x**2)
x
>>> pprint(powsimp(x**n*x**m*y**n*y**m))
m + n
(x*y)
Setting combine to the default value is the same as not setting it.
Note: See the Python docs for more information on function parameters.
3.6.1 help()
Although all docs are available at docs.sympy.org or on the SymPy Wiki, you can also get info
on functions from within the Python interpreter that runs SymPy. The easiest way to do this
is to do help(function), or function? if you are using ipython:
In [2]: # But in ipython, you can also use ?, which is better because it
In [3]: # it gives you more information
In [4]: powsimp?
These will give you the function parameters and docstring for powsimp(). The output will
look something like this:
sympy.simplify.simplify.powsimp(expr, deep=False, combine=’all’, force=False,
measure=<function count_ops>)
reduces expression by combining powers with similar bases and exponents.
Notes
If deep is True then powsimp() will also simplify arguments of functions. By default deep
is set to False.
If force is True then bases will be combined without checking for assumptions, e.g.
sqrt(x)*sqrt(y) -> sqrt(x*y) which is not true if x and y are both negative.
You can make powsimp() only combine bases or only combine exponents by changing
combine=’base’ or combine=’exp’. By default, combine=’all’, which does both. com-
bine=’base’ will only combine:
a a a 2x x
x * y => (x*y) as well as things like 2 => 4
a b (a + b)
x * x => x
combine=’exp’ will strictly only combine exponents in the way that used to be automatic.
Also use deep=True if you need the old behavior.
When combine=’all’, ‘exp’ is evaluated first. Consider the first example below for when
there could be an ambiguity relating to this. This is done so things like the second
example can be completely combined. If you want ‘base’ combined first, do something
like powsimp(powsimp(expr, combine=’base’), combine=’exp’).
Examples
>>> a=sqrt(x*sqrt(y))
>>> a*a**3 == a**4
True
But if an integer power of that radical has been autoexpanded then Mul does not join
the resulting factors:
3.6.2 source()
Another useful option is the source() function. This will print the source code of a func-
tion, including any docstring that it may have. You can also do function?? in ipython. For
example, from SymPy 0.6.5:
FOUR
4.1 Introduction
If you are new to SymPy, start with the Tutorial (page 5). If you went through it, now it’s time
to learn how SymPy works internally, which is what this guide is about. Once you grasp the
ideas behind SymPy, you will be able to use it effectively and also know how to extend it and
fix it. You may also be just interested in SymPy Modules Reference (page 95).
Let’s say I’d like to understand how x + y + x works and how it is possible that it gets sim-
plified to 2*x + y.
I write a simple script, I usually call it t.py (I don’t remember anymore why I call it that way):
e = x + y + x
print e
85
SymPy Documentation, Release 1.3
Then I step into (F7) and after a little debugging I get for example:
Tip: Make the winpdb window larger on your screen, it was just made smaller to fit in this
guide.
I see values of all local variables in the left panel, so it’s very easy to see what’s happening. You
can see, that the y + 2*x is emerging in the obj variable. Observing that obj is constructed
from c_part and nc_part and seeing what c_part contains (y and 2*x). So looking at the line
28 (the whole line is not visible on the screenshot, so here it is):
you can see that the simplification happens in cls.flatten. Now you can set the breakpoint
on the line 28, quit winpdb (it will remember the breakpoint), start it again, hit F5, this will
stop at this breakpoint, hit F7, this will go into the function Add.flatten():
@classmethod
def flatten(cls, seq):
”””
Takes the sequence ”seq” of nested Adds and returns a flatten list.
and then you can study how it works. I am going to stop here, this should be enough to get
you going – with the above technique, I am able to understand almost any Python code.
We try to make the sources easily understandable, so you can look into the sources and read
the doctests, it should be well documented and if you don’t understand something, ask on the
mailinglist.
You can find all the decisions archived in the issues, to see rationale for doing this and that.
4.3.1 Basics
All symbolic things are implemented using subclasses of the Basic class. First, you need to
create symbols using Symbol(”x”) or numbers using Integer(5) or Float(34.3). Then you
construct the expression using any class from SymPy. For example Add(Symbol(”a”), Sym-
bol(”b”)) gives an instance of the Add class. You can call all methods, which the particular
class supports.
For easier use, there is a syntactic sugar for expressions.
For example, cos(x) + 1 is equal to cos(x).__add__(1) is equal to Add(cos(x), Inte-
ger(1))
This works becauses Python allows you to override the behavior of built-in operations +, *,
-, / etc., by defining methods with names __add__, __mul__ etc., so cos(x) + 1 is evaluated
as cos(x).__add__(1).
Then, in the __add__, __mul__ methods defined for SymPy objects, new Add or Mul objects
are constructed. Python types are also converted to SymPy types through the sympify()
function. Thus, cos(x).__add__(1) returns Add(cos(x), Integer(1)).
Similarly, 2/cos(x) is equal to cos(x).__rdiv__(2) is equal to Mul(Integer(2),
Pow(cos(x), Integer(-1))).
Note that 2/cos(x) calls cos(x).__rdiv__(2) instead of (2).__div__(cos(x)) because 2
(type int) does not know how to be divided by cos(x). (Note: In Python 3, __div__ is replaced
by __truediv__ and __rdiv__ is replaced by __rtruediv__.)
So, you can write normal expressions using python arithmetics like this:
a = Symbol(”a”)
b = Symbol(”b”)
e = (a + b)**2
print e
but from the SymPy point of view, we just need the classes Add, Mul, Pow, Rational, Integer.
For computation, all expressions need to be in a canonical form, this is done during the cre-
ation of the particular instance and only inexpensive operations are performed, necessary to
put the expression in the canonical form. So the canonical form doesn’t mean the simplest
possible expression. The exact list of operations performed depend on the implementation.
Obviously, the definition of the canonical form is arbitrary, the only requirement is that all
equivalent expressions must have the same canonical form. We tried the conversion to a
canonical (standard) form to be as fast as possible and also in a way so that the result is what
you would write by hand - so for example b*a + -4 + b + a*b + 4 + (a + b)**2 becomes
2*a*b + b + (a + b)**2.
Whenever you construct an expression, for example Add(x, x), the Add.__new__() is called
and it determines what to return. In this case:
>>> type(e)
<class 'sympy.core.mul.Mul'>
4.3.3 Comparisons
>>> x + y == y - x
False
4.3.4 Debugging
Starting with 0.6.4, you can turn on/off debug messages with the environment variable
SYMPY_DEBUG, which is expected to have the values True or False. For example, to turn on
debugging, you would issue:
4.3.5 Functionality
There are no given requirements on classes in the library. For example, if they don’t imple-
ment the fdiff() method and you construct an expression using such a class, then trying to
use the Basic.series() method will raise an exception of not finding the fdiff() method
in your class. This “duck typing” has an advantage that you just implement the functionality
which you need.
You can define the cos class like this:
class cos(Function):
pass
and use it like 1 + cos(x), but if you don’t implement the fdiff() method, you will not be
able to call (1 + cos(x)).series().
The symbolic object is characterized (defined) by the things which it can do, so implementing
more methods like fdiff(), subs() etc., you are creating a “shape” of the symbolic object.
Useful things to implement in new classes are: hash() (to use the class in comparisons),
fdiff() (to use it in series expansion), subs() (to use it in expressions, where some parts are
being substituted) and series() (if the series cannot be computed using the general Basic.
series() method). When you create a new class, don’t worry about this too much - just try to
use it in your code and you will realize immediately which methods need to be implemented
in each situation.
All objects in sympy are immutable - in the sense that any operation just returns a new in-
stance (it can return the same instance only if it didn’t change). This is a common mistake to
change the current instance, like self.arg = self.arg + 1 (wrong!). Use arg = self.arg
+ 1; return arg instead. The object is immutable in the sense of the symbolic expression it
represents. It can modify itself to keep track of, for example, its hash. Or it can recalculate
anything regarding the expression it contains. But the expression cannot be changed. So you
can pass any instance to other objects, because you don’t have to worry that it will change,
or that this would break anything.
4.3.6 Conclusion
Above are the main ideas behind SymPy that we try to obey. The rest depends on the current
implementation and may possibly change in the future. The point of all of this is that the
interdependencies inside SymPy should be kept to a minimum. If one wants to add new
functionality to SymPy, all that is necessary is to create a subclass of Basic and implement
what you want.
4.3.7 Functions
class sign(Function):
nargs = 1
@classmethod
def eval(cls, arg):
if isinstance(arg, Basic.NaN):
return S.NaN
if isinstance(arg, Basic.Zero):
return S.Zero
if arg.is_positive:
return S.One
if arg.is_negative:
return S.NegativeOne
if isinstance(arg, Basic.Mul):
coeff, terms = arg.as_coeff_mul()
if not isinstance(coeff, Basic.One):
return cls(coeff) * cls(Basic.Mul(*terms))
is_finite = True
def _eval_conjugate(self):
return self
def _eval_is_zero(self):
return isinstance(self[0], Basic.Zero)
and that’s it. The _eval_* functions are called when something is needed. The eval is
called when the class is about to be instantiated and it should return either some simpli-
fied instance of some other class or if the class should be unmodified, return None (see
core/function.py in Function.__new__ for implementation details). See also tests in
sympy/functions/elementary/tests/test_interface.py that test this interface. You can use them
to create your own new functions.
The applied function sign(x) is constructed using
sign(x)
both inside and outside of SymPy. Unapplied functions sign is just the class itself:
sign
both inside and outside of SymPy. This is the current structure of classes in SymPy:
class BasicType(type):
pass
class MetaBasicMeths(BasicType):
...
class BasicMeths(AssumeMeths):
__metaclass__ = MetaBasicMeths
...
class Basic(BasicMeths):
...
class FunctionClass(MetaBasicMeths):
...
class Function(Basic, RelMeths, ArithMeths):
__metaclass__ = FunctionClass
...
The exact names of the classes and the names of the methods and how they work can be
changed in the future.
This is how to create a function with two variables:
class chebyshevt_root(Function):
nargs = 2
@classmethod
def eval(cls, n, k):
if not 0 <= k < n:
raise ValueError(”must have 0 <= k < n”)
return cos(S.Pi*(2*k + 1)/(2*n))
Note: the first argument of a @classmethod should be cls (i.e. not self).
So guess what this my_function is going to be? Well, it’s derivative is cos and the function
value at 0 is 0, but let’s pretend we don’t know:
>>> from sympy import pprint
>>> pprint(my_function(x).series(x, 0, 10))
3 5 7 9
x x x x / 10\
x - -- + --- - ---- + ------ + O\x /
6 120 5040 362880
Let’s try a more complicated example. Let’s define the derivative in terms of the function
itself:
>>> class what_am_i(Function):
... nargs = 1
...
... def fdiff(self, argindex = 1):
... return 1 - what_am_i(self.args[0])**2
...
... @classmethod
... def eval(cls, arg):
... arg = sympify(arg)
... if arg == 0:
... return sympify(0)
The new functions we just defined are regular SymPy objects, you can use them all over
SymPy, e.g.:
Please use the same way as is shown below all across SymPy.
accessing parameters:
>>> e = sign(x**2)
>>> e.args
(x**2,)
>>> e.args[0]
x**2
Number arguments (in Adds and Muls) will always be the first argument;
other arguments might be in arbitrary order:
>>> (1 + x + y*z).args[0]
1
>>> (1 + x + y*z).args[1] in (x, y*z)
True
>>> (y*z).args
(y, z)
>>> sin(y*z).args
(y*z,)
Never use internal methods or variables, prefixed with “_” (example: don’t use _args, use
.args instead).
testing the structure of a SymPy expression
Applied functions:
>>> f = exp
>>> g = Add
>>> g is Add
True
4.4 Contributing
We welcome every SymPy user to participate in it’s development. Don’t worry if you’ve never
contributed to any open source project, we’ll help you learn anything necessary, just ask on
our mailinglist.
Don’t be afraid to ask anything and don’t worry that you are wasting our time if you are new
to SymPy and ask questions that maybe most of the people know the answer to – you are
not, because that’s exactly what the mailinglist is for and people answer your emails because
they want to. Also we try hard to answer every email, so you’ll always get some feedback and
pointers what to do next.
Go to issues that are sorted by priority and simply find something that you would like to get
fixed and fix it. If you find something odd, please report it into issues first before fixing it.
Feel free to consult with us on the mailinglist. Then send your patch either to the issues or
the mailinglist.
Please read our excellent SymPy Patches Tutorial at our wiki for a guide on how to write
patches to SymPy, how to work with Git, and how to make your life easier as you get started
with SymPy.
Please see the documentation (page 95) how to fix and improve SymPy’s documentation. All
contribution is very welcome.
FIVE
Because every feature of SymPy must have a test case, when you are not sure how to use
something, just look into the tests/ directories, find that feature and read the tests for it,
that will tell you everything you need to know.
Most of the things are already documented though in this document, that is automatically
generated using SymPy’s docstrings.
Click the “modules” (modindex) link in the top right corner to easily access any SymPy module,
or use the list below:
5.1.1 sympify
sympify
Warning: Note that this function uses eval, and thus shouldn’t be used on unsani-
tized input.
If the argument is already a type that SymPy understands, it will do nothing but return
that value. This can be used at the beginning of a function to ensure you are working
with the correct type.
95
SymPy Documentation, Release 1.3
>>> sympify(2).is_integer
True
>>> sympify(2).is_real
True
>>> sympify(2.0).is_real
True
>>> sympify(”2.0”).is_real
True
>>> sympify(”2e-45”).is_real
True
>>> sympify(”x***2”)
Traceback (most recent call last):
...
SympifyError: SympifyError: ”could not parse u'x***2'”
Notes
Locals
The sympification happens with access to everything that is loaded by from sympy im-
port *; anything used in a string that is not defined by that import will be converted
to a symbol. In the following, the bitcount function is treated as a symbol and the O
is interpreted as the Order object (used with series) and it raises an error when used
improperly:
>>> s = 'bitcount(42)'
>>> sympify(s)
bitcount(42)
>>> sympify(”O(x)”)
O(x)
>>> sympify(”O + 1”)
(continues on next page)
In order to have the O interpreted as a Symbol, identify it as such in the namespace dic-
tionary. This can be done in a variety of ways; all three of the following are possibilities:
>>> from sympy import Symbol
>>> ns[”O”] = Symbol(”O”) # method 1
>>> exec_('from sympy.abc import O', ns) # method 2
>>> ns.update(dict(O=Symbol(”O”))) # method 3
>>> sympify(”O + 1”, locals=ns)
O + 1
If you want all single-letter and Greek-letter variables to be symbols then you can use the
clashing-symbols dictionaries that have been defined there as private variables: _clash1
(single-letter variables), _clash2 (the multi-letter Greek names) or _clash (both single
and multi-letter names that are defined in abc).
>>> from sympy.abc import _clash1
>>> _clash1
{'C': C, 'E': E, 'I': I, 'N': N, 'O': O, 'Q': Q, 'S': S}
>>> sympify('I & Q', _clash1)
I & Q
Strict
If the option strict is set to True, only the types for which an explicit conversion has
been defined are converted. In the other cases, a SympifyError is raised.
>>> print(sympify(None))
None
>>> sympify(None, strict=True)
Traceback (most recent call last):
...
SympifyError: SympifyError: None
Evaluation
If the option evaluate is set to False, then arithmetic and operators will be converted
into their SymPy equivalents and the evaluate=False option will be added. Nested Add
or Mul will be denested first. This is done via an AST transformation that replaces oper-
ators with their SymPy equivalents, so if an operand redefines any of those operations,
the redefined operators will not be used.
Extending
To extend sympify to convert custom objects (not derived from Basic), just define a
_sympy_ method to your class. You can do that even to classes that you do not own by
subclassing or adding the method at runtime.
If you do not have control over the class definition you could also use the converter
global dictionary. The key is the class and the value is a function that takes a single
argument and returns the desired SymPy object, e.g. converter[MyList] = lambda x:
Matrix(x).
>>> class MyList2(object): # XXX Do not do this if you control the class!
... def __iter__(self): # Use _sympy_!
... yield 1
... yield 2
... return
... def __getitem__(self, i): return list(self)[i]
>>> from sympy.core.sympify import converter
>>> converter[MyList2] = lambda x: Matrix(x)
>>> sympify(MyList2())
Matrix([
[1],
[2]])
5.1.2 assumptions
When the property cannot be determined (or when a method is not implemented) None will be
returned, e.g. a generic symbol, x, may or may not be positive so a value of None is returned
for x.is_positive.
By default, all symbolic values are in the largest set in the given context without specifying
the property. For example, a symbol that has a property being integer, is also real, complex,
etc.
Here follows a list of possible assumption names:
commutative object commutes with any other object with respect to multiplication opera-
tion.
complex object can have only values from the set of complex numbers.
imaginary object value is a number that can be written as a real number multiplied by the
imaginary unit I. See [R51]. Please note, that 0 is not considered to be an imaginary
number, see issue #7649.
real object can have only values from the set of real numbers.
integer object can have only values from the set of integers.
odd
even object can have only values from the set of odd (even) integers [R50].
prime object is a natural number greater than 1 that has no positive divisors other than 1
and itself. See [R54].
composite object is a positive integer that has at least one positive divisor other than 1 or
the number itself. See [R52].
zero object has the value of 0.
nonzero object is a real number that is not zero.
rational object can have only values from the set of rationals.
algebraic object can have only values from the set of algebraic numbers11 .
transcendental object can have only values from the set of transcendental numbers10 .
irrational object value cannot be represented exactly by Rational, see [R53].
finite
infinite object absolute value is bounded (arbitrarily large). See [R55], [R56], [R57].
negative
nonnegative object can have only negative (nonnegative) values [R49].
positive
11 http://en.wikipedia.org/wiki/Algebraic_number
10
http://en.wikipedia.org/wiki/Transcendental_number
Examples
See Also
See also:
sympy.core.numbers.ImaginaryUnit (page 154) sympy.core.numbers.Zero (page 149)
sympy.core.numbers.One (page 150)
Notes
Assumption values are stored in obj._assumptions dictionary or are returned by getter meth-
ods (with property decorators) or are attributes of objects/classes.
References
5.1.3 cache
cacheit
sympy.core.cache.cacheit(func)
5.1.4 basic
Basic
class sympy.core.basic.Basic
Base class for all objects in SymPy.
Conventions:
1) Always use .args, when accessing parameters of some instance:
>>> cot(x).args
(x,)
>>> cot(x).args[0]
x
>>> (x*y).args
(x, y)
>>> (x*y).args[1]
y
2) Never use internal methods or variables (the ones prefixed with _):
>>> cot(x)._args # do not use this, use cot(x).args instead
(x,)
args
Returns a tuple of arguments of ‘self’.
Notes
Never use self._args, always use self.args. Only use _args in __new__ when creating
a new function. Don’t override .args() from Basic (so that it’s easy to change the
interface in the future if needed).
Examples
>>> cot(x).args
(x,)
>>> cot(x).args[0]
x
>>> (x*y).args
(x, y)
>>> (x*y).args[1]
y
as_content_primitive(radical=False, clear=True)
A stub to allow Basic args (like Tuple) to be skipped when computing the content
and primitive components of an expression.
See also:
sympy.core.expr.Expr.as_content_primitive (page 118)
as_poly(*gens, **args)
Converts self to a polynomial or returns None.
assumptions0
Return object type assumptions.
For example:
Symbol(‘x’, real=True) Symbol(‘x’, integer=True)
are different objects. In other words, besides Python type (Symbol in this case), the
initial assumptions are also forming their typeinfo.
Examples
atoms(*types)
Returns the atoms that form the current object.
By default, only objects that are truly atomic and can’t be divided into smaller pieces
are returned: symbols, numbers, and number symbols like I and pi. It is possible to
request atoms of any type, however, as demonstrated below.
Examples
If one or more types are given, the results will contain only those types of atoms.
>>> from sympy import Number, NumberSymbol, Symbol
>>> (1 + x + 2*sin(y + I*pi)).atoms(Symbol)
{x, y}
Note that I (imaginary unit) and zoo (complex infinity) are special types of number
symbols and are not part of the NumberSymbol class.
The type can be given implicitly, too:
>>> (1 + x + 2*sin(y + I*pi)).atoms(x) # x is a Symbol
{x, y}
Be careful to check your assumptions when using the implicit option since S(1).
is_Integer = True but type(S(1)) is One, a special type of sympy atom, while
type(S(2)) is type Integer and will find all integers in an expression:
>>> from sympy import S
>>> (1 + x + 2*sin(y + I*pi)).atoms(S(1))
{1}
Finally, arguments to atoms() can select more than atomic atoms: any sympy type
(loaded in core/__init__.py) can be listed as an argument and those types of “atoms”
as found in scanning the arguments of the expression recursively:
>>> from sympy import Function, Mul
>>> from sympy.core.function import AppliedUndef
>>> f = Function('f')
>>> (1 + f(x) + 2*sin(y + I*pi)).atoms(Function)
{f(x), sin(y + I*pi)}
>>> (1 + f(x) + 2*sin(y + I*pi)).atoms(AppliedUndef)
{f(x)}
canonical_variables
Return a dictionary mapping any variable defined in self.variables as underscore-
suffixed numbers corresponding to their position in self.variables. Enough un-
derscores are added to ensure that there will be no clash with existing free symbols.
Examples
classmethod class_key()
Nice order of classes.
compare(other)
Return -1, 0, 1 if the object is smaller, equal, or greater than other.
Not in the mathematical sense. If the object is of a different type from the “other”
then their classes are ordered according to the sorted_classes list.
Examples
count(query)
Count the number of matching subexpressions.
count_ops(visual=None)
wrapper for count_ops that returns the operation count.
doit(**hints)
Evaluate objects that are not evaluated by default like limits, integrals, sums and
products. All objects of this kind will be evaluated recursively, unless some species
were excluded via ‘hints’ or unless the ‘deep’ hint was set to ‘False’.
>>> 2*Integral(x, x)
2*Integral(x, x)
dummy_eq(other, symbol=None)
Compare two expressions and handle dummy symbols.
Examples
>>> u = Dummy('u')
find(query, group=False)
Find all subexpressions matching a query.
free_symbols
Return from the atoms of self those which are free symbols.
For most expressions, all symbols are free symbols. For some classes this is not true.
e.g. Integrals use Symbols for the dummy variables which are bound variables, so
Integral has a method to return all symbols except those. Derivative keeps track of
symbols with respect to which it will perform a derivative; those are bound variables,
too, so it has its own free_symbols method.
Any other method that uses bound variables should implement a free_symbols
method.
classmethod fromiter(args, **assumptions)
Create a new object from an iterable.
This is a convenience function that allows one to create objects from any iterable,
without having to convert to a list or tuple first.
Examples
func
The top-level function in an expression.
The following should hold for all objects:
>> x == x.func(*x.args)
Examples
has(*patterns)
Test whether any subexpression matches any of the patterns.
Examples
>>> i.contains(4)
True
>>> i.contains(0)
False
>>> x.has()
False
is_comparable
Return True if self can be computed to a real number (or already is a real number)
with precision, else False.
Examples
A False result does not mean that self cannot be rewritten into a form that would
be comparable. For example, the difference computed below is zero but without
simplification it does not evaluate to a zero with precision:
match(pattern, old=False)
Pattern matching.
Wild symbols match all.
Return None when expression (self) does not match with pattern. Otherwise return
a dictionary such that:
pattern.xreplace(self.match(pattern)) == self
Examples
The old flag will give the old-style pattern matching where expressions and patterns
are essentially solved to give the match. Both of the following give None unless
old=True:
Examples
rcall(*args)
Apply on the argument recursively through the expression tree.
This method is used to simulate a common abuse of notation for operators. For
instance in SymPy the the following will not work:
(x+Lambda(y, 2*y))(z) == x+2*z,
however you can use
>>> from sympy import Lambda
>>> from sympy.abc import x, y, z
>>> (x + Lambda(y, 2*y)).rcall(z)
x + 2*z
Examples
Initial setup
>>> from sympy import log, sin, cos, tan, Wild, Mul, Add
>>> from sympy.abc import x, y
>>> f = log(sin(x)) + tan(sin(x**2))
>>> a = Wild('a')
>>> f.replace(sin(a), tan(a))
log(tan(x)) + tan(tan(x**2))
>>> f.replace(sin(a), tan(a/2))
log(tan(x/2)) + tan(tan(x**2/2))
>>> f.replace(sin(a), a)
log(x) + tan(x**2)
>>> (x*y).replace(a*x, a)
y
When the default value of False is used with patterns that have more than one
Wild symbol, non-intuitive results may be obtained:
>>> b = Wild('b')
>>> (2*x).replace(a*x + b, b - a)
2/x
For this reason, the exact option can be used to make the replacement only
when the match gives non-zero values for all Wild symbols:
>>> g = 2*sin(x**3)
>>> g.replace(lambda expr: expr.is_Number, lambda expr: expr**2)
4*sin(x**9)
The expression itself is also targeted by the query but is done in such a fashion that
changes are not made twice.
>>> e = x*(x*y + 1)
>>> e.replace(lambda x: x.is_Mul, lambda x: 2*x)
2*x*(2*x*y + 1)
rewrite(*args, **hints)
Rewrite functions in terms of other functions.
Rewrites expression containing applications of functions of one kind in terms of
functions of different kind. For example you can rewrite trigonometric functions as
complex exponentials or combinatorial functions as gamma function.
As a pattern this function accepts a list of functions to to rewrite (instances of De-
finedFunction class). As rule you can use string or a destination function instance
(in this case rewrite() will use the str() function).
There is also the possibility to pass hints on how to rewrite the given expressions.
For now there is only one such hint defined called ‘deep’. When ‘deep’ is set to False
it will forbid functions to rewrite their contents.
Examples
Unspecified pattern:
>>> sin(x).rewrite(exp)
-I*(exp(I*x) - exp(-I*x))/2
sort_key(order=None)
Return a sort key.
Examples
subs(*args, **kwargs)
Substitutes old for new in an expression after sympifying args.
args is either:
• two arguments, e.g. foo.subs(old, new)
• one iterable argument, e.g. foo.subs(iterable). The iterable may be
o an iterable container with (old, new) pairs. In this case the
replacements are processed in the order given with successive patterns
possibly affecting replacements already made.
o a dict or set whose key/value items correspond to old/new pairs.
In this case the old/new pairs will be sorted by op count and in case of a
tie, by number of args and the default_sort_key. The resulting sorted list
is then processed as an iterable container (see previous).
If the keyword simultaneous is True, the subexpressions will not be evaluated until
all the substitutions have been made.
See also:
Examples
To replace only the x**2 but not the x**4, use xreplace:
To delay evaluation until all substitutions have been made, set the keyword simul-
taneous to True:
This has the added feature of not allowing subsequent substitutions to affect those
already made:
>>> A = (sqrt(sin(2*x)), a)
>>> B = (sin(2*x), b)
>>> C = (cos(2*x), c)
>>> D = (x, d)
>>> E = (exp(x), e)
The resulting expression represents a literal replacement of the old arguments with
the new arguments. This may not reflect the limiting behavior of the expression:
rather than
as the former will ensure that the desired level of precision is obtained.
xreplace(rule)
Replace occurrences of objects within the expression.
Parameters rule : dict-like
Expresses a replacement rule
Returns xreplace : the result of the replacement
See also:
Examples
xreplace doesn’t differentiate between free and bound symbols. In the following,
subs(x, y) would not change x since it is a bound symbol, but xreplace does:
Atom
class sympy.core.basic.Atom
A parent class for atomic things. An atom is an expression with no subexpressions.
Examples
5.1.5 core
5.1.6 singleton
class sympy.core.singleton.SingletonRegistry
The registry for the singleton classes (accessible as S).
This class serves as two separate things.
The first thing it is is the SingletonRegistry. Several classes in SymPy appear so often
that they are singletonized, that is, using some metaprogramming they are made so
that they can only be instantiated once (see the sympy.core.singleton.Singleton class
for details). For instance, every time you create Integer(0), this will return the same
instance, sympy.core.numbers.Zero (page 149). All singleton instances are attributes
of the S object, so Integer(0) can also be accessed as S.Zero.
Singletonization offers two advantages: it saves memory, and it allows fast comparison.
It saves memory because no matter how many times the singletonized objects appear
in expressions in memory, they all point to the same single instance in memory. The
fast comparison comes from the fact that you can use is to compare exact instances in
Python (usually, you need to use == to compare things). is compares objects by memory
address, and is very fast. For instance
For the most part, the fact that certain objects are singletonized is an implementation
detail that users shouldn’t need to worry about. In SymPy library code, is comparison
is often used for performance purposes The primary advantage of S for end users is the
convenient access to certain instances that are otherwise difficult to type, like S.Half
(instead of Rational(1, 2)).
When using is comparison, make sure the argument is sympified. For instance,
>>> 0 is S.Zero
False
This problem is not an issue when using ==, which is recommended for most use-cases:
>>> 0 == S.Zero
True
5.1.7 expr
5.1.8 Expr
class sympy.core.expr.Expr
Base class for algebraic expressions.
Everything that requires arithmetic operations to be defined should subclass this class,
instead of Basic (which should be used only for argument storage and expression manip-
ulation, i.e. pattern matching, substitutions, etc).
See also:
sympy.core.basic.Basic (page 100)
apart(x=None, **args)
See the apart function in sympy.polys
args_cnc(cset=False, warn=True, split_1=True)
Return [commutative factors, non-commutative factors] of self.
self is treated as a Mul and the ordering of the factors is maintained. If cset is True
the commutative factors will be returned in a set. If there were repeated factors
(as may happen with an unevaluated Mul) then an error will be raised unless it is
explicitly suppressed by setting warn to False.
Note: -1 is always separated from a Number unless split_1 is False.
as_coeff_Add(rational=False)
Efficiently extract the coefficient of a summation.
as_coeff_Mul(rational=False)
Efficiently extract the coefficient of a product.
as_coeff_add(*deps)
Return the tuple (c, args) where self is written as an Add, a.
c should be a Rational added to any terms of the Add that are independent of deps.
args should be a tuple of all other terms of a; args is empty if self is a Number or if
self is independent of deps (when given).
This should be used when you don’t know if self is an Add or not but you want to
treat self as an Add or if you want to process the individual arguments of the tail of
self as an Add.
• if you know self is an Add and want only the head, use self.args[0];
• if you don’t want to process the arguments of the tail but need the tail then use
self.as_two_terms() which gives the head and tail.
• if you want to split self into an independent and dependent parts use self.
as_independent(*deps)
as_coeff_exponent(x)
c*x**e -> c,e where x can be any symbolic expression.
as_coeff_mul(*deps, **kwargs)
Return the tuple (c, args) where self is written as a Mul, m.
c should be a Rational multiplied by any factors of the Mul that are independent of
deps.
args should be a tuple of all other factors of m; args is empty if self is a Number or
if self is independent of deps (when given).
This should be used when you don’t know if self is a Mul or not but you want to treat
self as a Mul or if you want to process the individual arguments of the tail of self as
a Mul.
• if you know self is a Mul and want only the head, use self.args[0];
• if you don’t want to process the arguments of the tail but need the tail then use
self.as_two_terms() which gives the head and tail;
• if you want to split self into an independent and dependent parts use self.
as_independent(*deps)
as_coefficient(expr)
Extracts symbolic coefficient at the given expression. In other words, this functions
separates ‘self’ into the product of ‘expr’ and ‘expr’-free coefficient. If such separa-
tion is not possible it will return None.
See also:
Examples
>>> E.as_coefficient(E)
1
>>> (2*E).as_coefficient(E)
2
>>> (2*sin(E)*E).as_coefficient(E)
Two terms have E in them so a sum is returned. (If one were desiring the coefficient
of the term exactly matching E then the constant from the returned expression could
be selected. Or, for greater precision, a method of Poly can be used to indicate the
desired term from which the coefficient is desired.)
>>> (2*pi*I).as_coefficient(pi*I)
2
>>> (2*I).as_coefficient(pi*I)
as_coefficients_dict()
Return a dictionary mapping terms to their Rational coefficient. Since the dictio-
nary is a defaultdict, inquiries about terms which were not present will return a
coefficient of 0. If an expression is not an Add it is considered to have a single term.
Examples
as_content_primitive(radical=False, clear=True)
This method should recursively remove a Rational from all arguments and return
that (content) and the new self (primitive). The content should always be posi-
tive and Mul(*foo.as_content_primitive()) == foo. The primitive need not be
in canonical form and should try to preserve the underlying structure if possible (i.e.
expand_mul should not be applied to self).
Examples
>>> eq.as_content_primitive()
(2, x + 3*y*(y + 1) + 1)
If clear=False (default is True) then content will not be removed from an Add if it
can be distributed to leave one or more terms with integer coefficients.
as_expr(*gens)
Convert a polynomial to a SymPy expression.
Examples
>>> sin(x).as_expr()
sin(x)
as_independent(*deps, **hint)
A mostly naive separation of a Mul or Add into arguments that are not are dependent
on deps. To obtain as complete a separation of variables as possible, use a separation
method first, e.g.:
• separatevars() to change Mul, Add and Pow (including exp) into Mul
• .expand(mul=True) to change Add or Mul into Add
• .expand(log=True) to change log expr into an Add
The only non-naive thing that is done here is to respect noncommutative ordering
of variables and to always return (0, 0) for self of zero regardless of hints.
For nonzero self , the returned tuple (i, d) has the following interpretation:
• i will has no variable that appears in deps
• d will either have terms that contain variables that are in deps, or be equal to 0
(when self is an Add) or 1 (when self is a Mul)
• if self is an Add then self = i + d
• if self is a Mul then self = i*d
• otherwise (self, S.One) or (S.One, self) is returned.
To force the expression to be treated as an Add, use the hint as_Add=True
See also:
separatevars (page 1187), expand (page 126), Add.as_two_terms, Mul.
as_two_terms, as_coeff_add (page 116), as_coeff_mul (page 116)
Examples
– self is an Add
>>> from sympy import sin, cos, exp
>>> from sympy.abc import x, y, z
>>> (x + x*y).as_independent(x)
(0, x*y + x)
>>> (x + x*y).as_independent(y)
(x, x*y)
>>> (2*x*sin(x) + y + x + z).as_independent(x)
(y + z, 2*x*sin(x) + x)
(continues on next page)
– self is a Mul
>>> (x*sin(x)*cos(y)).as_independent(x)
(cos(y), x*sin(x))
>>> (sin(x)).as_independent(x)
(1, sin(x))
>>> (sin(x)).as_independent(y)
(sin(x), 1)
>>> exp(x+y).as_independent(x)
(1, exp(x + y))
Note how the below differs from the above in making the constant on the dep term
positive.
>>> (y*(-3+x)).as_independent(x)
(y, x - 3)
Note: when trying to get independent terms, a separation method might need to be
used first. In this case, it is important to keep track of what you send to this routine
so you know how to interpret the returned values
as_leading_term(*symbols)
Returns the leading (nonzero) term of the series expansion of self.
The _eval_as_leading_term routines are used to do this, and they must always return
a non-zero value.
Examples
as_numer_denom()
expression -> a/b -> a, b
This is just a stub that should be defined by an object’s class methods to get anything
else.
See also:
as_ordered_factors(order=None)
Return list of ordered factors (if Mul) else [self].
as_ordered_terms(order=None, data=False)
Transform an expression to an ordered list of terms.
Examples
as_powers_dict()
Return self as a dictionary of factors with each factor being treated as a power.
The keys are the bases of the factors and the values, the corresponding exponents.
The resulting dictionary should be used with caution if the expression is a Mul and
contains non- commutative factors since the order that they appeared will be lost in
the dictionary.
as_real_imag(deep=True, **hints)
Performs complex expansion on ‘self’ and returns a tuple containing collected both
real and imaginary parts. This method can’t be confused with re() and im() functions,
which does not perform complex expansion at evaluation.
However it is possible to expand both re() and im() functions and get exactly the
same results as with a single call to this function.
>>> (x + y*I).as_real_imag()
(x, y)
>>> (z + w*I).as_real_imag()
(re(z) - im(w), re(w) + im(z))
as_terms()
Transform an expression to a list of terms.
cancel(*gens, **args)
See the cancel function in sympy.polys
coeff(x, n=1, right=False)
Returns the coefficient from the term(s) containing x**n. If n is zero then all terms
independent of x will be returned.
When x is noncommutative, the coefficient to the left (default) or right of x can be
returned. The keyword ‘right’ is ignored when x is commutative.
See also:
as_coefficient (page 117) separate the expression into a coefficient and factor
as_coeff_Add (page 116) separate the additive constant from an expression
Examples
You can select terms that have an explicit negative in front of them:
>>> (-x + 2*y).coeff(-1)
x
>>> (x - 2*y).coeff(-1)
2*y
You can select terms that have a numerical term in front of them:
>>> (-x - 2*y).coeff(2)
-y
>>> from sympy import sqrt
>>> (x + sqrt(2)*x).coeff(sqrt(2))
x
Examples
count_ops(visual=None)
wrapper for count_ops that returns the operation count.
equals(other, failing_expression=False)
Return True if self == other, False if it doesn’t, or None. If failing_expression is True
then the expression which did not simplify to a 0 will be returned instead of None.
If self is a Number (or complex number) that is not zero, then the result is False.
If self is a number and has not evaluated to zero, evalf will be used to test whether
the expression evaluates to zero. If it does so and the result has significance (i.e.
the precision is either -1, for a Rational result, or is greater than 1) then the evalf
value will be used to return True or False.
expand(deep=True, modulus=None, power_base=True, power_exp=True,
mul=True, log=True, multinomial=True, basic=True, **hints)
Expand an expression using hints.
See the docstring of the expand() function in sympy.core.function for more informa-
tion.
expr_free_symbols
Like free_symbols, but returns the free symbols only if they are contained in an
expression node.
Examples
If the expression is contained in a non-expression object, don’t return the free sym-
bols. Compare:
extract_additively(c)
Return self - c if it’s possible to subtract c from self and make all matching coeffi-
cients move towards zero, else return None.
See also:
extract_multiplicatively (page 127), coeff (page 123), as_coefficient
(page 117)
Examples
extract_branch_factor(allow_half=False)
Try to write self as exp_polar(2*pi*I*n)*z in a nice way. Return (z, n).
>>> exp_polar(I*pi).extract_branch_factor(allow_half=True)
(1, 1/2)
>>> exp_polar(2*I*pi).extract_branch_factor(allow_half=True)
(1, 1)
>>> exp_polar(3*I*pi).extract_branch_factor(allow_half=True)
(1, 3/2)
>>> exp_polar(-I*pi).extract_branch_factor(allow_half=True)
(1, -1/2)
extract_multiplicatively(c)
Return None if it’s not possible to make self in the form c * something in a nice way,
i.e. preserving the properties of arguments of self.
>>> ((x*y)**3).extract_multiplicatively(x**2 * y)
x*y**2
>>> ((x*y)**3).extract_multiplicatively(x**4 * y)
>>> (2*x).extract_multiplicatively(2)
x
>>> (2*x).extract_multiplicatively(3)
factor(*gens, **args)
See the factor() function in sympy.polys.polytools
fourier_series(limits=None)
Compute fourier sine/cosine series of self.
See the docstring of the fourier_series() (page 128) in sympy.series.fourier for
more information.
fps(x=None, x0=0, dir=1, hyper=True, order=4, rational=True, full=False)
Compute formal power power series of self.
See the docstring of the fps() (page 128) function in sympy.series.formal for more
information.
gammasimp()
See the gammasimp function in sympy.simplify
getO()
Returns the additive O(..) symbol if there is one, else None.
getn()
Returns the order of the expression.
The order is determined either from the O(…) term. If there is no O(…) term, it
returns None.
Examples
integrate(*args, **kwargs)
See the integrate function in sympy.integrals
invert(g, *gens, **args)
Return the multiplicative inverse of self mod g where self (and g) may be symbolic
expressions).
See also:
sympy.core.numbers.mod_inverse, sympy.polys.polytools.invert (page 832)
is_algebraic_expr(*syms)
This tests whether a given expression is algebraic or not, in the given symbols, syms.
When syms is not given, all free symbols will be used. The rational function does
not have to be in expanded or in any kind of canonical form.
This function returns False for expressions that are “algebraic expressions” with
symbolic exponents. This is a simple extension to the is_rational_function, including
rational exponentiation.
See also:
is_rational_function (page 132)
References
• http://en.wikipedia.org/wiki/Algebraic_expression
Examples
This function does not attempt any nontrivial simplifications that may result in an
expression that does not appear to be an algebraic expression to become one.
is_constant(*wrt, **flags)
Return True if self is constant, False if not, or None if the constancy could not be
determined conclusively.
If an expression has no free symbols then it is a constant. If there are free symbols
it is possible that the expression is a constant, perhaps (but not necessarily) zero.
To test such expressions, two strategies are tried:
1) numerical evaluation at two random points. If two such evaluations give two dif-
ferent values and the values have a precision greater than 1 then self is not constant.
If the evaluations agree or could not be obtained with any precision, no decision is
made. The numerical testing is done only if wrt is different than the free symbols.
2) differentiation with respect to variables in ‘wrt’ (or all free symbols if omitted) to
see if the expression is constant or not. This will not always lead to an expression
that is zero even though an expression is constant (see added test in test_expr.py).
If all derivatives are zero then self is constant with respect to the given symbols.
If neither evaluation nor differentiation can prove the expression is constant, None
is returned unless two numerical values happened to be the same and the flag fail-
ing_number is True – in that case the numerical value will be returned.
If flag simplify=False is passed, self will not be simplified; the default is True since
self should be simplified before testing.
Examples
>>> (0**x).is_constant()
False
>>> x.is_constant()
False
>>> (x**x).is_constant()
False
>>> one = cos(x)**2 + sin(x)**2
>>> one.is_constant()
True
>>> ((one - 1)**(x + 1)).is_constant() in (True, False) # could be 0 or 1
True
is_number
Returns True if self has no free symbols and no undefined functions (AppliedUndef,
to be precise). It will be faster than if not self.free_symbols, however, since
is_number will fail as soon as it hits a free symbol or undefined function.
See also:
sympy.core.basic.is_comparable
Examples
>>> x.is_number
False
>>> f(1).is_number
False
>>> (2*x).is_number
False
>>> (2 + Integral(2, x)).is_number
False
>>> (2 + Integral(2, (x, 1, 2))).is_number
True
is_polynomial(*syms)
Return True if self is a polynomial in syms and False otherwise.
This checks if self is an exact polynomial in syms. This function returns False for
expressions that are “polynomials” with symbolic exponents. Thus, you should be
able to apply polynomial algorithms to expressions for which this returns True, and
Poly(expr, *syms) should work if and only if expr.is_polynomial(*syms) returns True.
The polynomial does not have to be in expanded form. If no symbols are given, all
free symbols in the expression will be used.
This is not part of the assumptions system. You cannot do Symbol(‘z’, polyno-
mial=True).
Examples
This function does not attempt any nontrivial simplifications that may result in an
expression that does not appear to be a polynomial to become one.
Examples
>>> (x/y).is_rational_function()
True
>>> (x**2).is_rational_function()
True
>>> (x/sin(y)).is_rational_function(y)
False
This function does not attempt any nontrivial simplifications that may result in an
expression that does not appear to be a rational function to become one.
Examples
The advantage of lseries() over nseries() is that many times you are just interested
in the next term in the series (i.e. the first term for example), but you don’t know
how many you should ask for in nseries() using the “n” parameter.
See also nseries().
nseries(x=None, x0=0, n=6, dir=’+’, logx=None)
Wrapper to _eval_nseries if assumptions allow, else to series.
If x is given, x0 is 0, dir=’+’, and self has x, then _eval_nseries is called. This calcu-
lates “n” terms in the innermost expressions and then builds up the final series just
by “cross-multiplying” everything out.
The optional logx parameter can be used to replace any log(x) in the returned series
with a symbolic value to avoid evaluating log(x) at 0. A symbol to use in place of
log(x) should be provided.
Advantage – it’s fast, because we don’t have to determine how many terms we need
to calculate in advance.
Disadvantage – you may end up with less terms than you may have expected, but
the O(x**n) term appended will always be correct and so the result, though perhaps
shorter, will also be correct.
If any of those assumptions is not met, this is treated like a wrapper to series which
will try harder to return the correct number of terms.
See also lseries().
Examples
Handling of the logx parameter — in the following example the expansion fails since
sin does not have an asymptotic expansion at -oo (the limit of log(x) as x approaches
0):
>>> e = sin(log(x))
>>> e.nseries(x, 0, 6)
Traceback (most recent call last):
...
PoleError: ...
...
>>> logx = Symbol('logx')
>>> e.nseries(x, 0, 6, logx=logx)
sin(logx)
In the following example, the expansion works but gives only an Order term unless
the logx parameter is used:
>>> e = x**y
>>> e.nseries(x, 0, 2)
O(log(x)**2)
>>> e.nseries(x, 0, 2, logx=logx)
exp(logx*y)
primitive()
Return the positive Rational that can be extracted non-recursively from every term
of self (i.e., self is treated like an Add). This is like the as_coeff_Mul() method but
primitive always extracts a positive Rational (never a negative or a Float).
Examples
radsimp(**kwargs)
See the radsimp function in sympy.simplify
ratsimp()
See the ratsimp function in sympy.simplify
refine(assumption=True)
See the refine function in sympy.assumptions
removeO()
Removes the additive O(..) symbol if there is one
round(p=0)
Return x rounded to the given decimal place.
If a complex number would results, apply round to the real and imaginary compo-
nents of the number.
Notes
Do not confuse the Python builtin function, round, with the SymPy method of the
same name. The former always returns a float (or raises an error if applied to a
complex value) while the latter returns either a Number or a complex number:
Examples
separate(deep=False, force=False)
See the separate function in sympy.simplify
series(x=None, x0=0, n=6, dir=’+’, logx=None)
Series expansion of “self” around x = x0 yielding either terms of the series one by
one (the lazy series given when n=None), else all the terms at once when n != None.
Returns the series expansion of “self” around the point x = x0 with respect to x up
to O((x - x0)**n, x, x0) (default n is 6).
If x=None and self is univariate, the univariate symbol will be supplied, otherwise
an error will be raised.
>>> term=cos(x).series(n=None)
>>> [next(term) for i in range(2)]
[1, -x**2/2]
For dir=+ (default) the series is calculated from the right and for dir=- the series
from the left. For smooth functions this flag will not alter the results.
>>> abs(x).series(dir=”+”)
x
(continues on next page)
5.1.9 UnevaluatedExpr
class sympy.core.expr.UnevaluatedExpr
Expression that is not evaluated unless released.
Examples
5.1.10 AtomicExpr
class sympy.core.expr.AtomicExpr
A parent class for object which are both atoms and Exprs.
For example: Symbol, Number, Rational, Integer, … But not: Add, Mul, Pow, …
5.1.11 symbol
Symbol
class sympy.core.symbol.Symbol
Assumptions: commutative = True
You can override the default assumptions in the constructor:
as_dummy()
Return a Dummy having the same name and same assumptions as self.
Wild
class sympy.core.symbol.Wild
A Wild symbol matches anything, or anything without whatever is explicitly excluded.
Parameters name : str
Name of the Wild instance.
exclude : iterable, optional
Instances in exclude will not be matched.
properties : iterable of functions, optional
Functions, each taking an expressions as input and returns a bool.
All functions in properties need to return True in order for the Wild
instance to match the expression.
Examples
Tips
When using Wild, be sure to use the exclude keyword to make the pattern more precise.
Without the exclude pattern, you may get matches that are technically correct, but not
what you wanted. For example, using the above without exclude:
This is technically correct, because (2/x)*x + 3*y == 2 + 3*y, but you probably wanted
it to not match at all. The issue is that you really didn’t want a and b to include x and
y, and the exclude parameter lets you specify exactly this. With the exclude parameter,
the pattern will not match.
>>> E = 2*x**3*y*z
>>> a, b = symbols('a b', cls=Wild)
>>> E.match(a*b)
{a_: 2*y*z, b_: x**3}
>>> a = Wild('a', exclude=[x, y])
>>> E.match(a*b)
{a_: z, b_: 2*x**3*y}
>>> a = Wild('a', exclude=[x, y, z])
>>> E.match(a*b)
{a_: 2, b_: x**3*y*z}
Dummy
class sympy.core.symbol.Dummy
Dummy symbols are each unique, even if they have the same name:
If a name is not supplied then a string value of an internal count will be used. This is
useful when a temporary variable is needed and the name of the variable used in the
expression is not important.
symbols
sympy.core.symbol.symbols(names, **args)
Transform strings into instances of Symbol (page 137) class.
symbols() (page 139) function returns a sequence of symbols with names taken from
names argument, which can be a comma or whitespace delimited string, or a sequence
of strings:
>>> x, y, z = symbols('x,y,z')
>>> a, b, c = symbols('a b c')
>>> symbols('x')
x
>>> symbols('x,')
(x,)
>>> symbols('x,y')
(x, y)
>>> symbols(('a', 'b', 'c'))
(a, b, c)
>>> symbols(['a', 'b', 'c'])
[a, b, c]
>>> symbols({'a', 'b', 'c'})
{a, b, c}
If an iterable container is needed for a single symbol, set the seq argument to True or
terminate the symbol name with a comma:
To reduce typing, range syntax is supported to create indexed symbols. Ranges are
indicated by a colon and the type of range is determined by the character to the right of
the colon. If the character is a digit then all contiguous digits to the left are taken as the
nonnegative starting value (or 0 if there is no digit left of the colon) and all contiguous
digits to the right are taken as 1 greater than the ending value:
>>> symbols('x:10')
(x0, x1, x2, x3, x4, x5, x6, x7, x8, x9)
>>> symbols('x5:10')
(x5, x6, x7, x8, x9)
>>> symbols('x5(:2)')
(x50, x51)
>>> symbols('x5:10,y:5')
(x5, x6, x7, x8, x9, y0, y1, y2, y3, y4)
If the character to the right of the colon is a letter, then the single letter to the left (or
‘a’ if there is none) is taken as the start and all characters in the lexicographic range
through the letter to the right are used as the range:
>>> symbols('x:z')
(x, y, z)
>>> symbols('x:c') # null range
(continues on next page)
>>> symbols(':c')
(a, b, c)
>>> symbols('x:2(1:3)')
(x01, x02, x11, x12)
>>> symbols(':3:2') # parsing is from left to right
(00, 01, 10, 11, 20, 21)
Only one pair of parentheses surrounding ranges are removed, so to include parentheses
around ranges, double them. And to include spaces, commas, or colons, escape them
with a backslash:
>>> symbols('x((a:b))')
(x(a), x(b))
>>> symbols(r'x(:1\,:2)') # or r'x((:1)\,(:2))'
(x(0,0), x(0,1))
Despite its name, symbols() (page 139) can create symbol-like objects like instances of
Function or Wild classes. To achieve this, set cls keyword argument to the desired type:
>>> type(_[0])
<class 'sympy.core.function.UndefinedFunction'>
var
sympy.core.symbol.var(names, **args)
Create symbols and inject them into the global namespace.
This calls symbols() (page 139) with the same arguments and puts the results into the
global namespace. It’s recommended not to use var() (page 141) in library code, where
symbols() (page 139) has to be used:
.. rubric:: Examples
>>> var('x')
x
>>> x
x
>>> var('a,ab,abc')
(a, ab, abc)
>>> abc
abc
See symbol() documentation for more details on what kinds of arguments can be passed
to var() (page 141).
5.1.12 numbers
Number
class sympy.core.numbers.Number
Represents atomic numbers in SymPy.
Floating point numbers are represented by the Float class. Rational numbers (of any
size) are represented by the Rational class. Integer numbers (of any size) are repre-
sented by the Integer class. Float and Rational are subclasses of Number; Integer is a
subclass of Rational.
For example, 2/3 is represented as Rational(2, 3) which is a different object from
the floating point number obtained with Python division 2/3. Even for numbers that
are exactly represented in binary, there is a difference between how two forms, such as
Rational(1, 2) and Float(0.5), are used in SymPy. The rational form is to be preferred
in symbolic computations.
Other kinds of numbers, such as algebraic numbers sqrt(2) or complex numbers 3 +
4*I, are not instances of Number class as they are not atomic.
See also:
Float (page 143), Integer (page 147), Rational (page 145)
as_coeff_Add(rational=False)
Efficiently extract the coefficient of a summation.
as_coeff_Mul(rational=False)
Efficiently extract the coefficient of a product.
cofactors(other)
Compute GCD and cofactors of self and other.
gcd(other)
Compute GCD of self and other.
lcm(other)
Compute LCM of self and other.
Float
class sympy.core.numbers.Float
Represent a floating-point number of arbitrary precision.
Notes
Floats are inexact by their nature unless their value is a binary-exact value.
For calculation purposes, evalf needs to be able to change the precision but this will not
increase the accuracy of the inexact value. The following is the most accurate 5-digit
approximation of a value of 0.1 that had only 1 digit of precision:
>>> approx.evalf(5)
0.099609
By contrast, 0.125 is exact in binary (as it is in base 10) and so it can be passed to Float
or evalf to obtain an arbitrary precision with matching accuracy:
>>> Float(exact, 5)
0.12500
>>> exact.evalf(20)
0.12500000000000000000
Trying to make a high-precision Float from a float is not disallowed, but one must keep
in mind that the underlying float (not the apparent decimal value) is being obtained with
high precision. For example, 0.3 does not have a finite binary representation. The closest
rational is the fraction 5404319552844595/2**54. So if you try to obtain a Float of 0.3
to 20 digits of precision you will not see the same thing as 0.3 followed by 19 zeros:
If you want a 20-digit value of the decimal 0.3 (not the floating point approximation of
0.3) you should send the 0.3 as a string. The underlying representation is still binary but
a higher precision than Python’s float is used:
Although you can increase the precision of an existing Float using Float it will not in-
crease the accuracy – the underlying value is not changed:
>>> show(t.evalf(20))
4915/2**14 at prec=70
>>> show(t.evalf(2))
307/2**10 at prec=10
Finally, Floats can be instantiated with an mpf tuple (n, c, p) to produce the number
(-1)**n*c*2**p:
>>> n, c, p = 1, 5, 0
>>> (-1)**n*c*2**p
-5
>>> Float((1, 5, 0))
-5.00000000000000
An actual mpf tuple also contains the number of bits in c as the last element of the tuple:
>>> _._mpf_
(1, 5, 0, 3)
This is not needed for instantiation and is not the same thing as the precision. The mpf
tuple and the precision are two separate quantities that Float tracks.
Examples
Creating Floats from strings (and Python int and long types) will give a minimum pre-
cision of 15 digits, but the precision will automatically increase to capture all digits
entered.
>>> Float(1)
1.00000000000000
>>> Float(10**20)
100000000000000000000.
(continues on next page)
However, floating-point numbers (Python float types) retain only 15 digits of precision:
>>> Float(1e20)
1.00000000000000e+20
>>> Float(1.23456789123456789)
1.23456789123457
>>> Float('1e-3', 3)
0.00100
>>> Float(100, 4)
100.0
Float can automatically count significant figures if a null string is sent for the precision;
space are also allowed in the string. (Auto- counting is only allowed for strings, ints and
longs).
If a number is written in scientific notation, only the digits before the exponent are
considered significant if a decimal appears, otherwise the “e” signifies only how to move
the decimal:
Attributes
is_irrational
is_rational
Rational
class sympy.core.numbers.Rational
Represents rational numbers (p/q) of any size.
See also:
sympify, sympy.simplify.simplify.nsimplify (page 1190)
Examples
>>> Rational(.5)
1/2
>>> Rational(.2)
3602879701896397/18014398509481984
If the simpler representation of the float is desired then consider limiting the denomina-
tor to the desired value or convert the float to a string (which is roughly equivalent to
limiting the denominator to 10**12):
>>> Rational(str(.2))
1/5
>>> Rational(.2).limit_denominator(10**12)
1/5
>>> Rational(”1.23”)
123/100
>>> Rational('1e-2')
1/100
>>> Rational(”.1”)
1/10
>>> Rational('1e-2/3.2')
1/320
The conversion of other types of strings can be handled by the sympify() function, and
conversion of floats to expressions or simple fractions can be handled with nsimplify:
But if the input does not reduce to a literal Rational, an error will be raised:
>>> Rational(pi)
Traceback (most recent call last):
...
TypeError: invalid input: pi
Low-level
>>> r = Rational(3, 4)
>>> r
3/4
>>> r.p
3
>>> r.q
4
Note that p and q return integers (not SymPy Integers) so some care is needed when
using them in expressions:
>>> r.p/r.q
0.75
as_coeff_Add(rational=False)
Efficiently extract the coefficient of a summation.
as_coeff_Mul(rational=False)
Efficiently extract the coefficient of a product.
as_content_primitive(radical=False, clear=True)
Return the tuple (R, self/R) where R is the positive Rational extracted from self.
Examples
Integer
class sympy.core.numbers.Integer
Represents integer numbers of any size.
Examples
If a float or a rational is passed to Integer, the fractional part will be discarded; the effect
is of rounding toward zero.
>>> Integer(3.8)
3
>>> Integer(-3.8)
-3
It is rarely needed to explicitly instantiate an Integer, because Python integers are au-
tomatically converted to Integer when they are used in SymPy expressions.
NumberSymbol
class sympy.core.numbers.NumberSymbol
approximation(number_cls)
Return an interval with number_cls endpoints that contains the value of Number-
Symbol. If not implemented, then return None.
RealNumber
sympy.core.numbers.RealNumber
alias of sympy.core.numbers.Float (page 143)
igcd
sympy.core.numbers.igcd(*args)
Computes nonnegative integer greatest common divisor.
The algorithm is based on the well known Euclid’s algorithm. To improve speed, igcd()
has its own caching mechanism implemented.
Examples
ilcm
sympy.core.numbers.ilcm(*args)
Computes integer least common multiple.
Examples
seterr
sympy.core.numbers.seterr(divide=False)
Should sympy raise an exception on 0/0 or return a nan?
divide == True …. raise an exception divide == False … return nan
Zero
class sympy.core.numbers.Zero
The number zero.
Zero is a singleton, and can be accessed by S.Zero
References
[R58]
Examples
as_coeff_Mul(rational=False)
Efficiently extract the coefficient of a summation.
One
class sympy.core.numbers.One
The number one.
One is a singleton, and can be accessed by S.One.
References
[R59]
Examples
NegativeOne
class sympy.core.numbers.NegativeOne
The number negative one.
NegativeOne is a singleton, and can be accessed by S.NegativeOne.
See also:
One (page 150)
References
[R60]
Examples
Half
class sympy.core.numbers.Half
The rational number 1/2.
Half is a singleton, and can be accessed by S.Half.
References
[R61]
Examples
NaN
class sympy.core.numbers.NaN
Not a Number.
This serves as a place holder for numeric values that are indeterminate. Most opera-
tions on NaN, produce another NaN. Most indeterminate forms, such as 0/0 or oo -
oo` produce NaN. Two exceptions are ``0**0 and oo**0, which all produce 1 (this
is consistent with Python’s float).
NaN is loosely related to floating point nan, which is defined in the IEEE 754 floating
point standard, and corresponds to the Python float('nan'). Differences are noted
below.
NaN is mathematically not equal to anything else, even NaN itself. This explains the
initially counter-intuitive results with Eq and == in the examples below.
NaN is not comparable so inequalities raise a TypeError. This is in constrast with floating
point nan where all inequalities are false.
NaN is a singleton, and can be accessed by S.NaN, or can be imported as nan.
References
[R62]
Examples
Attributes
is_algebraic
is_finite
is_integer
is_negative
is_positive
is_prime
is_rational
is_real
is_transcendental
is_zero
Infinity
class sympy.core.numbers.Infinity
Positive infinite quantity.
In real analysis the symbol ∞ denotes an unbounded limit: x → ∞ means that x grows
without bound.
Infinity is often used not only to define a limit but as a value in the affinely extended
real number system. Points labeled +∞ and −∞ can be added to the topological space of
the real numbers, producing the two-point compactification of the real numbers. Adding
algebraic properties to this gives us the extended real numbers.
Infinity is a singleton, and can be accessed by S.Infinity, or can be imported as oo.
See also:
NegativeInfinity (page 152), NaN (page 151)
References
[R63]
Examples
NegativeInfinity
class sympy.core.numbers.NegativeInfinity
Negative infinite quantity.
ComplexInfinity
class sympy.core.numbers.ComplexInfinity
Complex infinity.
In complex analysis the symbol ∞,
˜ called “complex infinity”, represents a quantity with
infinite magnitude, but undetermined complex phase.
ComplexInfinity is a singleton, and can be accessed by S.ComplexInfinity, or can be
imported as zoo.
See also:
Infinity (page 152)
Examples
Exp1
class sympy.core.numbers.Exp1
The e constant.
The transcendental number e = 2.718281828 . . . is the base of the natural logarithm and
of the exponential function, e = exp(1). Sometimes called Euler’s number or Napier’s
constant.
Exp1 is a singleton, and can be accessed by S.Exp1, or can be imported as E.
References
[R64]
Examples
ImaginaryUnit
class sympy.core.numbers.ImaginaryUnit
√
The imaginary unit, i = −1.
I is a singleton, and can be accessed by S.I, or can be imported as I.
References
[R65]
Examples
Pi
class sympy.core.numbers.Pi
The π constant.
The transcendental number π = 3.141592654 . . . represents the ratio of a circle’s circumfer-
ence to its diameter, the area of the unit circle, the half-period of trigonometric functions,
and many other things in mathematics.
Pi is a singleton, and can be accessed by S.Pi, or can be imported as pi.
References
[R66]
Examples
>>> from sympy import S, pi, oo, sin, exp, integrate, Symbol
>>> S.Pi
pi
>>> pi > 3
True
>>> pi.is_irrational
True
>>> x = Symbol('x')
>>> sin(x + 2*pi)
sin(x)
>>> integrate(exp(-x**2), (x, -oo, oo))
sqrt(pi)
EulerGamma
class sympy.core.numbers.EulerGamma
The Euler-Mascheroni constant.
γ = 0.5772157 . . . (also called Euler’s constant) is a mathematical constant recurring in
analysis and number theory. It is defined as the limiting difference between the harmonic
series and the natural logarithm:
( n )
∑1
γ = lim − ln n
n→∞ k
k=1
References
[R67]
Examples
Attributes
is_irrational
Catalan
class sympy.core.numbers.Catalan
Catalan’s constant.
References
[R68]
Examples
Attributes
is_irrational
GoldenRatio
class sympy.core.numbers.GoldenRatio
The golden ratio, ϕ.
√
ϕ = 1+2 5 is algebraic number. Two quantities are in the golden ratio if their ratio is the
same as the ratio of their sum to the larger of the two quantities, i.e. their maximum.
GoldenRatio is a singleton, and can be accessed by S.GoldenRatio.
References
[R69]
Examples
TribonacciConstant
class sympy.core.numbers.TribonacciConstant
The tribonacci constant.
The tribonacci numbers are like the Fibonacci numbers, but instead of starting with two
predetermined terms, the sequence starts with three predetermined terms and each
term afterwards is the sum of the preceding three terms.
The tribonacci constant is the ratio toward which adjacent tribonacci numbers tend. It
is a root of the polynomial x3 − x2 − x − 1 = 0, and also satisfies the equation x + x−3 = 2.
TribonacciConstant is a singleton, and can be accessed by S.TribonacciConstant.
References
[R70]
Examples
5.1.13 power
Pow
class sympy.core.power.Pow
Defines the expression x**y as “x raised to a power y”
Singleton definitions involving (0, 1, -1, oo, -oo, I, -I):
Because symbolic computations are more flexible that floating point calculations and we
prefer to never return an incorrect answer, we choose not to conform to all IEEE 754
conventions. This helps us avoid extra test-case code in the calculation of limits.
See also:
sympy.core.numbers.Infinity (page 152), sympy.core.numbers.NegativeInfinity
(page 152), sympy.core.numbers.NaN (page 151)
References
If base is 1/Integer, then return Integer, -exp. If this extra processing is not needed,
the base and exp properties will give the raw arguments
Examples
as_content_primitive(radical=False, clear=True)
Return the tuple (R, self/R) where R is the positive Rational extracted from self.
Examples
>>> eq = (2 + 2*x)**y
>>> s = expand_power_base(eq); s.is_Mul, s
(False, (2*x + 2)**y)
>>> eq.as_content_primitive()
(1, (2*(x + 1))**y)
>>> s = expand_power_base(_[1]); s.is_Mul, s
(True, 2**y*(x + 1)**y)
integer_nthroot
sympy.core.power.integer_nthroot(y, n)
Return a tuple containing x = floor(y**(1/n)) and a boolean indicating whether the result
is exact (that is, whether x**n == y).
See also:
sympy.ntheory.primetest.is_square, integer_log
Examples
5.1.14 mul
Mul
class sympy.core.mul.Mul
as_coeff_Mul(rational=False)
Efficiently extract the coefficient of a product.
as_coefficients_dict()
Return a dictionary mapping terms to their coefficient. Since the dictionary is a
defaultdict, inquiries about terms which were not present will return a coefficient
of 0. The dictionary is considered to have a single term.
Examples
as_content_primitive(radical=False, clear=True)
Return the tuple (R, self/R) where R is the positive Rational extracted from self.
Examples
Examples
>>> (2*x*y*sin(x)*cos(x)).as_ordered_factors()
[2, x, y, sin(x), cos(x)]
as_two_terms()
Return head and tail of self.
This is the most efficient way to get the head and tail of an expression.
• if you want only the head, use self.args[0];
• if you want to process the arguments of the tail then use self.as_coef_mul() which
gives the head and a tuple containing the arguments of the tail when treated as
a Mul.
• if you want the coefficient when self is treated as an Add then use
self.as_coeff_add()[0]
classmethod flatten(seq)
Return commutative, noncommutative and order arguments by combining related
terms.
Notes
Powers with compound bases may not find a single base to combine with
unless all arguments are processed at once. Post-processing may be nec-
essary in such cases. {c.f. https://github.com/sympy/sympy/issues/5728}
>>> a = sqrt(x*sqrt(y))
>>> a**3
(x*sqrt(y))**(3/2)
>>> Mul(a,a,a)
(x*sqrt(y))**(3/2)
>>> a*a*a
x*sqrt(y)*sqrt(x*sqrt(y))
>>> _.subs(a.base, z).subs(z, a.base)
(x*sqrt(y))**(3/2)
– If more than two terms are being multiplied then all the previous terms will
be re-processed for each new argument. So if each of a, b and c were Mul
(page 160) expression, then a*b*c (or building up the product with *=) will
process all the arguments of a and b twice: once when a*b is computed and
again when c is multiplied.
Using Mul(a, b, c) will process all arguments once.
• The results of Mul are cached according to arguments, so flatten will only be
called once for Mul(a, b, c). If you can structure a calculation so the argu-
ments are most likely to be repeats then this can save time in computing the
answer. For example, say you had a Mul, M, that you wished to divide by d[i]
and multiply by n[i] and you suspect there are many repeats in n. It would be
better to compute M*n[i]/d[i] rather than M/d[i]*n[i] since every time n[i]
is a repeat, the product, M*n[i] will be returned without flattening – the cached
value will be returned. If you divide by the d[i] first (and those are more unique
than the n[i]) then that will create a new Mul, M/d[i] the args of which will be
traversed again when it is multiplied by n[i].
{c.f. https://github.com/sympy/sympy/issues/5706}
This consideration is moot if the cache is turned off.
Nb
The validity of the above notes depends on the implementation details of Mul and
flatten which may change at any time. Therefore, you should only consider them
when your code is highly performance sensitive.
Removal of 1 from the sequence is already handled by AssocOp.__new__.
prod
sympy.core.mul.prod(a, start=1)
Return product of elements of a. Start with int 1 so if only ints are included then
an int result is returned.
Examples
5.1.15 add
Add
class sympy.core.add.Add
as_coeff_Add(rational=False)
Efficiently extract the coefficient of a summation.
as_coeff_add(*deps)
Returns a tuple (coeff, args) where self is treated as an Add and coeff is the Number
term and args is a tuple of all other terms.
Examples
as_coefficients_dict()
Return a dictionary mapping terms to their Rational coefficient. Since the dictio-
nary is a defaultdict, inquiries about terms which were not present will return a
coefficient of 0. If an expression is not an Add it is considered to have a single term.
Examples
as_content_primitive(radical=False, clear=True)
Return the tuple (R, self/R) where R is the positive Rational extracted from self. If
radical is True (default is False) then common radicals will be removed and included
as a factor of the primitive expression.
Examples
Examples
as_two_terms()
Return head and tail of self.
This is the most efficient way to get the head and tail of an expression.
• if you want only the head, use self.args[0];
• if you want to process the arguments of the tail then use self.as_coef_add() which
gives the head and a tuple containing the arguments of the tail when treated as
an Add.
• if you want the coefficient when self is treated as a Mul then use
self.as_coeff_mul()[0]
classmethod class_key()
Nice order of classes
extract_leading_order(symbols, point=None)
Returns the leading term and its order.
Examples
classmethod flatten(seq)
Takes the sequence “seq” of nested Adds and returns a flatten list.
Returns: (commutative_part, noncommutative_part, order_symbols)
Applies associativity, all terms are commutable with respect to addition.
NB: the removal of 0 is already handled by AssocOp.__new__
See also:
sympy.core.mul.Mul.flatten (page 161)
primitive()
Return (R, self/R) where R` is the Rational GCD of self`.
R is collected only from the leading coefficient of each term.
Examples
5.1.16 mod
Mod
class sympy.core.mod.Mod
Represents a modulo operation on symbolic expressions.
Receives two arguments, dividend p and divisor q.
The convention used is the same as Python’s: the remainder always has the same sign
as the divisor.
Examples
5.1.17 relational
Rel
sympy.core.relational.Rel
alias of sympy.core.relational.Relational
Eq
sympy.core.relational.Eq
alias of sympy.core.relational.Equality (page 167)
Ne
sympy.core.relational.Ne
alias of sympy.core.relational.Unequality (page 174)
Lt
sympy.core.relational.Lt
alias of sympy.core.relational.StrictLessThan (page 178)
Le
sympy.core.relational.Le
alias of sympy.core.relational.LessThan (page 171)
Gt
sympy.core.relational.Gt
alias of sympy.core.relational.StrictGreaterThan (page 175)
Ge
sympy.core.relational.Ge
alias of sympy.core.relational.GreaterThan (page 168)
Equality
class sympy.core.relational.Equality
An equal relation between two objects.
Represents that two objects are equal. If they can be easily shown to be definitively equal
(or unequal), this will reduce to True (or False). Otherwise, the relation is maintained
as an unevaluated Equality object. Use the simplify function on this object for more
nontrivial evaluation of the equality relation.
As usual, the keyword argument evaluate=False can be used to prevent any evaluation.
See also:
Notes
This class is not the same as the == operator. The == operator tests for exact structural
equality between two expressions; this class compares expressions mathematically.
If either object defines an e valE q method, it can be used in place of the default algorithm.
If lhs.e valE q(rhs) or rhs.e valE q(lhs) returns anything other than None, that return value
will be substituted for the Equality. If None is returned by e valE q, an Equality object will
be created as usual.
Examples
GreaterThan
class sympy.core.relational.GreaterThan
Class representations of inequalities.
The *Than classes represent inequal relationships, where the left-hand side is generally
bigger or smaller than the right-hand side. For example, the GreaterThan class repre-
sents an inequal relationship where the left-hand side is at least as big as the right side,
if not bigger. In mathematical notation:
lhs >= rhs
In total, there are four *Than classes, to represent the four inequalities:
In addition to the normal .lhs and .rhs of Relations, *Than inequality objects also have
the .lts and .gts properties, which represent the “less than side” and “greater than side”
of the operator. Use of .lts and .gts in an algorithm rather than .lhs and .rhs as an
assumption of inequality direction will make more explicit the intent of a certain section
of code, and will make it similarly more robust to client code changes:
>>> from sympy import GreaterThan, StrictGreaterThan
>>> from sympy import LessThan, StrictLessThan
>>> from sympy import And, Ge, Gt, Le, Lt, Rel, S
>>> from sympy.abc import x, y, z
>>> from sympy.core.relational import Relational
>>> e = GreaterThan(x, 1)
>>> e
x >= 1
(continues on next page)
Notes
Due to the order that Python parses a statement, it may not immediately find
two objects comparable. When “1 < x” is evaluated, Python recognizes that the
number 1 is a native number and that x is not. Because a native Python number
does not know how to compare itself with a SymPy object Python will try the
reflective operation, “x > 1” and that is the form that gets evaluated, hence
returned.
If the order of the statement is important (for visual output to the console, per-
haps), one can work around this annoyance in a couple ways:
(1) “sympify” the literal before comparison
>>> S(1) < x
1 < x
(2) use one of the wrappers or less succinct methods described above
>>> Lt(1, x)
1 < x
>>> Relational(1, x, ”<”)
1 < x
The second gotcha involves writing equality tests between relationals when one or both
sides of the test involve a literal relational:
>>> e = x < 1; e
x < 1
>>> e == e # neither side is a literal
True
>>> e == x < 1 # expecting True, too
False
>>> e != x < 1 # expecting False
x < 1
>>> x < 1 != x < 1 # expecting False or the same thing as before
Traceback (most recent call last):
...
TypeError: cannot determine truth value of Relational
The third gotcha involves chained inequalities not involving ‘==’ or ‘!=’. Occasionally,
one may be tempted to write:
Although this can also be done with the ‘&’ operator, it cannot be done with the
‘and’ operarator:
Examples
One generally does not instantiate these classes directly, but uses various convenience
methods:
Another option is to use the Python inequality operators (>=, >, <=, <) directly. Their
main advantage over the Ge, Gt, Le, and Lt counterparts, is that one can write a more
“mathematical looking” statement rather than littering the math with oddball function
calls. However there are certain (minor) caveats of which to be aware (search for
‘gotcha’, below).
>>> x >= 2
x >= 2
>>> _ == Ge(x, 2)
True
However, it is also perfectly valid to instantiate a *Than class less succinctly and less
conveniently:
>>> StrictGreaterThan(x, 1)
x > 1
>>> GreaterThan(x, 1)
x >= 1
>>> LessThan(x, 1)
x <= 1
>>> StrictLessThan(x, 1)
x < 1
LessThan
class sympy.core.relational.LessThan
Class representations of inequalities.
The *Than classes represent inequal relationships, where the left-hand side is generally
bigger or smaller than the right-hand side. For example, the GreaterThan class repre-
sents an inequal relationship where the left-hand side is at least as big as the right side,
if not bigger. In mathematical notation:
lhs >= rhs
In total, there are four *Than classes, to represent the four inequalities:
In addition to the normal .lhs and .rhs of Relations, *Than inequality objects also have
the .lts and .gts properties, which represent the “less than side” and “greater than side”
of the operator. Use of .lts and .gts in an algorithm rather than .lhs and .rhs as an
assumption of inequality direction will make more explicit the intent of a certain section
of code, and will make it similarly more robust to client code changes:
>>> e = GreaterThan(x, 1)
>>> e
x >= 1
>>> '%s >= %s is the same as %s <= %s' % (e.gts, e.lts, e.lts, e.gts)
'x >= 1 is the same as 1 <= x'
Notes
>>> 1 < x
x > 1
Due to the order that Python parses a statement, it may not immediately find
two objects comparable. When “1 < x” is evaluated, Python recognizes that the
number 1 is a native number and that x is not. Because a native Python number
does not know how to compare itself with a SymPy object Python will try the
reflective operation, “x > 1” and that is the form that gets evaluated, hence
returned.
If the order of the statement is important (for visual output to the console, per-
haps), one can work around this annoyance in a couple ways:
(1) “sympify” the literal before comparison
(2) use one of the wrappers or less succinct methods described above
>>> Lt(1, x)
1 < x
>>> Relational(1, x, ”<”)
1 < x
The second gotcha involves writing equality tests between relationals when one or both
sides of the test involve a literal relational:
>>> e = x < 1; e
x < 1
>>> e == e # neither side is a literal
True
>>> e == x < 1 # expecting True, too
False
>>> e != x < 1 # expecting False
x < 1
>>> x < 1 != x < 1 # expecting False or the same thing as before
Traceback (most recent call last):
(continues on next page)
>>> e == (x < 1)
True
>>> e != (x < 1)
False
>>> (x < 1) != (x < 1)
False
The third gotcha involves chained inequalities not involving ‘==’ or ‘!=’. Occasionally,
one may be tempted to write:
Although this can also be done with the ‘&’ operator, it cannot be done with the
‘and’ operarator:
Examples
One generally does not instantiate these classes directly, but uses various convenience
methods:
Another option is to use the Python inequality operators (>=, >, <=, <) directly. Their
main advantage over the Ge, Gt, Le, and Lt counterparts, is that one can write a more
“mathematical looking” statement rather than littering the math with oddball function
calls. However there are certain (minor) caveats of which to be aware (search for
‘gotcha’, below).
>>> x >= 2
x >= 2
>>> _ == Ge(x, 2)
True
However, it is also perfectly valid to instantiate a *Than class less succinctly and less
conveniently:
>>> StrictGreaterThan(x, 1)
x > 1
>>> GreaterThan(x, 1)
x >= 1
>>> LessThan(x, 1)
x <= 1
>>> StrictLessThan(x, 1)
x < 1
Unequality
class sympy.core.relational.Unequality
An unequal relation between two objects.
Represents that two objects are not equal. If they can be shown to be definitively equal,
this will reduce to False; if definitively unequal, this will reduce to True. Otherwise, the
relation is maintained as an Unequality object.
See also:
Equality (page 167)
Notes
This class is not the same as the != operator. The != operator tests for exact structural
equality between two expressions; this class compares expressions mathematically.
This class is effectively the inverse of Equality. As such, it uses the same algorithms,
including any available e valE q methods.
Examples
StrictGreaterThan
class sympy.core.relational.StrictGreaterThan
Class representations of inequalities.
The *Than classes represent inequal relationships, where the left-hand side is generally
bigger or smaller than the right-hand side. For example, the GreaterThan class repre-
sents an inequal relationship where the left-hand side is at least as big as the right side,
if not bigger. In mathematical notation:
lhs >= rhs
In total, there are four *Than classes, to represent the four inequalities:
In addition to the normal .lhs and .rhs of Relations, *Than inequality objects also have
the .lts and .gts properties, which represent the “less than side” and “greater than side”
of the operator. Use of .lts and .gts in an algorithm rather than .lhs and .rhs as an
assumption of inequality direction will make more explicit the intent of a certain section
of code, and will make it similarly more robust to client code changes:
>>> from sympy import GreaterThan, StrictGreaterThan
>>> from sympy import LessThan, StrictLessThan
>>> from sympy import And, Ge, Gt, Le, Lt, Rel, S
>>> from sympy.abc import x, y, z
>>> from sympy.core.relational import Relational
>>> e = GreaterThan(x, 1)
>>> e
x >= 1
>>> '%s >= %s is the same as %s <= %s' % (e.gts, e.lts, e.lts, e.gts)
'x >= 1 is the same as 1 <= x'
Notes
Due to the order that Python parses a statement, it may not immediately find
two objects comparable. When “1 < x” is evaluated, Python recognizes that the
number 1 is a native number and that x is not. Because a native Python number
does not know how to compare itself with a SymPy object Python will try the
reflective operation, “x > 1” and that is the form that gets evaluated, hence
returned.
If the order of the statement is important (for visual output to the console, per-
haps), one can work around this annoyance in a couple ways:
(1) “sympify” the literal before comparison
(2) use one of the wrappers or less succinct methods described above
>>> Lt(1, x)
1 < x
>>> Relational(1, x, ”<”)
1 < x
The second gotcha involves writing equality tests between relationals when one or both
sides of the test involve a literal relational:
>>> e = x < 1; e
x < 1
>>> e == e # neither side is a literal
True
>>> e == x < 1 # expecting True, too
False
>>> e != x < 1 # expecting False
x < 1
>>> x < 1 != x < 1 # expecting False or the same thing as before
Traceback (most recent call last):
...
TypeError: cannot determine truth value of Relational
>>> e == (x < 1)
True
>>> e != (x < 1)
False
>>> (x < 1) != (x < 1)
False
The third gotcha involves chained inequalities not involving ‘==’ or ‘!=’. Occasionally,
one may be tempted to write:
Although this can also be done with the ‘&’ operator, it cannot be done with the
‘and’ operarator:
>>> (x < y) & (y < z)
(x < y) & (y < z)
>>> (x < y) and (y < z)
Traceback (most recent call last):
...
TypeError: cannot determine truth value of Relational
Examples
One generally does not instantiate these classes directly, but uses various convenience
methods:
>>> for f in [Ge, Gt, Le, Lt]: # convenience wrappers
... print(f(x, 2))
x >= 2
x > 2
x <= 2
x < 2
Another option is to use the Python inequality operators (>=, >, <=, <) directly. Their
main advantage over the Ge, Gt, Le, and Lt counterparts, is that one can write a more
“mathematical looking” statement rather than littering the math with oddball function
calls. However there are certain (minor) caveats of which to be aware (search for
‘gotcha’, below).
>>> x >= 2
x >= 2
>>> _ == Ge(x, 2)
True
However, it is also perfectly valid to instantiate a *Than class less succinctly and less
conveniently:
>>> Rel(x, 1, ”>”)
x > 1
>>> Relational(x, 1, ”>”)
x > 1
>>> StrictGreaterThan(x, 1)
x > 1
>>> GreaterThan(x, 1)
x >= 1
>>> LessThan(x, 1)
x <= 1
>>> StrictLessThan(x, 1)
x < 1
StrictLessThan
class sympy.core.relational.StrictLessThan
Class representations of inequalities.
The *Than classes represent inequal relationships, where the left-hand side is generally
bigger or smaller than the right-hand side. For example, the GreaterThan class repre-
sents an inequal relationship where the left-hand side is at least as big as the right side,
if not bigger. In mathematical notation:
lhs >= rhs
In total, there are four *Than classes, to represent the four inequalities:
In addition to the normal .lhs and .rhs of Relations, *Than inequality objects also have
the .lts and .gts properties, which represent the “less than side” and “greater than side”
of the operator. Use of .lts and .gts in an algorithm rather than .lhs and .rhs as an
assumption of inequality direction will make more explicit the intent of a certain section
of code, and will make it similarly more robust to client code changes:
>>> from sympy import GreaterThan, StrictGreaterThan
>>> from sympy import LessThan, StrictLessThan
>>> from sympy import And, Ge, Gt, Le, Lt, Rel, S
>>> from sympy.abc import x, y, z
>>> from sympy.core.relational import Relational
>>> e = GreaterThan(x, 1)
>>> e
x >= 1
>>> '%s >= %s is the same as %s <= %s' % (e.gts, e.lts, e.lts, e.gts)
'x >= 1 is the same as 1 <= x'
Notes
Due to the order that Python parses a statement, it may not immediately find
two objects comparable. When “1 < x” is evaluated, Python recognizes that the
number 1 is a native number and that x is not. Because a native Python number
does not know how to compare itself with a SymPy object Python will try the
reflective operation, “x > 1” and that is the form that gets evaluated, hence
returned.
If the order of the statement is important (for visual output to the console, per-
haps), one can work around this annoyance in a couple ways:
(1) “sympify” the literal before comparison
(2) use one of the wrappers or less succinct methods described above
>>> Lt(1, x)
1 < x
>>> Relational(1, x, ”<”)
1 < x
The second gotcha involves writing equality tests between relationals when one or both
sides of the test involve a literal relational:
>>> e = x < 1; e
x < 1
>>> e == e # neither side is a literal
True
>>> e == x < 1 # expecting True, too
False
>>> e != x < 1 # expecting False
x < 1
>>> x < 1 != x < 1 # expecting False or the same thing as before
Traceback (most recent call last):
...
TypeError: cannot determine truth value of Relational
>>> e == (x < 1)
True
>>> e != (x < 1)
False
>>> (x < 1) != (x < 1)
False
The third gotcha involves chained inequalities not involving ‘==’ or ‘!=’. Occasionally,
one may be tempted to write:
Although this can also be done with the ‘&’ operator, it cannot be done with the
‘and’ operarator:
>>> (x < y) & (y < z)
(x < y) & (y < z)
>>> (x < y) and (y < z)
Traceback (most recent call last):
...
TypeError: cannot determine truth value of Relational
Examples
One generally does not instantiate these classes directly, but uses various convenience
methods:
>>> for f in [Ge, Gt, Le, Lt]: # convenience wrappers
... print(f(x, 2))
x >= 2
x > 2
x <= 2
x < 2
Another option is to use the Python inequality operators (>=, >, <=, <) directly. Their
main advantage over the Ge, Gt, Le, and Lt counterparts, is that one can write a more
“mathematical looking” statement rather than littering the math with oddball function
calls. However there are certain (minor) caveats of which to be aware (search for
‘gotcha’, below).
>>> x >= 2
x >= 2
>>> _ == Ge(x, 2)
True
However, it is also perfectly valid to instantiate a *Than class less succinctly and less
conveniently:
>>> Rel(x, 1, ”>”)
x > 1
>>> Relational(x, 1, ”>”)
x > 1
>>> StrictGreaterThan(x, 1)
x > 1
>>> GreaterThan(x, 1)
x >= 1
>>> LessThan(x, 1)
x <= 1
>>> StrictLessThan(x, 1)
x < 1
5.1.18 multidimensional
vectorize
class sympy.core.multidimensional.vectorize(*mdargs)
Generalizes a function taking scalars to accept multidimensional arguments.
For example
>>> @vectorize(0)
... def vsin(x):
... return sin(x)
>>> @vectorize(0, 1)
... def vdiff(f, y):
... return diff(f, y)
5.1.19 function
Lambda
class sympy.core.function.Lambda
Lambda(x, expr) represents a lambda function similar to Python’s ‘lambda x: expr’. A
function of several variables is written as Lambda((x, y, …), expr).
A simple example:
>>> p = x, y, z
>>> f = Lambda(p, x + y*z)
>>> f(*p)
x + y*z
expr
The return value of the function
is_identity
Return True if this Lambda is an identity function.
variables
The variables used in the internal representation of the function
WildFunction
Examples
To match functions with a given number of arguments, set nargs to the desired value at
instantiation:
To match functions with a range of arguments, set nargs to a tuple containing the desired
number of arguments, e.g. if nargs = (1, 2) then functions with 1 or 2 arguments will
be matched.
Derivative
class sympy.core.function.Derivative
Carries out differentiation of the given expression with respect to symbols.
expr must define ._eval_derivative(symbol) method that returns the differentiation result.
This function only needs to consider the non-trivial case where expr contains symbol and
it should call the diff() method internally (not _eval_derivative); Derivative should be the
only one to call _eval_derivative.
Simplification of high-order derivatives:
Because there can be a significant amount of simplification that can be done when mul-
tiple differentiations are performed, results will be automatically simplified in a fairly
conservative fashion unless the keyword simplify is set to False.
Ordering of variables:
If evaluate is set to True and the expression can not be evaluated, the list of differen-
tiation symbols will be sorted, that is, the expression is assumed to have continuous
derivatives up to the order asked. This sorting assumes that derivatives wrt Symbols
commute, derivatives wrt non-Symbols commute, but Symbol and non-Symbol deriva-
tives don’t commute with each other.
Derivative wrt non-Symbols:
This class also allows derivatives wrt non-Symbols that have _diff_wrt set to True, such
as Function and Derivative. When a derivative wrt a non- Symbol is attempted, the non-
Symbol is temporarily converted to a Symbol while the differentiation is performed.
Note that this may seem strange, that Derivative allows things like f(g(x)).diff(g(x)), or
even f(cos(x)).diff(cos(x)). The motivation for allowing this syntax is to make it easier
to work with variational calculus (i.e., the Euler-Lagrange method). The best way to
understand this is that the action of derivative with respect to a non-Symbol is defined
by the above description: the object is substituted for a Symbol and the derivative is
taken with respect to that. This action is only allowed for objects for which this can be
done unambiguously, for example Function and Derivative objects. Note that this leads
to what may appear to be mathematically inconsistent results. For example:
This appears wrong because in fact 2*cos(x) and 2*sqrt(1 - sin(x)**2) are identically
equal. However this is the wrong way to think of this. Think of it instead as if we have
something like this:
Here, the Symbols c and s act just like the functions cos(x) and sin(x), respectively.
Think of 2*cos(x) as f(c).subs(c, cos(x)) (or f(c) at c = cos(x)) and 2*sqrt(1 - sin(x)**2)
as g(s).subs(s, sin(x)) (or g(s) at s = sin(x)), where f(u) == 2*u and g(u) == 2*sqrt(1
- u**2). Here, we define the function first and evaluate it at the function, but we can
actually unambiguously do this in reverse in SymPy, because expr.subs(Function, Sym-
bol) is well-defined: just structurally replace the function everywhere it appears in the
expression.
This is the same notational convenience used in the Euler-Lagrange method when one
says F(t, f(t), f’(t)).diff(f(t)). What is actually meant is that the expression in question is
represented by some F(t, u, v) at u = f(t) and v = f’(t), and F(t, f(t), f’(t)).diff(f(t)) simply
means F(t, u, v).diff(u) at u = f(t).
We do not allow derivatives to be taken with respect to expressions where this is not
so well defined. For example, we do not allow expr.diff(x*y) because there are multiple
ways of structurally defining where x*y appears in an expression, some of which may
surprise the reader (for example, a very strict definition would have that (x*y*z).diff(x*y)
== 0).
Note that this definition also fits in nicely with the definition of the chain rule. Note how
Finally, note that, to be consistent with variational calculus, and to ensure that the defini-
tion of substituting a Function for a Symbol in an expression is well-defined, derivatives
of functions are assumed to not be related to the function. In other words, we have:
Note, any class can allow derivatives to be taken with respect to itself. See the docstring
of Expr._diff_wrt.
Examples
Examples
The default step size and number of points are 1 and order + 1 respectively. We
can change the step size by passing a symbol as a parameter:
>>> f(x).diff(x).as_finite_difference(h)
-f(-h/2 + x)/h + f(h/2 + x)/h
The algorithm is not restricted to use equidistant spacing, nor do we need to make
the approximation around x0, but we can get an expression estimating the derivative
at an offset:
>>> y = Symbol('y')
>>> d2fdxdy=f(x,y).diff(x,y)
>>> d2fdxdy.as_finite_difference(wrt=x)
-Derivative(f(x - 1/2, y), y) + Derivative(f(x + 1/2, y), y)
doit_numerically(z0)
Evaluate the derivative at z numerically.
When we can represent derivatives at a point, this should be folded into the normal
evalf. For now, we need a special method.
diff
References
http://reference.wolfram.com/legacy/v5_2/Built-inFunctions/AlgebraicComputation/
Calculus/D.html
Examples
>>> diff(sin(x), x)
cos(x)
>>> diff(f(x), x, x, x)
Derivative(f(x), (x, 3))
>>> diff(f(x), x, 3)
(continues on next page)
>>> diff(sin(x))
cos(x)
>>> diff(sin(x*y))
Traceback (most recent call last):
...
ValueError: specify differentiation variables to differentiate sin(x*y)
Note that diff(sin(x)) syntax is meant only for convenience in interactive sessions and
should be avoided in library code.
FunctionClass
Examples
If the function can take any number of arguments, the set of whole numbers is re-
turned:
>>> Function('f').nargs
Naturals0
If the function was initialized to accept one or more arguments, a corresponding set
will be returned:
The undefined function, after application, also has the nargs attribute; the actual
number of arguments is always available by checking the args attribute:
>>> f = Function('f')
>>> f(1).nargs
Naturals0
>>> len(f(1).args)
1
Function
class sympy.core.function.Function
Base class for applied mathematical functions.
It also serves as a constructor for undefined function classes.
Examples
First example shows how to use Function as a constructor for undefined function classes:
Note that assumptions on a function are unrelated to the assumptions on the variable it is
called on. If you want to add a relationship, subclass Function and define the appropriate
_eval_is_assumption methods.
In the following example Function is used as a base class for my_func that represents a
mathematical function my_func. Suppose that it is well known, that my_func(0) is 1 and
my_func at infinity goes to 0, so we want those two simplifications to occur automatically.
Suppose also that my_func(x) is real exactly when x is real. Here is an implementation
that honours those requirements:
In order for my_func to become useful, several other methods would need to be im-
plemented. See source code of some of the already implemented functions for more
complete examples.
Also, if the function can take more than one argument, then nargs must be defined, e.g.
if my_func can take one or two arguments then,
as_base_exp()
Returns the method as the 2-tuple (base, exponent).
fdiff(argindex=1)
Returns the first derivative of the function.
is_commutative
Returns whether the function is commutative.
If you want to see which functions appear in an expression you can use the atoms method:
If you just want the function you defined, not SymPy functions, the thing to search for is
AppliedUndef:
Subs
class sympy.core.function.Subs
Represents unevaluated substitutions of an expression.
Subs(expr, x, x0) receives 3 arguments: an expression, a variable or list of distinct
variables and a point or list of evaluation points corresponding to those variables.
Subs objects are generally useful to represent unevaluated derivatives calculated at a
point.
The variables may be expressions, but they are subjected to the limitations of subs(), so
it is usually a good practice to use only symbols for variables, since in that case there
can be no ambiguity.
There’s no automatic expansion - use the method .doit() to effect all possible substitutions
of the object and also of objects inside the expression.
When evaluating derivatives at a point that is not a symbol, a Subs object is returned.
One is also able to calculate derivatives of Subs objects - in this case the expression is
always expanded (for the unevaluated form, use Derivative()).
A simple example:
expr
The expression on which the substitution operates
point
The values for which the variables are to be substituted
variables
The variables to be evaluated
expand
Notes
• Hints are applied in an arbitrary, but consistent order (in the current implemen-
tation, they are applied in alphabetical order, except multinomial comes before
mul, but this may change). Because of this, some hints may prevent expansion by
other hints if they are applied first. For example, mul may distribute multiplica-
tions and prevent log and power_base from expanding them. Also, if mul is applied
before multinomial`, the expression might not be fully distributed. The
solution is to use the various ``expand_hint helper functions or to use
hint=False to this function to finely control which hints are applied. Here are some
examples:
Here, we see that log was applied before mul. To get the mul expanded form, either
of the following will work:
To get the power_base expanded form, either of the following will work:
• The modulus meta-hint can be used to reduce the coefficients of an expression post-
expansion:
• Either expand() the function or .expand() the method can be used. Both are equiv-
alent:
Examples
>>> b = MyClass(1, 2, 3, 4, 5)
>>> b.expand(double=True)
MyClass(1, 2, 3, 4, 5)
>>> b.expand(double=True, force=True)
MyClass(1, 2, 3, 4, 5, 1, 2, 3, 4, 5)
Hints
Mul
Multinomial
Power_exp
Power_base
>>> ((x*y)**z).expand(power_base=True)
(x*y)**z
>>> ((x*y)**z).expand(power_base=True, force=True)
x**z*y**z
>>> ((2*y)**z).expand(power_base=True)
2**z*y**z
Note that in some cases where this expansion always holds, SymPy performs it automat-
ically:
>>> (x*y)**2
x**2*y**2
Log
Pull out power of an argument as a coefficient and split logs products into sums of logs.
Note that these only work if the arguments of the log function have the proper
assumptions–the arguments must be positive and the exponents must be real–or else
the force hint must be True:
Basic
This hint is intended primarily as a way for custom subclasses to enable expansion by
default.
These hints are not run by default:
Complex
>>> x, y = symbols('x,y')
>>> (x + y).expand(complex=True)
re(x) + re(y) + I*im(x) + I*im(y)
>>> cos(x).expand(complex=True)
-I*sin(re(x))*sinh(im(x)) + cos(re(x))*cosh(im(x))
Note that this is just a wrapper around as_real_imag(). Most objects that wish to
redefine _eval_expand_complex() should consider redefining as_real_imag() instead.
Func
Trig
Do trigonometric expansions.
Note that the forms of sin(n*x) and cos(n*x) in terms of sin(x) and cos(x) are not
2 2
unique, due to the identity sin (x) + cos (x) = 1. The current implementation uses the
form obtained from Chebyshev polynomials, but this may change. See this MathWorld
article for more information.
Api
Objects can define their own expand hints by defining _eval_expand_hint(). The func-
tion should take the form:
See also the example below. Objects should define _eval_expand_hint() methods only
if hint applies to that specific object. The generic _eval_expand_hint() method defined
in Expr will handle the no-op case.
Each hint should be responsible for expanding that hint only. Furthermore, the expansion
should be applied to the top-level expression only. expand() takes care of the recursion
that happens when deep=True.
You should only call _eval_expand_hint() methods directly if you are 100% sure
that the object has the method, as otherwise you are liable to get unexpected At-
tributeError``s. Note, again, that you do not need to recursively apply
the hint to args of your object: this is handled automatically by ``ex-
pand(). _eval_expand_hint() should generally not be used at all outside of an
_eval_expand_hint() method. If you want to apply a specific expansion from within
another method, use the public expand() function, method, or expand_hint() functions.
In order for expand to work, objects must be rebuildable by their args, i.e., obj.
func(*obj.args) == obj must hold.
Expand methods are passed **hints so that expand hints may use ‘metahints’–hints
that control how different expand methods are applied. For example, the force=True
hint described above that causes expand(log=True) to ignore assumptions is such a
metahint. The deep meta-hint is handled exclusively by expand() and is not passed to
_eval_expand_hint() methods.
Note that expansion hints should generally be methods that perform some kind of ‘ex-
pansion’. For hints that simply rewrite an expression, use the .rewrite() API.
PoleError
class sympy.core.function.PoleError
count_ops
sympy.core.function.count_ops(expr, visual=False)
Return a representation (integer or expression) of the operations in expr.
If visual is False (default) then the sum of the coefficients of the visual expression will
be returned.
If visual is True then the number of each type of operation is shown with the core class
types (or their virtual equivalent) multiplied by the number of times they occur.
If expr is an iterable, the sum of the op counts of the items will be returned.
Examples
Although there isn’t a SUB object, minus signs are interpreted as either negations or
subtractions:
>>> (x - y).count_ops(visual=True)
SUB
>>> (-x).count_ops(visual=True)
NEG
>>> (1 + a + b**2).count_ops(visual=True)
2*ADD + POW
for a total of 5:
Note that “what you type” is not always what you get. The expression 1/x/y is translated
by sympy into 1/(x*y) so it gives a DIV and MUL rather than two DIVs:
>>> (1/x/y).count_ops(visual=True)
DIV + MUL
The visual option can be used to demonstrate the difference in operations for expressions
in different forms. Here, the Horner representation is compared with the expanded form
of a polynomial:
expand_mul
sympy.core.function.expand_mul(expr, deep=True)
Wrapper around expand that only uses the mul hint. See the expand docstring for more
information.
Examples
expand_log
Examples
expand_func
sympy.core.function.expand_func(expr, deep=True)
Wrapper around expand that only uses the func hint. See the expand docstring for more
information.
Examples
expand_trig
sympy.core.function.expand_trig(expr, deep=True)
Wrapper around expand that only uses the trig hint. See the expand docstring for more
information.
Examples
expand_complex
sympy.core.function.expand_complex(expr, deep=True)
Wrapper around expand that only uses the complex hint. See the expand docstring for
more information.
See also:
Expr.as_real_imag
Examples
expand_multinomial
sympy.core.function.expand_multinomial(expr, deep=True)
Wrapper around expand that only uses the multinomial hint. See the expand docstring
for more information.
Examples
expand_power_exp
sympy.core.function.expand_power_exp(expr, deep=True)
Wrapper around expand that only uses the power_exp hint.
See the expand docstring for more information.
Examples
expand_power_base
>>> (x*y)**2
x**2*y**2
>>> (2*x)**y
(2*x)**y
>>> expand_power_base(_)
2**y*x**y
>>> expand_power_base((x*y)**z)
(x*y)**z
>>> expand_power_base((x*y)**z, force=True)
x**z*y**z
>>> expand_power_base(sin((x*y)**z), deep=False)
sin((x*y)**z)
>>> expand_power_base(sin((x*y)**z), force=True)
sin(x**z*y**z)
>>> expand_power_base((2*exp(y))**x)
2**x*exp(y)**x
>>> expand_power_base((2*cos(x))**y)
2**y*cos(x)**y
Notice that sums are left untouched. If this is not the desired behavior, apply full ex-
pand() to the expression:
>>> expand_power_base(((x+y)*z)**2)
z**2*(x + y)**2
>>> (((x+y)*z)**2).expand()
x**2*z**2 + 2*x*y*z**2 + y**2*z**2
>>> expand_power_base((2*y)**(1+z))
2**(z + 1)*y**(z + 1)
(continues on next page)
nfloat
Examples
5.1.20 evalf
PrecisionExhausted
class sympy.core.evalf.PrecisionExhausted
Examples
5.1.21 containers
Tuple
class sympy.core.containers.Tuple
Wrapper around the builtin tuple object
The Tuple is a subclass of Basic, so that it works well in the SymPy framework. The
wrapped tuple is available as self.args, but you can also access elements or slices with
[:] syntax.
Parameters sympify : bool
If False, sympify is not called on args. This can be used for speedups
for very large tuples where the elements are known to already be
sympy objects.
Example
Dict
class sympy.core.containers.Dict
Wrapper around the builtin dict object
The Dict is a subclass of Basic, so that it works well in the SymPy framework. Because it
is immutable, it may be included in sets, but its values must all be given at instantiation
and cannot be changed afterwards. Otherwise it behaves identically to the Python dict.
The args are sympified so the 1 and 2 are Integers and the values are Symbols. Queries
automatically sympify args so the following work:
>>> 1 in D
True
>>> D.has('one') # searches keys and values
True
>>> 'one' in D # not in the keys
False
>>> D[1]
one
5.1.22 compatibility
iterable
Examples
is_sequence
sympy.core.compatibility.is_sequence(i, include=None)
Return a boolean indicating whether i is a sequence in the SymPy sense. If anything
that fails the test below should be included as being a sequence for your application, set
‘include’ to that object’s type; multiple types should be passed as a tuple of types.
Note: although generators can generate a sequence, they often need special handling to
make sure their elements are captured before the generator is exhausted, so these are
not included by default in the definition of a sequence.
See also: iterable
Examples
as_int
sympy.core.compatibility.as_int(n)
Convert the argument to a builtin integer.
The return value is guaranteed to be equal to the input. ValueError is raised if the input
has a non-integral value.
Examples
5.1.23 exprtools
gcd_terms
Examples
The clear flag was ignored in this case because the returned expression was a rational
expression, not a simple sum.
factor_terms
Examples
When clear is False, a rational will only be factored out of an Add expression if all terms
of the Add have coefficients that are fractions:
>>> factor_terms(x/2 + 1, clear=False)
x/2 + 1
>>> factor_terms(x/2 + 1, clear=True)
(x + 2)/2
If a -1 is all that can be factored out, to not factor it out, the flag sign must be False:
>>> factor_terms(-x - y)
-(x + y)
>>> factor_terms(-x - y, sign=False)
-x - y
>>> factor_terms(-2*x - 2*y, sign=False)
-2*(x + y)
5.2.1 Contents
Partitions
class sympy.combinatorics.partitions.Partition
This class represents an abstract partition.
A partition is a set of disjoint sets whose union equals a given set.
See also:
sympy.utilities.iterables.partitions (page 1470), sympy.utilities.iterables.
multiset_partitions (page 1466)
Attributes
is_Complement
is_EmptySet
is_Intersection
is_UniversalSet
RGS
Returns the “restricted growth string” of the partition.
The RGS is returned as a list of indices, L, where L[i] indicates the block in which
element i appears. For example, in a partition of 3 elements (a, b, c) into 2 blocks
([c], [a, b]) the RGS is [1, 1, 0]: “a” is in block 1, “b” is in block 1 and “c” is in block
0.
Examples
Examples
partition
Return partition as a sorted list of lists.
Examples
rank
Gets the rank of a partition.
Examples
sort_key(order=None)
Return a canonical key that can be used for sorting.
Ordering is based on the size and sorted elements of the partition and ties are broken
with the rank.
Examples
class sympy.combinatorics.partitions.IntegerPartition
This class represents an integer partition.
In number theory and combinatorics, a partition of a positive integer, n, also called an
integer partition, is a way of writing n as a list of positive integers that sum to n. Two
partitions that differ only in the order of summands are considered to be the same par-
tition; if order matters then the partitions are referred to as compositions. For example,
4 has five partitions: [4], [3, 1], [2, 2], [2, 1, 1], and [1, 1, 1, 1]; the compositions [1, 2,
1] and [1, 1, 2] are the same as partition [2, 1, 1].
See also:
sympy.utilities.iterables.partitions (page 1470), sympy.utilities.iterables.
multiset_partitions (page 1466)
Reference http://en.wikipedia.org/wiki/Partition_%28number_theory%29
as_dict()
Return the partition as a dictionary whose keys are the partition integers and the
values are the multiplicity of that integer.
Examples
as_ferrers(char=’#’)
Prints the ferrer diagram of a partition.
Examples
conjugate
Computes the conjugate partition of itself.
Examples
next_lex()
Return the next partition of the integer, n, in lexical order, wrapping around to [n]
if the partition is [1, …, 1].
Examples
prev_lex()
Return the previous partition of the integer, n, in lexical order, wrapping around to
[1, …, 1] if the partition is [n].
Examples
sympy.combinatorics.partitions.random_integer_partition(n, seed=None)
Generates a random integer partition summing to n as a list of reverse-sorted integers.
Examples
For the following, a seed is given so a known value can be shown; in practice, the seed
would not be given.
sympy.combinatorics.partitions.RGS_generalized(m)
Computes the m + 1 generalized unrestricted growth strings and returns them as rows
in matrix.
Examples
sympy.combinatorics.partitions.RGS_enum(m)
RGS_enum computes the total number of restricted growth strings possible for a super-
set of size m.
Examples
We can check that the enumeration is correct by actually generating the partitions. Here,
the 15 partitions of 4 items are generated:
>>> a = Partition(list(range(4)))
>>> s = set()
>>> for i in range(20):
... s.add(a)
... a += 1
...
>>> assert len(s) == 15
sympy.combinatorics.partitions.RGS_unrank(rank, m)
Gives the unranked restricted growth string for a given superset size.
Examples
sympy.combinatorics.partitions.RGS_rank(rgs)
Computes the rank of a restricted growth string.
Examples
Permutations
class sympy.combinatorics.permutations.Permutation
A permutation, alternatively known as an ‘arrangement number’ or ‘ordering’ is an ar-
rangement of the elements of an ordered list into a one-to-one mapping with itself. The
permutation of a given arrangement is given by indicating the positions of the elements
after re-arrangement [R30]. For example, if one started with elements [x, y, a, b] (in that
order) and they were reordered as [x, y, b, a] then the permutation would be [0, 1, 3, 2].
Notice that (in SymPy) the first element is always referred to as 0 and the permutation
uses the indices of the elements in the original ordering, not the elements (a, b, etc…)
themselves.
See also:
Cycle (page 236)
References
Permutations Notation
In the 2-line form, the elements and their final positions are shown as a matrix with 2
rows:
[0 1 2 … n-1] [p(0) p(1) p(2) … p(n-1)]
Since the first line is always range(n), where n is the size of p, it is sufficient to represent
the permutation by the second line, referred to as the “array form” of the permutation.
This is entered in brackets as the argument to the Permutation class:
The composite of two permutations p*q means first apply p, then q, so i^(p*q) = (i^p)^q
which is i^p^q according to Python precedence rules:
One can use also the notation p(i) = i^p, but then the composition rule is (p*q)(i) =
q(p(i)), not p(q(i)):
In disjoint cycle notation, only the elements that have shifted are indicated. In the above
case, the 2 and 1 switched places. This can be entered in two ways:
The disjoint cycle notation is convenient when representing permutations that have sev-
eral cycles in them:
It also provides some economy in entry when computing products of permutations that
are written in disjoint cycle notation:
Caution: when the cycles have common elements between them then the order
in which the permutations are applied matters. The convention is that the per-
mutations are applied from right to left. In the following, the transposition of
elements 2 and 3 is followed by the transposition of elements 1 and 2:
If the first and second elements had been swapped first, followed by the swap-
ping of the second and third, the result would have been [0, 2, 3, 1]. If, for
some reason, you want to apply the cycles in the order they are entered, you
can simply reverse the order of cycles:
Cyclic-form entry:
Caution: no singleton containing an element larger than the largest in any previous cycle
can be entered. This is an important difference in how Permutation and Cycle handle the
__call__ syntax. A singleton argument at the start of a Permutation performs instantiation
of the Permutation and is permitted:
>>> Permutation(5)
Permutation([], size=6)
A singleton entered after instantiation is a call to the permutation – a function call – and
if the argument is out of range it will trigger an error. For this reason, it is better to start
the cycle with the singleton:
The following fails because there is is no element 3:
This is ok: only the call to an out of range singleton is prohibited; otherwise the permu-
tation autosizes:
>>> Permutation(3)(1, 2)
Permutation([0, 2, 1, 3])
>>> Permutation(1, 2)(3, 4) == Permutation(3, 4)(1, 2)
True
Equality Testing
The array forms must be the same in order for permutations to be equal:
Identity Permutation
Watch out for entering the range inside a set of brackets (which is cycle notation):
>>> I == Permutation([range(4)])
False
Permutation Printing
There are a few things to note about how Permutations are printed.
1) If you prefer one form (array or cycle) over another, you can set that with the
print_cyclic flag.
2) Regardless of the setting, a list of elements in the array for cyclic form can be obtained
and either of those can be copied and supplied as the argument to Permutation:
>>> p.array_form
[0, 2, 1, 4, 5, 3]
>>> p.cyclic_form
[[1, 2], [3, 4, 5]]
>>> Permutation(_) == p
True
3) Printing is economical in that as little as possible is printed while retaining all infor-
mation about the size of the permutation:
The 2 was not printed but it is still there as can be seen with the array_form and size
methods:
>>> p.array_form
[1, 0, 2, 3]
>>> p.size
4
The permutation can act as a bijective function, telling what element is located at a given
position
>>> p.full_cyclic_form
[[0, 1], [2], [3]]
>>> p.size
4
>>> p.cardinality
24
A given permutation has a rank among all the possible permutations of the same ele-
ments, but what that rank is depends on how the permutations are enumerated. (There
are a number of different methods of doing so.) The lexicographic rank is given by the
rank method and this rank is used to increment a permutation with addition/subtraction:
>>> p.rank()
6
>>> p + 1
Permutation([1, 0, 3, 2])
>>> p.next_lex()
Permutation([1, 0, 3, 2])
>>> _.rank()
7
>>> p.unrank_lex(p.size, rank=7)
Permutation([1, 0, 3, 2])
The permutation can be ‘applied’ to any list-like object, not only Permutations:
If you have a list of arbitrary elements, the corresponding permutation can be found with
the from_sequence method:
>>> Permutation.from_sequence('SymPy')
Permutation([1, 3, 2, 0, 4])
array_form
Return a copy of the attribute _array_form Examples ========
ascents()
Returns the positions of ascents in a permutation, ie, the location where p[i] < p[i+1]
See also:
descents (page 221), inversions (page 225), min (page 229), max (page 229)
Examples
atoms()
Returns all the elements of a permutation
Examples
cardinality
Returns the number of all possible permutations.
See also:
length (page 228), order (page 230), rank (page 231), size (page 234)
Examples
commutator(x)
Return the commutator of self and x: ~x*~self*x*self
If f and g are part of a group, G, then the commutator of f and g is the group identity
iff f and g commute, i.e. fg == gf.
References
http://en.wikipedia.org/wiki/Commutator
Examples
>>> I = Permutation(3)
>>> p = [I + i for i in range(6)]
>>> for i in range(len(p)):
... for j in range(len(p)):
... c = p[i].commutator(p[j])
... if p[i]*p[j] == p[j]*p[i]:
... assert c == I
... else:
... assert c != I
...
commutes_with(other)
Checks if the elements are commuting.
Examples
cycle_structure
Return the cycle structure of the permutation as a dictionary indicating the multi-
plicity of each cycle length.
Examples
cycles
Returns the number of cycles contained in the permutation (including singletons).
See also:
sympy.functions.combinatorial.numbers.stirling (page 495)
Examples
cyclic_form
This is used to convert to the cyclic notation from the canonical notation. Singletons
are omitted.
See also:
array_form (page 218), full_cyclic_form (page 222)
Examples
descents()
Returns the positions of descents in a permutation, ie, the location where p[i] >
p[i+1]
See also:
ascents (page 219), inversions (page 225), min (page 229), max (page 229)
Examples
classmethod from_inversion_vector(inversion)
Calculates the permutation from the inversion vector.
Examples
Examples
>>> Permutation.from_sequence('SymPy')
(4)(0 1 3)
>>> _(sorted(”SymPy”))
['S', 'y', 'm', 'P', 'y']
>>> Permutation.from_sequence('SymPy', key=lambda x: x.lower())
(4)(0 2)(1 3)
full_cyclic_form
Return permutation in cyclic form including singletons.
Examples
get_adjacency_distance(other)
Computes the adjacency distance between two permutations.
This metric counts the number of times a pair i,j of jobs is adjacent in both p and p’.
If n_adj is this quantity then the adjacency distance is n - n_adj - 1 [1]
[1] Reeves, Colin R. Landscapes, Operators and Heuristic search, Annals of Opera-
tional Research, 86, pp 473-490. (1999)
See also:
get_precedence_matrix (page 224), get_precedence_distance (page 224),
get_adjacency_matrix (page 223)
Examples
get_adjacency_matrix()
Computes the adjacency matrix of a permutation.
If job i is adjacent to job j in a permutation p then we set m[i, j] = 1 where m is the
adjacency matrix of p.
See also:
get_precedence_matrix (page 224), get_precedence_distance (page 224),
get_adjacency_distance (page 222)
Examples
get_positional_distance(other)
Computes the positional distance between two permutations.
See also:
get_precedence_distance (page 224), get_adjacency_distance (page 222)
Examples
get_precedence_distance(other)
Computes the precedence distance between two permutations.
Suppose p and p’ represent n jobs. The precedence metric counts the number of
times a job j is preceded by job i in both p and p’. This metric is commutative.
See also:
get_precedence_matrix (page 224), get_adjacency_matrix (page 223),
get_adjacency_distance (page 222)
Examples
get_precedence_matrix()
Gets the precedence matrix. This is used for computing the distance between two
permutations.
See also:
get_precedence_distance (page 224), get_adjacency_matrix (page 223),
get_adjacency_distance (page 222)
Examples
index()
Returns the index of a permutation.
The index of a permutation is the sum of all subscripts j such that p[j] is greater than
p[j+1].
Examples
inversion_vector()
Return the inversion vector of the permutation.
The inversion vector consists of elements whose value indicates the number of ele-
ments in the permutation that are lesser than it and lie on its right hand side.
The inversion vector is the same as the Lehmer encoding of a permutation.
See also:
from_inversion_vector (page 222)
Examples
The inversion vector increases lexicographically with the rank of the permutation,
the -ith element cycling through 0..i.
>>> p = Permutation(2)
>>> while p:
... print('%s %s %s' % (p, p.inversion_vector(), p.rank()))
... p = p.next_lex()
...
Permutation([0, 1, 2]) [0, 0] 0
Permutation([0, 2, 1]) [0, 1] 1
Permutation([1, 0, 2]) [1, 0] 2
Permutation([1, 2, 0]) [1, 1] 3
Permutation([2, 0, 1]) [2, 0] 4
Permutation([2, 1, 0]) [2, 1] 5
inversions()
Computes the number of inversions of a permutation.
An inversion is where i > j but p[i] < p[j].
For small length of p, it iterates over all i and j values and calculates the number of
inversions. For large length of p, it uses a variation of merge sort to calculate the
number of inversions.
See also:
descents (page 221), ascents (page 219), min (page 229), max (page 229)
References
[1] http://www.cp.eng.chula.ac.th/~piak/teaching/algo/algo2008/count-inv.htm
Examples
is_Empty
Checks to see if the permutation is a set with zero elements
See also:
is_Singleton (page 226)
Examples
is_Identity
Returns True if the Permutation is an identity permutation.
See also:
order (page 230)
Examples
is_Singleton
Checks to see if the permutation contains only one number and is thus the only
possible permutation of this set of numbers
See also:
is_Empty (page 226)
Examples
is_even
Checks if a permutation is even.
See also:
is_odd (page 227)
Examples
is_odd
Checks if a permutation is odd.
See also:
is_even (page 227)
Examples
choices chosen
======== ======
012345
01 345 2
01 34 25
01 4 253
0 4 2531
0 25314
253140
References
1. http://en.wikipedia.org/wiki/Flavius_Josephus
2. http://en.wikipedia.org/wiki/Josephus_problem
3. http://www.wou.edu/~burtonl/josephus.html
Examples
length()
Returns the number of integers moved by a permutation.
See also:
min (page 229), max (page 229), support (page 234), cardinality (page 219), order
(page 230), rank (page 231), size (page 234)
Examples
list(size=None)
Return the permutation as an explicit list, possibly trimming unmoved elements if
size is less than the maximum element in the permutation; if this is desired, setting
size=-1 will guarantee such trimming.
Examples
Passing a length too small will trim trailing, unchanged elements in the permutation:
max()
The maximum element moved by the permutation.
See also:
min (page 229), descents (page 221), ascents (page 219), inversions (page 225)
Examples
min()
The minimum element moved by the permutation.
See also:
max (page 229), descents (page 221), ascents (page 219), inversions (page 225)
Examples
mul_inv(other)
other*~self, self and other have _array_form
next_lex()
Returns the next permutation in lexicographical order. If self is the last permutation
in lexicographical order it returns None. See [4] section 2.4.
See also:
rank (page 231), unrank_lex (page 235)
Examples
next_nonlex()
Returns the next permutation in nonlex order [3]. If self is the last permutation in
this order it returns None.
See also:
rank_nonlex (page 232), unrank_nonlex (page 235)
Examples
next_trotterjohnson()
Returns the next permutation in Trotter-Johnson order. If self is the last permutation
it returns None. See [4] section 2.4. If it is desired to generate all such permutations,
they can be generated in order more quickly with the generate_bell function.
See also:
rank_trotterjohnson (page 232), unrank_trotterjohnson (page 235), sympy.
utilities.iterables.generate_bell (page 1460)
Examples
order()
Computes the order of a permutation.
When the permutation is raised to the power of its order it equals the identity per-
mutation.
See also:
identity, cardinality (page 219), length (page 228), rank (page 231), size
(page 234)
Examples
parity()
Computes the parity of a permutation.
The parity of a permutation reflects the parity of the number of inversions in the
permutation, i.e., the number of pairs of x and y such that x > y but p[x] < p[y].
See also:
_af_parity
Examples
classmethod random(n)
Generates a random permutation of length n.
Uses the underlying Python pseudo-random number generator.
Examples
rank()
Returns the lexicographic rank of the permutation.
See also:
next_lex (page 229), unrank_lex (page 235), cardinality (page 219), length
(page 228), order (page 230), size (page 234)
Examples
rank_nonlex(inv_perm=None)
This is a linear time ranking algorithm that does not enforce lexicographic order [3].
See also:
next_nonlex (page 230), unrank_nonlex (page 235)
Examples
rank_trotterjohnson()
Returns the Trotter Johnson rank, which we get from the minimal change algorithm.
See [4] section 2.4.
See also:
unrank_trotterjohnson (page 235), next_trotterjohnson (page 230)
Examples
static rmul(*args)
Return product of Permutations [a, b, c, …] as the Permutation whose ith value is
a(b(c(i))).
a, b, c, … can be Permutation objects or tuples.
Notes
All items in the sequence will be parsed by Permutation as necessary as long as the
first item is a Permutation:
Examples
classmethod rmul_with_af(*args)
same as rmul, but the elements of args are Permutation objects which have _ar-
ray_form
runs()
Returns the runs of a permutation.
An ascending sequence in a permutation is called a run [5].
Examples
signature()
Gives the signature of the permutation needed to place the elements of the permu-
tation in canonical order.
The signature is calculated as (-1)^<number of inversions>
See also:
inversions (page 225)
Examples
size
Returns the number of elements in the permutation.
See also:
cardinality (page 219), length (page 228), order (page 230), rank (page 231)
Examples
support()
Return the elements in permutation, P, for which P[i] != i.
Examples
transpositions()
Return the permutation decomposed into a list of transpositions.
It is always possible to express a permutation as the product of transpositions, see
[1]
References
1. http://en.wikipedia.org/wiki/Transposition_%28mathematics%29#Properties
Examples
Examples
classmethod unrank_nonlex(n, r)
This is a linear time unranking algorithm that does not respect lexicographic order
[3].
See also:
next_nonlex (page 230), rank_nonlex (page 232)
Examples
Examples
class sympy.combinatorics.permutations.Cycle(*args)
Wrapper around dict which provides the functionality of a disjoint cycle.
A cycle shows the rule to use to move subsets of elements to obtain a permutation. The
Cycle class is more flexible than Permutation in that 1) all elements need not be present in
order to investigate how multiple cycles act in sequence and 2) it can contain singletons:
The array form of a Cycle can be obtained by calling the list method (or passing it to the
list function) and all elements from 0 will be shown:
>>> a = Cycle(1, 2)
>>> a.list()
[0, 2, 1]
>>> list(a)
[0, 2, 1]
If a larger (or smaller) range is desired use the list method and provide the desired size
– but the Cycle cannot be truncated to a size smaller than the largest element that is out
of place:
Singletons are not shown when printing with one exception: the largest element is al-
ways shown – as a singleton if necessary:
The array form can be used to instantiate a Permutation so other properties of the per-
mutation can be investigated:
See also:
Permutation (page 213)
Notes
The underlying structure of the Cycle is a dictionary and although the __iter__ method
has been redefined to give the array form of the cycle, the underlying dictionary items
are still available with the such methods as items():
>>> list(Cycle(1, 2).items())
[(1, 2), (2, 1)]
list(size=None)
Return the cycles as an explicit list starting from 0 up to the greater of the largest
value in the cycles and size.
Truncation of trailing unmoved items will occur when size is less than the maximum
element in the cycle; if this is desired, setting size=-1 will guarantee such trimming.
Examples
Passing a length too small will trim trailing, unchanged elements in the permutation:
>>> Cycle(2, 4)(1, 2, 4).list(-1)
[0, 2, 1]
Generators
generators.symmetric()
Generates the symmetric group of order n, Sn.
Examples
generators.cyclic()
Generates the cyclic group of order n, Cn.
See also:
dihedral (page 238)
Examples
generators.alternating()
Generates the alternating group of order n, An.
Examples
generators.dihedral()
Generates the dihedral group of order 2n, Dn.
The result is given as a subgroup of Sn, except for the special cases n=1 (the group S2)
and n=2 (the Klein 4-group) where that’s not possible and embeddings in S2 and S4
respectively are given.
See also:
cyclic (page 237)
Examples
Permutation Groups
class sympy.combinatorics.perm_groups.PermutationGroup
The class defining a Permutation group.
PermutationGroup([p1, p2, …, pn]) returns the permutation group generated by the list
of permutations. This group can be supplied to Polyhedron if one desires to decorate the
elements to which the indices of the permutation refer.
See also:
sympy.combinatorics.polyhedron.Polyhedron (page 267), sympy.combinatorics.
permutations.Permutation (page 213)
References
[1] Holt, D., Eick, B., O’Brien, E. “Handbook of Computational Group Theory”
[2] Seress, A. “Permutation Group Algorithms”
[3] http://en.wikipedia.org/wiki/Schreier_vector
[4] http://en.wikipedia.org/wiki/Nielsen_transformation #Prod-
uct_replacement_algorithm
[5] Frank Celler, Charles R.Leedham-Green, Scott H.Murray, Alice C.Niemeyer, and
E.A.O’Brien. “Generating Random Elements of a Finite Group”
[6] http://en.wikipedia.org/wiki/Block_%28permutation_group_theory%29
[7] http://www.algorithmist.com/index.php/Union_Find
[8] http://en.wikipedia.org/wiki/Multiply_transitive_group#Multiply_transitive_groups
[9] http://en.wikipedia.org/wiki/Center_%28group_theory%29
[10] http://en.wikipedia.org/wiki/Centralizer_and_normalizer
[11] http://groupprops.subwiki.org/wiki/Derived_subgroup
[12] http://en.wikipedia.org/wiki/Nilpotent_group
[13] http://www.math.colostate.edu/~hulpke/CGT/cgtnotes.pdf
Examples
The permutations corresponding to motion of the front, right and bottom face of a 2x2
Rubik’s cube are defined:
>>> G = PermutationGroup(F, R, D)
>>> G.order()
3674160
The group can be supplied to a Polyhedron in order to track the objects being moved.
An example involving the 2x2 Rubik’s cube is given there, but here is a simple demon-
stration:
>>> a = Permutation(2, 1)
>>> b = Permutation(1, 0)
>>> G = PermutationGroup(a, b)
>>> P = Polyhedron(list('ABC'), pgroup=G)
>>> P.corners
(A, B, C)
>>> P.rotate(0) # apply permutation 0
>>> P.corners
(A, C, B)
>>> P.reset()
>>> P.corners
(A, B, C)
Or one can make a permutation as a product of selected permutations and apply them
to an iterable directly:
base
Return a base from the Schreier-Sims algorithm.
For a permutation group G, a base is a sequence of points B = (b1 , b2 , ..., bk ) such that
no element of G apart from the identity fixes all the points in B. The concepts of a
base and strong generating set and their applications are discussed in depth in [1],
pp. 87-89 and [2], pp. 55-57.
An alternative way to think of B is that it gives the indices of the stabilizer cosets
that contain more than the identity permutation.
See also:
strong_gens (page 264), basic_transversals (page 242), basic_orbits
(page 241), basic_stabilizers (page 242)
Examples
Notes
The deterministic version of the algorithm is discussed in [1], pp. 102-103; the
randomized version is discussed in [1], p.103, and [2], p.98. It is of Las Vegas type.
Notice that [1] contains a mistake in the pseudocode and discussion of BASESWAP:
⟨T ⟩ ⟨T ⟩
on line 3 of the pseudocode, |βi+1 | should be replaced by |βi |, and the same for the
discussion of the algorithm.
Examples
>>> S1 = PermutationGroup(gens)
>>> _verify_bsgs(S1, base, gens)
True
basic_orbits
Return the basic orbits relative to a base and strong generating set.
If (b1 , b2 , ..., bk ) is a base for a group G, and G(i) = Gb1 ,b2 ,...,bi−1 is the i-th basic stabilizer
(so that G(1) = G), the i-th basic orbit relative to this base is the orbit of bi under G(i) .
See [1], pp. 87-89 for more information.
See also:
base (page 240), strong_gens (page 264), basic_transversals (page 242), ba-
sic_stabilizers (page 242)
Examples
basic_stabilizers
Return a chain of stabilizers relative to a base and strong generating set.
The i-th basic stabilizer G(i) relative to a base (b1 , b2 , ..., bk ) is Gb1 ,b2 ,...,bi−1 . For more
information, see [1], pp. 87-89.
See also:
base (page 240), strong_gens (page 264), basic_orbits (page 241), ba-
sic_transversals (page 242)
Examples
basic_transversals
Return basic transversals relative to a base and strong generating set.
The basic transversals are transversals of the basic orbits. They are provided as a
list of dictionaries, each dictionary having keys - the elements of one of the basic
orbits, and values - the corresponding transversal elements. See [1], pp. 87-89 for
more information.
See also:
strong_gens (page 264), base (page 240), basic_orbits (page 241), ba-
sic_stabilizers (page 242)
Examples
center()
Return the center of a permutation group.
The center for a group G is defined as Z(G) = {z ∈ G|∀g ∈ G, zg = gz}, the set of
elements of G that commute with all elements of G. It is equal to the centralizer of
G inside G, and is naturally a subgroup of G ([9]).
See also:
centralizer (page 243)
Notes
Examples
centralizer(other)
Return the centralizer of a group/set/element.
The centralizer of a set of permutations S inside a group G is the set of elements of
G that commute with all elements of S:
See also:
subgroup_search (page 265)
Notes
Examples
commutator(G, H)
Return the commutator of two subgroups.
For a permutation group K and subgroups G, H, the commutator of G and H is defined
as the group generated by all the commutators [g, h] = hgh−1 g −1 for g in G and h in H.
It is naturally a subgroup of K ([1], p.27).
See also:
derived_subgroup (page 248)
Notes
The commutator of two subgroups H, G is equal to the normal closure of the com-
mutators of all the generators, i.e. hgh−1 g −1 for h a generator of H and g a generator
of G ([1], p.28)
Examples
contains(g, strict=True)
Test if permutation g belong to self, G.
If g is an element of G it can be written as a product of factors drawn from the cosets
of G’s stabilizers. To see if g is one of the actual generators defining the group use
G.has(g).
If strict is not True, g will be resized, if necessary, to match the size of permutations
in self.
See also:
coset_factor (page 245), has, in
Examples
>>> a = Permutation(1, 2)
>>> b = Permutation(2, 3, 1)
>>> G = PermutationGroup(a, b, degree=5)
>>> G.contains(G[0]) # trivial check
True
>>> elem = Permutation([[2, 3]], size=5)
>>> G.contains(elem)
True
>>> G.contains(Permutation(4)(0, 1, 2, 3))
False
>>> H = PermutationGroup(Permutation(5))
>>> H.contains(Permutation(3))
False
>>> H.contains(Permutation(3), strict=False)
True
coset_factor(g, factor_index=False)
Return G’s (self’s) coset factorization of g
If g is an element of G then it can be written as the product of permutations drawn
from the Schreier-Sims coset decomposition,
The permutations returned in f are those for which the product gives g: g = f[n]*.
..f[1]*f[0] where n = len(B) and B = G.base. f[i] is one of the permutations in
self._basic_orbits[i].
If factor_index==True, returns a tuple [b[0],..,b[n]], where b[i] belongs to
self._basic_orbits[i]
Examples
Define g:
>>> g = Permutation(7)(1, 2, 4)(3, 6, 5)
Thus, it can be written as a product of factors (up to 3) drawn from u. See below
that a factor from u1 and u2 and the Identity permutation have been used:
>>> f = G.coset_factor(g)
>>> f[2]*f[1]*f[0] == g
True
>>> f1 = G.coset_factor(g, True); f1
[0, 4, 4]
>>> tr = G.basic_transversals
>>> f[0] == tr[0][f1[0]]
True
see util._strip
coset_rank(g)
rank using Schreier-Sims representation
The coset rank of g is the ordering number in which it appears in the lexicographic
listing according to the coset decomposition
The ordering is the same as in G.generate(method=’coset’). If g does not belong to
the group it returns None.
See also:
coset_factor (page 245)
Examples
coset_table(H)
Return the standardised (right) coset table of self in H as a list of lists.
coset_transversal(H)
Return a transversal of the right cosets of self by its subgroup H using the second
method described in [1], Subsection 4.6.7
coset_unrank(rank, af=False)
unrank using Schreier-Sims representation
coset_unrank is the inverse operation of coset_rank if 0 <= rank < order; otherwise
it returns None.
degree
Returns the size of the permutations in the group.
The number of permutations comprising the group is given by len(group); the num-
ber of permutations that can be generated by the group is given by group.order().
See also:
order (page 259)
Examples
derived_series()
Return the derived series for the group.
The derived series for a group G is defined as G = G0 > G1 > G2 > . . . where Gi =
[Gi−1 , Gi−1 ], i.e. Gi is the derived subgroup of Gi−1 , for i ∈ N. When we have Gk = Gk−1
for some k ∈ N, the series terminates.
Returns A list of permutation groups containing the members of the derived
series in the order G = G0 , G1 , G2 , . . ..
See also:
derived_subgroup (page 248)
Examples
derived_subgroup()
Compute the derived subgroup.
The derived subgroup, or commutator subgroup is the subgroup generated by all
commutators [g, h] = hgh−1 g −1 for g, h ∈ G ; it is equal to the normal closure of the set
of commutators of the generators ([1], p.28, [11]).
See also:
derived_series (page 247)
Examples
elements
Returns all the elements of the permutation group as a set
Examples
generate(method=’coset’, af=False)
Return iterator to generate the elements of the group
Iteration is done with one of these methods:
Examples
The permutation group given in the tetrahedron object is also true groups:
>>> G = tetrahedron.pgroup
>>> G.is_group
True
Also the group generated by the permutations in the tetrahedron pgroup – even the
first two – is a proper group:
>>> H = PermutationGroup(G[0], G[1])
>>> J = PermutationGroup(list(H.generate())); J
PermutationGroup([
(0 1)(2 3),
(1 2 3),
(1 3 2),
(0 3 1),
(0 2 3),
(0 3)(1 2),
(0 1 3),
(3)(0 2 1),
(0 3 2),
(3)(0 1 2),
(0 2)(1 3)])
>>> _.is_group
True
generate_dimino(af=False)
Yield group elements using Dimino’s algorithm
If af == True it yields the array form of the permutations
References
[1] The Implementation of Various Algorithms for Permutation Groups in the Com-
puter Algebra System: AXIOM, N.J. Doye, M.Sc. Thesis
Examples
generate_schreier_sims(af=False)
Yield group elements using the Schreier-Sims representation in coset_rank order
If af = True it yields the array form of the permutations
Examples
generator_product(g, original=False)
Return a list of strong generators [s1, ..., sn] s.t g = sn ∗ ... ∗ s1. If original = T rue, make
the list contain only the original group generators
generators
Returns the generators of the group.
Examples
identity
Return the identity element of the permutation group.
is_abelian
Test if the group is Abelian.
Examples
is_alt_sym(eps=0.05, _random_prec=None)
Monte Carlo test for the symmetric/alternating group for degrees >= 8.
More specifically, it is one-sided Monte Carlo with the answer True (i.e., G is sym-
metric/alternating) guaranteed to be correct, and the answer False being incorrect
with probability eps.
For degree < 8, the order of the group is checked so the test is deterministic.
See also:
_check_cycles_alt_sym
Notes
The algorithm itself uses some nontrivial results from group theory and number
theory: 1) If a transitive group G of degree n contains an element with a cycle of
length n/2 < p < n-2 for p a prime, G is the symmetric or alternating group ([1],
pp. 81-82) 2) The proportion of elements in the symmetric/alternating group having
the property described in 1) is approximately log(2)/ log(n) ([1], p.82; [2], pp. 226-
227). The helper function _check_cycles_alt_sym is used to go over the cycles in
a permutation and look for ones satisfying 1).
Examples
is_nilpotent
Test if the group is nilpotent.
A group G is nilpotent if it has a central series of finite length. Alternatively, G is
nilpotent if its lower central series terminates with the trivial group. Every nilpotent
group is also solvable ([1], p.29, [12]).
See also:
lower_central_series (page 254), is_solvable (page 253)
Examples
is_normal(gr, strict=True)
Test if G=self is a normal subgroup of gr.
G is normal in gr if for each g2 in G, g1 in gr, g = g1*g2*g1**-1 belongs to G It is
sufficient to check this for each g1 in gr.generators and g2 in G.generators.
Examples
is_primitive(randomized=True)
Test if a group is primitive.
A permutation group G acting on a set S is called primitive if S contains no nontrivial
block under the action of G (a block is nontrivial if its cardinality is more than 1).
See also:
minimal_block (page 256), random_stab (page 261)
Notes
The algorithm is described in [1], p.83, and uses the function minimal_block to
search for blocks of the form {0, k} for k ranging over representatives for the or-
bits of G0 , the stabilizer of 0. This algorithm has complexity O(n2 ) where n is the
degree of the group, and will perform badly if G0 is small.
There are two implementations offered: one finds G0 deterministically using the
function stabilizer, and the other (default) produces random elements of G0 using
random_stab, hoping that they generate a subgroup of G0 with not too many more
orbits than G0 (this is suggested in [1], p.83). Behavior is changed by the randomized
flag.
Examples
is_solvable
Test if the group is solvable.
G is solvable if its derived series terminates with the trivial group ([1], p.29).
See also:
is_nilpotent (page 251), derived_series (page 247)
Examples
is_subgroup(G, strict=True)
Return True if all elements of self belong to G.
If strict is False then if self’s degree is smaller than G’s, the elements will be
resized to have the same degree.
Examples
Testing is strict by default: the degree of each group must be the same:
>>> p = Permutation(0, 1, 2, 3, 4, 5)
>>> G1 = PermutationGroup([Permutation(0, 1, 2), Permutation(0, 1)])
>>> G2 = PermutationGroup([Permutation(0, 2), Permutation(0, 1, 2)])
>>> G3 = PermutationGroup([p, p**2])
>>> assert G1.order() == G2.order() == G3.order() == 6
>>> G1.is_subgroup(G2)
True
>>> G1.is_subgroup(G3)
False
>>> G3.is_subgroup(PermutationGroup(G3[1]))
False
>>> G3.is_subgroup(PermutationGroup(G3[0]))
True
>>> S3 = SymmetricGroup(3)
>>> S5 = SymmetricGroup(5)
>>> S3.is_subgroup(S5, strict=False)
True
>>> C7 = CyclicGroup(7)
>>> G = S5*C7
>>> S5.is_subgroup(G, False)
True
>>> C7.is_subgroup(G, 0)
False
is_transitive(strict=True)
Test if the group is transitive.
A group is transitive if it has a single orbit.
If strict is False the group is transitive if it has a single orbit of length different
from 1.
Examples
is_trivial
Test if the group is the trivial group.
This is true if the group contains only the identity permutation.
Examples
lower_central_series()
Return the lower central series for the group.
The lower central series for a group G is the series G = G0 > G1 > G2 > . . . where
Gk = [G, Gk−1 ], i.e. every term after the first is equal to the commutator of G and the
previous term in G1 ([1], p.29).
Returns A list of permutation groups in the order G = G0 , G1 , G2 , . . .
See also:
commutator (page 244), derived_series (page 247)
Examples
make_perm(n, seed=None)
Multiply n randomly selected permutations from pgroup together, starting with the
identity permutation. If n is a list of integers, those integers will be used to select
the permutations and they will be applied in L to R order: make_perm((A, B, C)) will
give CBA(I) where I is the identity permutation.
seed is used to set the seed for the random selection of permutations from pgroup. If
this is a list of integers, the corresponding permutations from pgroup will be selected
in the order give. This is mainly used for testing purposes.
See also:
random (page 260)
Examples
max_div
Maximum proper divisor of the degree of a permutation group.
See also:
minimal_block (page 256), _union_find_merge
Notes
Obviously, this is the degree divided by its minimal proper divisor (larger than 1, if
one exists). As it is guaranteed to be prime, the sieve from sympy.ntheory is used.
This function is also used as an optimization tool for the functions minimal_block
and _union_find_merge.
Examples
minimal_block(points)
For a transitive group, finds the block system generated by points.
If a group G acts on a set S, a nonempty subset B of S is called a block under the
action of G if for all g in G we have gB = B (g fixes B) or gB and B have no common
points (g moves B entirely). ([1], p.23; [6]).
The distinct translates gB of a block B for g in G partition the set S and this set of
translates is known as a block system. Moreover, we obviously have that all blocks
in the partition have the same size, hence the block size divides |S| ([1], p.23). A G-
congruence is an equivalence relation ~ on the set S such that a ~ b implies g(a) ~
g(b) for all g in G. For a transitive group, the equivalence classes of a G-congruence
and the blocks of a block system are the same thing ([1], p.23).
The algorithm below checks the group for transitivity, and then finds the G-
congruence generated by the pairs (p_0, p_1), (p_0, p_2), ..., (p_0,p_{k-
1}) which is the same as finding the maximal block system (i.e., the one with mini-
mum block size) such that p_0, ..., p_{k-1} are in the same block ([1], p.83).
It is an implementation of Atkinson’s algorithm, as suggested in [1], and manipulates
an equivalence relation on the set S using a union-find data structure. The running
time is just above O(|points||S|). ([1], pp. 83-87; [7]).
See also:
_union_find_rep, _union_find_merge, is_transitive (page 254), is_primitive
(page 252)
Examples
minimal_blocks(randomized=True)
For a transitive group, return the list of all minimal block systems. If a group is
intransitive, return F alse.
See also:
minimal_block (page 256), is_transitive (page 254), is_primitive (page 252)
Examples
normal_closure(other, k=10)
Return the normal closure of a subgroup/set of permutations.
If S is a subset of a group G, the normal closure of A in G is defined as the intersection
of all normal subgroups of G that contain A ([1], p.14). Alternatively, it is the group
generated by the conjugates x^{-1}yx for x a generator of G and y a generator
of the subgroup \left\langle S\right\rangle generated by S (for some chosen
generating set for \left\langle S\right\rangle) ([1], p.73).
Parameters other
a subgroup/list of permutations/single permutation
k
an implementation-specific parameter that determines the number of
conjugates that are adjoined to other at once
See also:
commutator (page 244), derived_subgroup (page 248), random_pr (page 260)
Notes
The algorithm is described in [1], pp. 73-74; it makes use of the generation of ran-
dom elements for permutation groups by the product replacement algorithm.
Examples
orbit(alpha, action=’tuples’)
Compute the orbit of alpha {g(α)|g ∈ G} as a set.
The time complexity of the algorithm used here is O(|Orb| ∗ r) where |Orb| is the size
of the orbit and r is the number of generators of the group. For a more detailed
analysis, see [1], p.78, [2], pp. 19-21. Here alpha can be a single point, or a list of
points.
If alpha is a single point, the ordinary orbit is computed. if alpha is a list of points,
there are three available options:
‘union’ - computes the union of the orbits of the points in the list ‘tuples’ - computes
the orbit of the list interpreted as an ordered tuple under the group action ( i.e.,
g((1,2,3)) = (g(1), g(2), g(3)) ) ‘sets’ - computes the orbit of the list interpreted as a
sets
See also:
orbit_transversal (page 258)
Examples
Examples
orbit_transversal(alpha, pairs=False)
Computes a transversal for the orbit of alpha as a set.
For a permutation group G, a transversal for the orbit Orb = {g(α)|g ∈ G} is a set
{gβ |gβ (α) = β} for β ∈ Orb. Note that there may be more than one possible transversal.
If pairs is set to True, it returns the list of pairs (β, gβ ). For a proof of correctness,
see [1], p.79
See also:
orbit (page 258)
Examples
orbits(rep=False)
Return the orbits of self, ordered according to lowest element in each orbit.
Examples
order()
Return the order of the group: the number of permutations that can be generated
from elements of the group.
The number of permutations comprising the group is given by len(group); the
length of each permutation in the group is given by group.size.
See also:
degree (page 247)
Examples
pointwise_stabilizer(points, incremental=True)
Return the pointwise stabilizer for a set of points.
For a permutation group G and a set of points {p1 , p2 , . . . , pk }, the pointwise stabilizer
of p1 , p2 , . . . , pk is defined as Gp1 ,...,pk = {g ∈ G|g(pi ) = pi ∀i ∈ {1, 2, . . . , k}} ([1],p20). It is
a subgroup of G.
See also:
stabilizer (page 263), schreier_sims_incremental (page 261)
Notes
When incremental == True, rather than the obvious implementation using succes-
sive calls to .stabilizer(), this uses the incremental Schreier-Sims algorithm to
obtain a base with starting segment - the given points.
Examples
presentation(eliminate_gens=True)
Return an F pGroup presentation of the group.
The algorithm is described in [1], Chapter 6.1.
random(af=False)
Return a random group element
random_pr(gen_count=11, iterations=50, _random_prec=None)
Return a random group element using product replacement.
For the details of the product replacement algorithm, see _random_pr_init In ran-
dom_pr the actual ‘product replacement’ is performed. Notice that if the attribute
_random_gens is empty, it needs to be initialized by _random_pr_init.
See also:
_random_pr_init
random_stab(alpha, schreier_vector=None, _random_prec=None)
Random element from the stabilizer of alpha.
The schreier vector for alpha is an optional argument used for speeding up repeated
calls. The algorithm is described in [1], p.81
See also:
random_pr (page 260), orbit_rep (page 258)
schreier_sims()
Schreier-Sims algorithm.
It computes the generators of the chain of stabilizers G > Gb1 > .. > Gb1,..,br > 1 in
which Gb1 ,..,bi stabilizes b1 , .., bi , and the corresponding s cosets. An element of the
group can be written as the product h1 ∗ .. ∗ hs .
We use the incremental Schreier-Sims algorithm.
Examples
See also:
schreier_sims (page 261), schreier_sims_random (page 262)
Notes
This version of the Schreier-Sims algorithm runs in polynomial time. There are cer-
tain assumptions in the implementation - if the trivial group is provided, base and
gens are returned immediately, as any sequence of points is a base for the trivial
group. If the identity is present in the generators gens, it is removed as it is a re-
dundant generator. The implementation is described in [1], pp. 90-93.
Examples
Notes
The algorithm is described in detail in [1], pp. 97-98. It extends the orbits orbs and
the permutation groups stabs to basic orbits and basic stabilizers for the base and
strong generating set produced in the end. The idea of the extension process is to
“sift” random group elements through the stabilizer chain and amend the stabiliz-
ers/orbits along the way when a sift is not successful. The helper function _strip
is used to attempt to decompose a random group element according to the current
state of the stabilizer chain and report whether the element was fully decomposed
(successful sift) or not (unsuccessful sift). In the latter case, the level at which the
sift failed is reported and used to amend stabs, base, gens and orbs accordingly.
The halting condition is for consec_succ consecutive successful sifts to pass. This
makes sure that the current base and gens form a BSGS with probability at least
1 − 1/consec_succ.
Examples
schreier_vector(alpha)
Computes the schreier vector for alpha.
The Schreier vector efficiently stores information about the orbit of alpha. It can
later be used to quickly obtain elements of the group that send alpha to a particular
element in the orbit. Notice that the Schreier vector depends on the order in which
the group generators are listed. For a definition, see [3]. Since list indices start from
zero, we adopt the convention to use “None” instead of 0 to signify that an element
doesn’t belong to the orbit. For the algorithm and its correctness, see [2], pp.78-80.
See also:
orbit (page 258)
Examples
stabilizer(alpha)
Return the stabilizer subgroup of alpha.
The stabilizer of α is the group Gα = {g ∈ G|g(α) = α}. For a proof of correctness, see
[1], p.79.
See also:
orbit (page 258)
Examples
strong_gens
Return a strong generating set from the Schreier-Sims algorithm.
A generating set S = {g1 , g2 , ..., gt } for a permutation group G is a strong generating
set relative to the sequence of points (referred to as a “base”) (b1 , b2 , ..., bk ) if, for
1 ≤ i ≤ k we have that the intersection of the pointwise stabilizer G(i+1) := Gb1 ,b2 ,...,bi
with S generates the pointwise stabilizer G(i+1) . The concepts of a base and strong
generating set and their applications are discussed in depth in [1], pp. 87-89 and
[2], pp. 55-57.
See also:
base (page 240), basic_transversals (page 242), basic_orbits (page 241), ba-
sic_stabilizers (page 242)
Examples
strong_presentation()
Return a strong finite presentation of G. The generators of the returned group are
in the same order as the strong generators of G.
The algorithm is based on Sims’ Verify algorithm described in [1], Chapter 6.
See also:
presentation (page 260), _verify
Example
subgroup(gens)
Return the subgroup generated by gens which is a list of elements of the group
subgroup_search(prop, base=None, strong_gens=None, tests=None,
init_subgroup=None)
Find the subgroup of all elements satisfying the property prop.
This is done by a depth-first search with respect to base images that uses several
tests to prune the search tree.
Parameters prop
The property to be used. Has to be callable on group elements and
always return True or False. It is assumed that all group elements
satisfying prop indeed form a subgroup.
base
A base for the supergroup.
strong_gens
A strong generating set for the supergroup.
tests
A list of callables of length equal to the length of base. These are
used to rule out group elements by partial base images, so that
tests[l](g) returns False if the element g is known not to satisfy
prop base on where g sends the first l + 1 base points.
init_subgroup
if a subgroup of the sought group is known in advance, it can be
passed to the function as this parameter.
Returns res
The subgroup of all elements satisfying prop. The generating set for
this group is guaranteed to be a strong generating set relative to the
base base.
Notes
This function is extremely lengthy and complicated and will require some careful
attention. The implementation is described in [1], pp. 114-117, and the comments
for the code here follow the lines of the pseudocode in the book for clarity.
The complexity is exponential in general, since the search process by itself visits
all members of the supergroup. However, there are a lot of tests which are used to
prune the search tree, and users can define their own tests via the tests parameter,
so in practice, and for some computations, it’s not terrible.
A crucial part in the procedure is the frequent base change performed (this is line
11 in the pseudocode) in order to obtain a new basic stabilizer. The book mentiones
that this can be done by using .baseswap(...), however the current implementation
uses a more straightforward way to find the next basic stabilizer - calling the function
.stabilizer(...) on the previous basic stabilizer.
Examples
sylow_subgroup(p)
Return a p-Sylow subgroup of the group.
The algorithm is described in [1], Chapter 4, Section 7
Examples
>>> D = DihedralGroup(6)
>>> S = D.sylow_subgroup(2)
>>> S.order()
4
>>> G = SymmetricGroup(6)
>>> S = G.sylow_subgroup(5)
>>> S.order()
5
>>> G1 = AlternatingGroup(3)
>>> G2 = AlternatingGroup(5)
>>> G3 = AlternatingGroup(9)
>>> S1 = G1.sylow_subgroup(3)
>>> S2 = G2.sylow_subgroup(3)
>>> S3 = G3.sylow_subgroup(3)
transitivity_degree
Compute the degree of transitivity of the group.
A permutation group G acting on Ω = {0, 1, ..., n − 1} is k-fold transitive, if, for any k
points (a1 , a2 , ..., ak ) ∈ Ω and any k points (b1 , b2 , ..., bk ) ∈ Ω there exists g ∈ G such that
g(a1 ) = b1 , g(a2 ) = b2 , ..., g(ak ) = bk The degree of transitivity of G is the maximum k
such that G is k-fold transitive. ([8])
See also:
is_transitive (page 254), orbit (page 258)
Examples
Polyhedron
class sympy.combinatorics.polyhedron.Polyhedron
Represents the polyhedral symmetry group (PSG).
The PSG is one of the symmetry groups of the Platonic solids. There are three polyhedral
groups: the tetrahedral group of order 12, the octahedral group of order 24, and the
icosahedral group of order 60.
All doctests have been given in the docstring of the constructor of the object.
References
http://mathworld.wolfram.com/PolyhedralGroup.html
array_form
Return the indices of the corners.
The indices are given relative to the original position of corners.
See also:
corners (page 268), cyclic_form (page 268)
Examples
>>> tetrahedron.rotate(0)
>>> tetrahedron.array_form
[0, 2, 3, 1]
>>> tetrahedron.pgroup[0].array_form
[0, 2, 3, 1]
corners
Get the corners of the Polyhedron.
The method vertices is an alias for corners.
See also:
array_form (page 267), cyclic_form (page 268)
Examples
cyclic_form
Return the indices of the corners in cyclic notation.
The indices are given relative to the original position of corners.
See also:
corners (page 268), array_form (page 267)
edges
Given the faces of the polyhedra we can get the edges.
Examples
faces
Get the faces of the Polyhedron.
pgroup
Get the permutations of the Polyhedron.
reset()
Return corners to their original positions.
Examples
rotate(perm)
Apply a permutation to the polyhedron in place. The permutation may be given as
a Permutation instance or an integer indicating which permutation from pgroup of
the Polyhedron should be applied.
This is an operation that is analogous to rotation about an axis by a fixed increment.
Notes
Examples
>>> cube.reset()
>>> cube.rotate(Permutation([[1, 2]], size=8))
>>> cube.corners
(0, 2, 1, 3, 4, 5, 6, 7)
Polyhedron can be used to follow elements of set that are identified by letters instead
of integers:
size
Get the number of corners of the Polyhedron.
vertices
Get the corners of the Polyhedron.
The method vertices is an alias for corners.
See also:
array_form (page 267), cyclic_form (page 268)
Examples
Prufer Sequences
class sympy.combinatorics.prufer.Prufer
The Prufer correspondence is an algorithm that describes the bijection between labeled
trees and the Prufer code. A Prufer code of a labeled tree is unique up to isomorphism
and has a length of n - 2.
Prufer sequences were first used by Heinz Prufer to give a proof of Cayley’s formula.
References
[R36]
static edges(*runs)
Return a list of edges and the number of nodes from the given runs that connect
nodes in an integer-labelled tree.
All node numbers will be shifted so that the minimum node is 0. It is not a problem
if edges are repeated in the runs; only unique edges are returned. There is no
assumption made about what the range of the node labels should be, but all nodes
from the smallest through the largest must be present.
Examples
next(delta=1)
Generates the Prufer sequence that is delta beyond the current one.
See also:
prufer_rank (page 272), rank (page 272), prev (page 271), size (page 273)
Examples
nodes
Returns the number of nodes in the tree.
Examples
prev(delta=1)
Generates the Prufer sequence that is -delta before the current one.
See also:
prufer_rank (page 272), rank (page 272), next (page 271), size (page 273)
Examples
prufer_rank()
Computes the rank of a Prufer sequence.
See also:
rank (page 272), next (page 271), prev (page 271), size (page 273)
Examples
prufer_repr
Returns Prufer sequence for the Prufer object.
This sequence is found by removing the highest numbered vertex, recording the
node it was attached to, and continuing until only two vertices remain. The Prufer
sequence is the list of recorded nodes.
See also:
to_prufer (page 273)
Examples
rank
Returns the rank of the Prufer sequence.
See also:
prufer_rank (page 272), next (page 271), prev (page 271), size (page 273)
Examples
size
Return the number of possible trees of this Prufer object.
See also:
prufer_rank (page 272), rank (page 272), next (page 271), prev (page 271)
Examples
static to_prufer(tree, n)
Return the Prufer sequence for a tree given as a list of edges where n is the number
of nodes in the tree.
See also:
Examples
static to_tree(prufer)
Return the tree (as a list of edges) of the given Prufer sequence.
See also:
References
• https://hamberg.no/erlend/posts/2010-11-06-prufer-sequence-compact-tree-representation.
html
Examples
tree_repr
Returns the tree representation of the Prufer object.
See also:
to_tree (page 273)
Examples
classmethod unrank(rank, n)
Finds the unranked Prufer sequence.
Examples
Subsets
class sympy.combinatorics.subsets.Subset
Represents a basic subset object.
We generate subsets using essentially two techniques, binary enumeration and lexico-
graphic enumeration. The Subset class takes two arguments, the first one describes the
initial subset to consider and the second describes the superset.
Examples
Examples
cardinality
Returns the number of all possible subsets.
See also:
subset (page 279), superset (page 279), size (page 278), superset_size
(page 280)
Examples
iterate_binary(k)
This is a helper function. It iterates over the binary subsets by k steps. This variable
can be both positive or negative.
See also:
next_binary (page 276), prev_binary (page 277)
Examples
iterate_graycode(k)
Helper function used for prev_gray and next_gray. It performs k step overs to get
the respective Gray codes.
See also:
next_gray (page 276), prev_gray (page 277)
Examples
next_binary()
Generates the next binary ordered subset.
See also:
prev_binary (page 277), iterate_binary (page 275)
Examples
next_gray()
Generates the next Gray code ordered subset.
See also:
iterate_graycode (page 275), prev_gray (page 277)
Examples
next_lexicographic()
Generates the next lexicographically ordered subset.
See also:
prev_lexicographic (page 277)
Examples
prev_binary()
Generates the previous binary ordered subset.
See also:
next_binary (page 276), iterate_binary (page 275)
Examples
prev_gray()
Generates the previous Gray code ordered subset.
See also:
iterate_graycode (page 275), next_gray (page 276)
Examples
prev_lexicographic()
Generates the previous lexicographically ordered subset.
See also:
next_lexicographic (page 276)
Examples
rank_binary
Computes the binary ordered rank.
See also:
iterate_binary (page 275), unrank_binary (page 280)
Examples
rank_gray
Computes the Gray code ranking of the subset.
See also:
iterate_graycode (page 275), unrank_gray (page 280)
Examples
rank_lexicographic
Computes the lexicographic ranking of the subset.
Examples
size
Gets the size of the subset.
See also:
subset (page 279), superset (page 279), superset_size (page 280), cardinality
(page 275)
Examples
subset
Gets the subset represented by the current instance.
See also:
superset (page 279), size (page 278), superset_size (page 280), cardinality
(page 275)
Examples
Examples
Examples
superset
Gets the superset of the subset.
See also:
subset (page 279), size (page 278), superset_size (page 280), cardinality
(page 275)
Examples
superset_size
Returns the size of the superset.
See also:
subset (page 279), superset (page 279), size (page 278), cardinality (page 275)
Examples
Examples
Examples
subsets.ksubsets(k)
Finds the subsets of size k in lexicographic order.
This uses the itertools generator.
See also:
class Subset
Examples
Gray Code
class sympy.combinatorics.graycode.GrayCode
A Gray code is essentially a Hamiltonian walk on a n-dimensional cube with edge length
of one. The vertices of the cube are represented by vectors whose values are binary.
The Hamilton walk visits each vertex exactly once. The Gray code for a 3d cube is
[‘000’,‘100’,‘110’,‘010’,‘011’,‘111’,‘101’, ‘001’].
A Gray code solves the problem of sequentially generating all possible subsets of n ob-
jects in such a way that each subset is obtained from the previous one by either deleting
or adding a single object. In the above example, 1 indicates that the object is present,
and 0 indicates that its absent.
Gray codes have applications in statistics as well when we want to compute various
statistics related to subsets in an efficient manner.
References: [1] Nijenhuis,A. and Wilf,H.S.(1978). Combinatorial Algorithms. Academic
Press. [2] Knuth, D. (2011). The Art of Computer Programming, Vol 4 Addison Wesley
Examples
current
Returns the currently referenced Gray code as a bit string.
Examples
generate_gray(**hints)
Generates the sequence of bit vectors of a Gray Code.
[1] Knuth, D. (2011). The Art of Computer Programming, Vol 4, Addison Wesley
See also:
skip (page 283)
Examples
n
Returns the dimension of the Gray code.
Examples
next(delta=1)
Returns the Gray code a distance delta (default = 1) from the current value in
canonical order.
Examples
rank
Ranks the Gray code.
Examples
selections
Returns the number of bit vectors in the Gray code.
Examples
skip()
Skips the bit generation.
See also:
generate_gray (page 282)
Examples
Examples
graycode.random_bitstring()
Generates a random bitlist of length n.
Examples
graycode.gray_to_bin()
Convert from Gray coding to binary coding.
We assume big endian encoding.
See also:
bin_to_gray (page 284)
Examples
graycode.bin_to_gray()
Convert from binary coding to gray coding.
We assume big endian encoding.
See also:
gray_to_bin (page 284)
Examples
graycode.get_subset_from_bitstring(bitstring)
Gets the subset defined by the bitstring.
See also:
graycode_subsets (page 285)
Examples
graycode.graycode_subsets()
Generates the subsets as enumerated by a Gray code.
See also:
get_subset_from_bitstring (page 285)
Examples
Named Groups
sympy.combinatorics.named_groups.SymmetricGroup(n)
Generates the symmetric group on n elements as a permutation group.
The generators taken are the n-cycle (0 1 2 ... n-1) and the transposition (0 1) (in
cycle notation). (See [1]). After the group is generated, some of its basic properties are
set.
See also:
CyclicGroup (page 286), DihedralGroup (page 286), AlternatingGroup (page 287)
References
[1] http://en.wikipedia.org/wiki/Symmetric_group#Generators_and_relations
Examples
sympy.combinatorics.named_groups.CyclicGroup(n)
Generates the cyclic group of order n as a permutation group.
The generator taken is the n-cycle (0 1 2 ... n-1) (in cycle notation). After the group
is generated, some of its basic properties are set.
See also:
SymmetricGroup (page 285), DihedralGroup (page 286), AlternatingGroup (page 287)
Examples
sympy.combinatorics.named_groups.DihedralGroup(n)
Generates the dihedral group Dn as a permutation group.
The dihedral group Dn is the group of symmetries of the regular n-gon. The generators
taken are the n-cycle a = (0 1 2 ... n-1) (a rotation of the n-gon) and b = (0 n-1)(1
n-2)... (a reflection of the n-gon) in cycle rotation. It is easy to see that these satisfy
a**n = b**2 = 1 and bab = ~a so they indeed generate Dn (See [1]). After the group
is generated, some of its basic properties are set.
See also:
SymmetricGroup (page 285), CyclicGroup (page 286), AlternatingGroup (page 287)
References
[1] http://en.wikipedia.org/wiki/Dihedral_group
Examples
sympy.combinatorics.named_groups.AlternatingGroup(n)
Generates the alternating group on n elements as a permutation group.
For n > 2, the generators taken are (0 1 2), (0 1 2 ... n-1) for n odd and (0 1 2),
(1 2 ... n-1) for n even (See [1], p.31, ex.6.9.). After the group is generated, some of
its basic properties are set. The cases n = 1, 2 are handled separately.
See also:
SymmetricGroup (page 285), CyclicGroup (page 286), DihedralGroup (page 286)
References
Examples
sympy.combinatorics.named_groups.AbelianGroup(*cyclic_orders)
Returns the direct product of cyclic groups with the given orders.
According to the structure theorem for finite abelian groups ([1]), every finite abelian
group can be written as the direct product of finitely many cyclic groups.
See also:
DirectProduct
References
[1] http://groupprops.subwiki.org/wiki/Structure_theorem_for_finitely_generated_
abelian_groups
Examples
Utilities
sympy.combinatorics.util._base_ordering(base, degree)
Order {0, 1, ..., n − 1} so that base points come first and in order.
Parameters ‘‘base‘‘ - the base
‘‘degree‘‘ - the degree of the associated permutation group
Returns A list base_ordering such that base_ordering[point] is the
number of point in the ordering.
Notes
This is used in backtrack searches, when we define a relation << on the underlying set
for a permutation group of degree n, {0, 1, ..., n − 1}, so that if (b1 , b2 , ..., bk ) is a base we
have bi << bj whenever i < j and bi << a for all i ∈ {1, 2, ..., k} and a is not in the base. The
idea is developed and applied to backtracking algorithms in [1], pp.108-132. The points
that are not in the base are taken in increasing order.
References
[1] Holt, D., Eick, B., O’Brien, E. “Handbook of computational group theory”
Examples
sympy.combinatorics.util._check_cycles_alt_sym(perm)
Checks for cycles of prime length p with n/2 < p < n-2.
Here n is the degree of the permutation. This is a helper function for the function
is_alt_sym from sympy.combinatorics.perm_groups.
See also:
sympy.combinatorics.perm_groups.PermutationGroup.is_alt_sym (page 251)
Examples
sympy.combinatorics.util._distribute_gens_by_base(base, gens)
Distribute the group elements gens by membership in basic stabilizers.
Notice that for a base (b1 , b2 , ..., bk ), the basic stabilizers are defined as G(i) = Gb1 ,...,bi−1 for
i ∈ {1, 2, ..., k}.
Parameters ‘‘base‘‘ - a sequence of points in ‘{0, 1, …, n-1}‘
‘‘gens‘‘ - a list of elements of a permutation group of degree ‘n‘.
Returns List of length k, where k is
the length of base. The i-th entry contains those elements in
gens which fix the first i elements of base (so that the
0-th entry is equal to gens itself). If no element fixes the first
i elements of base, the i-th element is set to a list containing
the identity element.
See also:
_strong_gens_from_distr (page 293), _orbits_transversals_from_bsgs (page 290),
_handle_precomputed_bsgs (page 290)
Examples
sympy.combinatorics.util._handle_precomputed_bsgs(base, strong_gens,
transversals=None,
basic_orbits=None,
strong_gens_distr=None)
Calculate BSGS-related structures from those present.
The base and strong generating set must be provided; if any of the transversals, basic
orbits or distributed strong generators are not provided, they will be calculated from the
base and strong generating set.
Parameters ‘‘base‘‘ - the base
‘‘strong_gens‘‘ - the strong generators
‘‘transversals‘‘ - basic transversals
‘‘basic_orbits‘‘ - basic orbits
‘‘strong_gens_distr‘‘ - strong generators distributed by membership
in basic
stabilizers
Returns (transversals, basic_orbits, strong_gens_distr) where
transversals
are the basic transversals, basic_orbits are the basic orbits, and
strong_gens_distr are the strong generators distributed by membership
in basic stabilizers.
See also:
_orbits_transversals_from_bsgs (page 290), distribute_gens_by_base
Examples
sympy.combinatorics.util._orbits_transversals_from_bsgs(base,
strong_gens_distr,
transver-
sals_only=False,
slp=False)
Compute basic orbits and transversals from a base and strong generating set.
The generators are provided as distributed across the basic stabilizers. If the optional
argument transversals_only is set to True, only the transversals are returned.
Parameters ‘‘base‘‘ - the base
‘‘strong_gens_distr‘‘ - strong generators distributed by membership
in basic
stabilizers
‘‘transversals_only‘‘ - a flag switching between returning only the
transversals/ both orbits and transversals
‘‘slp‘‘ - if ‘‘True‘‘, return a list of dictionaries containing the
generator presentations of the elements of the transversals, i.e. the
list of indices of generators from strongg ensd istr[i] such that their prod-
uct is the relevant transversal element
See also:
_distribute_gens_by_base (page 289), _handle_precomputed_bsgs (page 290)
Examples
Notes
References
[1] Holt, D., Eick, B., O’Brien, E. “Handbook of computational group theory”
Examples
Notes
The algorithm is described in [1],pp.89-90. The reason for returning both the current
state of the element being decomposed and the level at which the sifting ends is that
they provide important information for the randomized version of the Schreier-Sims al-
gorithm.
References
[1] Holt, D., Eick, B., O’Brien, E. “Handbook of computational group theory”
Examples
sympy.combinatorics.util._strong_gens_from_distr(strong_gens_distr)
Retrieve strong generating set from generators of basic stabilizers.
This is just the union of the generators of the first and second basic stabilizers.
Parameters ‘‘strong_gens_distr‘‘ - strong generators distributed by
membership in basic
stabilizers
See also:
_distribute_gens_by_base (page 289)
Examples
Group constructors
sympy.combinatorics.group_constructs.DirectProduct(*groups)
Returns the direct product of several groups as a permutation group.
This is implemented much like the __mul__ procedure for taking the direct product of
two permutation groups, but the idea of shifting the generators is realized in the case
of an arbitrary number of groups. A call to DirectProduct(G1, G2, …, Gn) is generally
expected to be faster than a call to G1*G2*…*Gn (and thus the need for this algorithm).
See also:
__mul__
Examples
Test Utilities
sympy.combinatorics.testutil._cmp_perm_lists(first, second)
Compare two lists of permutations as sets.
This is used for testing purposes. Since the array form of a permutation is currently a
list, Permutation is not hashable and cannot be put into a set.
Examples
Examples
Examples
Tensor Canonicalization
Examples
two types of indices [a, b, c, d, e, f ] and [m, n], in this order, both with commuting metric
f abc antisymmetric, commuting
Ama no symmetry, commuting
T = f c da ∗ f f eb ∗ Am d ∗ Amb ∗ An a ∗ Ane
ord = [c,f,a,-a,b,-b,d,-d,e,-e,m,-m,n,-n]
Algorithm
First one uses canonical_free to get the minimum tensor under lexicographic order, us-
ing only the slot symmetries. If the component tensors have not minimal BSGS, it is
attempted to find it; if the attempt fails canonicalize_naive is used instead.
Compute the residual slot symmetry keeping fixed the free indices using ten-
sor_gens(base, gens, list_free_indices, sym).
Reduce the problem eliminating the free indices.
Then use double_coset_can_rep and lift back the result reintroducing the free indices.
sympy.combinatorics.tensor_can.double_coset_can_rep(dummies, sym, b_S,
sgens, S_transversals, g)
Butler-Portugal algorithm for tensor canonicalization with dummy indices
dummies list of lists of dummy indices, one list for each type of index; the
dummy indices are put in order contravariant, covariant [d0, -d0, d1, -d1,
…].
sym list of the symmetries of the index metric for each type.
possible symmetries of the metrics
• 0 symmetric
• 1 antisymmetric
• None no symmetry
b_S base of a minimal slot symmetry BSGS.
sgens generators of the slot symmetry BSGS.
S_transversals transversals for the slot BSGS.
g permutation representing the tensor.
Return 0 if the tensor is zero, else return the array form of the permutation representing
the canonical form of the tensor.
A tensor with dummy indices can be represented in a number of equivalent ways which
typically grows exponentially with the number of indices. To be able to establish if two
tensors with many indices are equal becomes computationally very slow in absence of
an efficient algorithm.
Example in which the tensor is zero, same slot symmetries as above: T d3 d1,d2 d1 d3 d2
= −T d3 d1,d3 d1 d2 d2 under slot symmetry −(2, 4);
= Td3d1 d3d1 d2 d2 under slot symmetry −(0, 2);
= T d3 d1d3 d1 d2 d2 symmetric metric;
= 0 since two of these lines have tensors differ only for the sign.
The double coset D*g*S consists of permutations h = d ∗ g ∗ s corresponding to equivalent
tensors; if there are two h which are the same apart from the sign, return zero; otherwise
choose as representative the tensor with indices ordered lexicographically according to
[d1, −d1, d2, −d2, d3, −d3] that is rep = min(D ∗ g ∗ S) = min([d ∗ g ∗ sf ordinDf orsinS])
The indices are fixed one by one; first choose the lowest index for slot 0, then the lowest
remaining index for slot 1, etc. Doing this one obtains a chain of stabilizers
S− > Sb0 − > Sb0,b1 − > ... and D− > Dp0 − > Dp0,p1 − > ...
where [b0, b1, ...] = range(b) is a base of the symmetric group; the strong base bS of S is an
ordered sublist of it; therefore it is sufficient to compute once the strong base generators
of S using the Schreier-Sims algorithm; the stabilizers of the strong base generators are
the strong base generators of the stabilizer subgroup.
dbase = [p0, p1, ...] is not in general in lexicographic order, so that one must recompute
the strong base generators each time; however this is trivial, there is no need to use the
Schreier-Sims algorithm for D.
The algorithm keeps a TAB of elements (si , di , hi ) where hi = di ∗ g ∗ si satisfying hi [j] = pj
for 0 <= j < i starting from s0 = id, d0 = id, h0 = g.
The equations h0 [0] = p0 , h1 [1] = p1 , ... are solved in this order, choosing each time the
lowest possible value of p_i
For j < i di ∗ g ∗ si ∗ Sb0 ,...,bi−1 ∗ bj = Dp0 ,...,pi−1 ∗ pj so that for dx in Dp0 ,...,pi−1 and sx in
Sbase[0],...,base[i−1] one has dx ∗ di ∗ g ∗ si ∗ sx ∗ bj = pj
Search for dx, sx such that this equation holds for j = i; it can be written as si ∗ sx ∗ bj =
J, dx ∗ di ∗ g ∗ J = pj sx ∗ bj = si ∗ ∗ − 1 ∗ J; sx = trace(si ∗ ∗ − 1, Sb0 ,...,bi−1 ) dx ∗ ∗ − 1 ∗ pj = di ∗ g ∗ J; dx =
trace(di ∗ g ∗ J, Dp0 ,...,pi−1 )
si+1 = si ∗ trace(si ∗ ∗ − 1 ∗ J, Sb0 ,...,bi−1 ) di+1 = trace(di ∗ g ∗ J, Dp0 ,...,pi−1 ) ∗ ∗ − 1 ∗ di hi+1 ∗ bi =
di+1 ∗ g ∗ si+1 ∗ bi = pi
hn ∗ bj = pj for all j, so that hn is the solution.
Add the found (s, d, h) to TAB1.
At the end of the iteration sort TAB1 with respect to the h; if there are two consecutive
h in TAB1 which differ only for the sign, the tensor is zero, so return 0; if there are two
consecutive h which are equal, keep only one.
Then stabilize the slot generators under i and the dummy generators under pi .
Assign T AB = T AB1 at the end of the iteration step.
At the end T AB contains a unique (s, d, h), since all the slots of the tensor h have been
fixed to have the minimum value according to the symmetries. The algorithm returns h.
It is important that the slot BSGS has lexicographic minimal base, otherwise there is an
i which does not belong to the slot base for which pi is fixed by the dummy symmetry
only, while i is not invariant from the slot stabilizer, so pi is not in general the minimal
value.
This algorithm differs slightly from the original algorithm [3]: the canonical
form is minimal lexicographically, and the BSGS has minimal base under lexico-
graphic order. Equal tensors h are eliminated from TAB.
Examples
sympy.combinatorics.tensor_can.get_symmetric_group_sgs(n, antisym=False)
Return base, gens of the minimal BSGS for (anti)symmetric tensor
n rank of the tensor
antisym = False symmetric tensor antisym = True antisymmetric tensor
Examples
Examples
Introduction
This module presents the functionality designed for computing with finitely- presented groups
(fp-groups for short). The name of the corresponding SymPy object is FpGroup. The functions
or classes described here are studied under computational group theory. All code exam-
ples assume:
Overview of Facilities
The facilities provided for fp-groups fall into a number of natural groupings
• The construction of fp-groups using a free group and a list of words in generators of that
free group.
• Index determination using the famous Todd-Coxeter procedure.
• The construction of all subgroups having index less than some (small) specified positive
integer, using the Low-Index Subgroups algorithm.
• Algorithms for computing presentations of a subgroup of finite index in a group defined
by finite presentation.
For a description of fundamental algorithms of finitely presented groups we often make use
of Handbook of Computational Group Theory.
Finitely presented groups are constructed by factoring a free group by a set of relators. The
set of relators is taken in as a list of words in generators of free group in SymPy, using a list
provides ordering to the relators. If the list of relators is empty, the associated free group is
returned.
Example of construction of a finitely-presented group. The symmetric group of degree 4
may be represented as a two generator group with presentation ⟨a, b | a2 , b3 , (ab)4 ⟩. Giving the
relations as a list of relators, group in SymPy would be specified as:
Currently groups with relators having presentation like ⟨r, s, t | r2 , s2 , t2 , rst = str = trs⟩ will
have to be specified as:
Obviously this is not a unique way to make that particular group, but the point is that in case
of equality with non-identity the user has to manually do that.
creates a free group F of rank 2 and assigns the variables x and y to the two generators.
creates a free group F of rank 2, with tuple of generators F.generators, and inserts x and y
as generators into the global namespace.
Construction of words
This section is applicable to words of FreeGroup as well as FpGroup. When we say word in
SymPy, it actually means a reduced word , since the words are automatically reduced. Given
a group G defined on n generators x1 , x2 , x3 , . . . , xn , a word is constructed as sr11 sr22 · · · srkk where
si ∈ {x1 , x2 , . . . , xn } , ri ∈ Z for all k.
Each word can be constructed in a variety of ways, since after reduction they may be equiv-
alent.
This section describes the use of coset enumeration techniques in SymPy. The algorithm
used for coset enumeration procedure is Todd-Coxeter algorithm and is developed in Sympy
using [Ho05] and [CDHW73]. The reader should consult [CDHW73] and [Hav91] for a general
description of the algorithm.
We have two strategies of coset enumeration relator-based and coset-table based and the two
have been implemented as coset_enumeration_r, coset_enumeration_c respectively. The
two strategies differ in the way they make new definitions for the cosets.
Though from the user point of view it is suggested to rather use the .coset_enumeration
method of FpGroup and specify the strategy argument.
strategy: (default=”relator_based”) specifies the strategy of coset enumeration to be used,
possible values are “relator_based” or “coset_table_based”.
CosetTable
Class used to manipulate the information regarding the coset enumeration of the finitely
presented group G on the cosets of the subgroup H.
Basically a coset table CosetTable(G,H), is the permutation representation of the finitely
presented group on the cosets of a subgroup. Most of the set theoretic and group functions
use the regular representation of G, i.e., the coset table of G over the trivial subgroup.
The actual mathematical coset table is obtained using .table attribute and is a list of lists.
For each generator g of G it contains a column and the next column corresponds to g**-1 and
so on for other generators, so in total it has 2*G.rank() columns. Each column is simply a
list of integers. If l is the generator list for the generator g and if l[i] = j then generator g
takes the coset i to the coset j by multiplication from the right.
For finitely presented groups, a coset table is computed by a Todd-Coxeter coset enumeration.
Note that you may influence the performance of that enumeration by changing the values of
the variable CosetTable.coset_table_max_limit.
Attributes of CosetTable
built-in. This is controlled by this variable. To change it, use maxc osets keyword. For
example:
For any two entries i, j with i < j in coset table, the first occurrence of i in a coset table
precedes the first occurrence of j with respect to the usual row-wise ordering of the table
entries. We call such a table a standard coset table. To standardize a CosetTable we use the
.standardize method.
Note the method alters the given table, it does not create a copy.
The functionality in this section are concerned with the construction of subgroups of finite
index. We describe a method for computing all subgroups whose index does not exceed some
(modest) integer bound.
This returns the coset tables of subgroups of satisfying the property that index, index, of
subgroup in group is ≤ n.
Bibliography
class sympy.ntheory.generate.Sieve
An infinite list of prime numbers, implemented as a dynamically growing sieve of Eratos-
thenes. When a lookup is requested involving an odd number that has not been sieved,
the sieve is automatically extended up to that number.
Examples
extend(n)
Grow the sieve to cover all primes <= n (a real number).
Examples
extend_to_no(i)
Extend to include the ith prime number.
i must be an integer.
The list is extended by 50% if it is too short, so it is likely that it will be longer than
requested.
Examples
mobiusrange(a, b)
Generate all mobius numbers for the range [a, b).
Examples
primerange(a, b)
Generate all prime numbers in the range [a, b).
Examples
search(n)
Return the indices i, j of the primes that bound n.
If n is prime then i == j.
Although n can be an expression, if ceiling cannot convert it to an integer then an n
error will be raised.
Examples
totientrange(a, b)
Generate all totient numbers for the range [a, b).
Examples
sympy.ntheory.generate.prime(nth)
Return the nth prime, with the primes indexed as prime(1) = 2, prime(2) = 3, etc…. The
nth prime is approximately n*log(n).
Logarithmic integral of x is a pretty nice approximation for number of primes <= x, i.e.
li(x) ~ pi(x) In fact, for the numbers we are concerned about( x<1e11 ), li(x) - pi(x) <
50000
Also, li(x) > pi(x) can be safely assumed for the numbers which can be evaluated by this
function.
Here, we find the least integer m such that li(m) > n using binary search. Now pi(m-1)
< li(m-1) <= n,
We find pi(m - 1) using primepi function.
Starting from m, we have to find n - pi(m-1) more primes.
For the inputs this implementation can handle, we will have to test primality for at max
about 10**5 numbers, to get our answer.
See also:
References
• https://en.wikipedia.org/wiki/Prime_number_theorem#Table_of_.CF.80.28x.29.2C_
x_.2F_log_x.2C_and_li.28x.29
• https://en.wikipedia.org/wiki/Prime_number_theorem#Approximations_for_the_
nth_prime_number
• https://en.wikipedia.org/wiki/Skewes%27_number
Examples
sympy.ntheory.generate.primepi(n)
Return the value of the prime counting function pi(n) = the number of prime numbers
less than or equal to n.
Algorithm Description:
In sieve method, we remove all multiples of prime p except p itself.
Let phi(i,j) be the number of integers 2 <= k <= i which remain after sieving from primes
less than or equal to j. Clearly, pi(n) = phi(n, sqrt(n))
If j is not a prime, phi(i,j) = phi(i, j - 1)
if j is a prime, We remove all numbers(except j) whose smallest prime factor is j.
Let x= j*a be such a number, where 2 <= a<= i / j Now, after sieving from primes <= j
- 1, a must remain (because x, and hence a has no prime factor <= j - 1) Clearly, there
are phi(i / j, j - 1) such a which remain on sieving from primes <= j - 1
Now, if a is a prime less than equal to j - 1, x= j*a has smallest prime factor = a, and has
already been removed(by sieving from a). So, we don’t need to remove it again. (Note:
there will be pi(j - 1) such x)
Thus, number of x, that will be removed are: phi(i / j, j - 1) - phi(j - 1, j - 1) (Note that pi(j
- 1) = phi(j - 1, j - 1))
=> phi(i,j) = phi(i, j - 1) - phi(i / j, j - 1) + phi(j - 1, j - 1)
So,following recursion is used and implemented as dp:
phi(a, b) = phi(a, b - 1), if b is not a prime phi(a, b) = phi(a, b-1)-phi(a / b, b-1) + phi(b-1,
b-1), if b is prime
Clearly a is always of the form floor(n / k), which can take at most 2*sqrt(n) values. Two
arrays arr1,arr2 are maintained arr1[i] = phi(i, j), arr2[i] = phi(n // i, j)
Finally the answer is arr2[1]
See also:
Examples
sympy.ntheory.generate.nextprime(n, ith=1)
Return the ith prime greater than n.
i must be an integer.
See also:
Notes
Potential primes are located at 6*j +/- 1. This property is used during searching.
sympy.ntheory.generate.prevprime(n)
Return the largest prime smaller than n.
See also:
Notes
Potential primes are located at 6*j +/- 1. This property is used during searching.
sympy.ntheory.generate.primerange(a, b)
Generate a list of all prime numbers in the range [a, b).
If the range exists in the default sieve, the values will be returned from there; otherwise
values will be returned but will not modify the sieve.
See also:
Notes
Some famous conjectures about the occurrence of primes in a given range are [1]:
• Twin primes: though often not, the following will give 2 primes
an infinite number of times: primerange(6*n - 1, 6*n + 2)
• Legendre’s: the following always yields at least one prime primerange(n**2,
(n+1)**2+1)
• Bertrand’s (proven): there is always a prime in the range primerange(n,
2*n)
• Brocard’s: there are at least four primes in the range
primerange(prime(n)**2, prime(n+1)**2)
The average gap between primes is log(n) [2]; the gap between primes can be arbitrarily
large since sequences of composite numbers are arbitrarily large, e.g. the numbers in
the sequence n! + 2, n! + 3 … n! + n are all composite.
References
1. http://en.wikipedia.org/wiki/Prime_number
2. http://primes.utm.edu/notes/gaps.html
Examples
The Sieve method, primerange, is generally faster but it will occupy more memory as
the sieve stores values. The default instance of Sieve, named sieve, can be used:
sympy.ntheory.generate.randprime(a, b)
Return a random prime number in the range [a, b).
Bertrand’s postulate assures that randprime(a, 2*a) will always succeed for a > 1.
See also:
References
• http://en.wikipedia.org/wiki/Bertrand’s_postulate
Examples
sympy.ntheory.generate.primorial(n, nth=True)
Returns the product of the first n primes (default) or the primes less than or equal to n
(when nth=False).
One can argue that the primes are infinite since if you take a set of primes and multiply
them together (e.g. the primorial) and then add or subtract 1, the result cannot be
divided by any of the original factors, hence either 1 or more new primes must divide
this product of primes.
In this case, the number itself is a new prime:
>>> factorint(primorial(4) + 1)
{211: 1}
>>> factorint(primorial(4) - 1)
{11: 1, 19: 1}
Here, some primes smaller and larger than the primes multiplied together are obtained:
See also:
cycle (lambda) and the length of terms before the cycle begins (mu); if values is True
then the terms of the sequence will be returned instead. The sequence is started with
value x0.
Note: more than the first lambda + mu terms may be returned and this is the cost of
cycle detection with Brent’s method; there are, however, generally less terms calculated
than would have been calculated if the proper ending point were determined, e.g. by
using Floyd’s method.
A function is defined:
sympy.ntheory.generate.composite(nth)
Return the nth composite number, with the composite numbers indexed as composite(1)
= 4, composite(2) = 6, etc….
See also:
Examples
sympy.ntheory.generate.compositepi(n)
Return the number of positive composite numbers less than or equal to n. The first
positive composite is 4, i.e. compositepi(4) = 1.
See also:
Examples
sympy.ntheory.factor_.smoothness(n)
Return the B-smooth and B-power smooth values of n.
The smoothness of n is the largest prime factor of n; the power- smoothness is the largest
divisor raised to its multiplicity.
See also:
factorint (page 319), smoothness_p (page 313)
This string can also be generated directly from a factorization dictionary and vice versa:
>>> factorint(17*9)
{3: 2, 17: 1}
>>> smoothness_p(_)
'p**i=3**2 has p-1 B=2, B-pow=2\np**i=17**1 has p-1 B=2, B-pow=16'
>>> smoothness_p(_)
{3: 2, 17: 1}
Visual
See also:
factorint (page 319), smoothness (page 313)
sympy.ntheory.factor_.trailing(n)
Count the number of trailing zero digits in the binary representation of n, i.e. determine
the largest power of 2 that divides n.
Examples
sympy.ntheory.factor_.multiplicity(p, n)
Find the greatest integer m such that p**m divides n.
Examples
Examples
e.g. 1, 2, 3, 4, 5, 3, 4, 5 – this leader and loop look a bit like the Greek letter rho, and
thus the name, ‘rho’.
For a given function, very different leader-loop values can be obtained so it is a good
idea to allow for retries:
>>> x=2
>>> for i in range(9):
... x=(x**2+12)%17
... print(x)
...
16
13
11
14
4
11
14
4
11
>>> next(cycle_length(lambda x: (x**2+12)%17, 2))
(3, 2)
>>> list(cycle_length(lambda x: (x**2+12)%17, 2, values=True))
[16, 13, 11, 14, 4]
Instead of checking the differences of all generated values for a gcd with n, only the kth
and 2*kth numbers are checked, e.g. 1st and 2nd, 2nd and 4th, 3rd and 6th until it has
been detected that the loop has been traversed. Loops may be many thousands of steps
long before rho finds a factor or reports failure. If max_steps is specified, the iteration
is cancelled with a failure after the specified number of steps.
References
• Richard Crandall & Carl Pomerance (2005), “Prime Numbers: A Computational Per-
spective”, Springer, 2nd edition, 229-231
Examples
If retries is > 0 then perhaps the problem will correct itself when new values are gener-
ated for a:
But if the value of a is changed we find that only multiples of 257 work, e.g.:
Checking different a values shows that all the ones that didn’t work had a gcd value not
equal to n but equal to one of the factors:
>>> aM = pow(255, M, n)
>>> [(d, aM%Pow(*d.args)) for d in factorint(n, visual=True).args]
[(257**1, 1), (1009**1, 1)]
No, only one of them. So perhaps the principle is that a root will be found for a given
value of B provided that:
1) the power smoothness of the p - 1 value next to the root does not exceed B
2) a**M % p != 1 for any of the divisors of n.
By trying more than one a it is possible that one of them will yield a factor.
References
• Richard Crandall & Carl Pomerance (2005), “Prime Numbers: A Computational Per-
spective”, Springer, 2nd edition, 236-238
• http://modular.math.washington.edu/edu/2007/spring/ent/ent-html/node81.html
• http://www.cs.toronto.edu/~yuvalf/Factorization.pdf
Examples
The B and B-pow are the same for the p - 1 factorizations of the divisors because those
factorizations had a very large prime factor:
>>> factorint(4410317 - 1)
{2: 2, 617: 1, 1787: 1}
>>> factorint(4869863-1)
{2: 1, 2434931: 1}
Note that until B reaches the B-pow value of 1787, the number is not cracked;
The B value has to do with the factors of the number next to the divisor, not the divi-
sors themselves. A worst case scenario is that the number next to the factor p has a
large prime divisisor or is a perfect power. If these conditions apply then the power-
smoothness will be about p/2 or p. The more realistic is that there will be a large prime
factor next to p requiring a B value on the order of p/2. Although primes may have been
searched for up to this level, the p/2 is a factor of p - 1, something that we don’t know.
The modular.math reference below states that 15% of numbers in the range of 10**15
to 15**15 + 10**4 are 10**6 power smooth so a B of 10**6 will fail 85% of the time in
that range. From 10**8 to 10**8 + 10**3 the percentages are nearly reversed…but in
that range the simple trial division is quite fast.
sympy.ntheory.factor_.factorint(n, limit=None, use_trial=True, use_rho=True,
use_pm1=True, verbose=False, visual=None,
multiple=False)
Given a positive integer n, factorint(n) returns a dict containing the prime factors of
n as keys and their respective multiplicities as values. For example:
This number has a small factor and a residual perfect power whose base is greater than
the limit:
List of Factors:
If multiple is set to True then a list containing the prime factors including multiplicities
is returned.
Visual Factorization:
If visual is set to True, then it will return a visual factorization of the integer. For
example:
Note that this is achieved by using the evaluate=False flag in Mul and Pow. If you do other
manipulations with an expression where evaluate=False, it may evaluate. Therefore, you
should use the visual option only for visualization, and use the normal dictionary returned
by visual=False if you want to perform operations on the factors.
You can easily switch between the two forms by sending them back to factorint:
If you want to send a number to be factored in a partially factored form you can do so
with a dictionary or unevaluated expression:
See also:
smoothness (page 313), smoothness_p (page 313), divisors (page 322)
Notes
Algorithm:
The function switches between multiple algorithms. Trial division quickly finds small
factors (of the order 1-5 digits), and finds all large factors if given enough time. The
Pollard rho and p-1 algorithms are used to find large factors ahead of time; they will
often find factors of the order of 10 digits within a few seconds:
Any of these methods can optionally be disabled with the following boolean parameters:
• use_trial: Toggle use of trial division
• use_rho: Toggle use of Pollard’s rho method
• use_pm1: Toggle use of Pollard’s p-1 method
factorint also periodically checks if the remaining part is a prime number or a perfect
power, and in those cases stops.
For unevaluated factorial, it uses Legendre’s formula(theorem).
If verbose is set to True, detailed progress is printed.
sympy.ntheory.factor_.primefactors(n, limit=None, verbose=False)
Return a sorted list of n’s prime factors, ignoring multiplicity and any composite factor
that remains if the limit was set too low for complete factorization. Unlike factorint(),
primefactors() does not return -1 or 0.
See also:
divisors (page 322)
Examples
>>> sorted(factorint(123456).items())
[(2, 6), (3, 1), (643, 1)]
>>> primefactors(123456)
[2, 3, 643]
sympy.ntheory.factor_.divisors(n, generator=False)
Return all divisors of n sorted from 1..n by default. If generator is True an unordered
generator is returned.
The number of divisors of n can be quite large if there are many prime factors (counting
repeated factors). If only the number of factors is desired use divisor_count(n).
See also:
primefactors (page 321), factorint (page 319), divisor_count (page 322)
Examples
References
• http://www.mayer.dial.pipex.com/maths/formulae.htm
sympy.ntheory.factor_.udivisors(n, generator=False)
Return all unitary divisors of n sorted from 1..n by default. If generator is True an un-
ordered generator is returned.
The number of unitary divisors of n can be quite large if there are many prime factors.
If only the number of unitary divisors is desired use udivisor_count(n).
See also:
primefactors (page 321), factorint (page 319), divisors (page 322), divisor_count
(page 322), udivisor_count (page 323)
References
• http://en.wikipedia.org/wiki/Unitary_divisor
• http://mathworld.wolfram.com/UnitaryDivisor.html
Examples
sympy.ntheory.factor_.udivisor_count(n)
Return the number of unitary divisors of n.
See also:
factorint (page 319), divisors (page 322), udivisors (page 323), divisor_count
(page 322), totient (page 324)
References
• http://mathworld.wolfram.com/UnitaryDivisorFunction.html
sympy.ntheory.factor_.antidivisors(n, generator=False)
Return all antidivisors of n sorted from 1..n by default.
Antidivisors [R433] of n are numbers that do not divide n by the largest possible margin.
If generator is True an unordered generator is returned.
See also:
primefactors (page 321), factorint (page 319), divisors (page 322), divisor_count
(page 322), antidivisor_count (page 324)
References
[R433]
Examples
sympy.ntheory.factor_.antidivisor_count(n)
Return the number of antidivisors [R434] of n.
See also:
factorint (page 319), divisors (page 322), antidivisors (page 323), divisor_count
(page 322), totient (page 324)
References
[R434]
Examples
class sympy.ntheory.factor_.totient
Calculate the Euler totient function phi(n)
totient(n) or ϕ(n) is the number of positive integers ≤ n that are relatively prime to n.
See also:
divisor_count (page 322)
References
[R435], [R436]
Examples
class sympy.ntheory.factor_.reduced_totient
Calculate the Carmichael reduced totient function lambda(n)
reduced_totient(n) or λ(n) is the smallest m > 0 such that k m ≡ 1 mod n for all k
relatively prime to n.
See also:
totient (page 324)
References
[R437], [R438]
Examples
class sympy.ntheory.factor_.divisor_sigma
Calculate the divisor function σk (n) for positive integer n
divisor_sigma(n, k) is equal to sum([x**k for x in divisors(n)])
If n’s prime factorization is:
∏
ω
n= pm
i ,
i
i=1
then
∏
ω
i + · · · + pi
mi k
σk (n) = (1 + pki + p2k ).
i=1
References
[R439]
Examples
class sympy.ntheory.factor_.udivisor_sigma
Calculate the unitary divisor function σk∗ (n) for positive integer n
udivisor_sigma(n, k) is equal to sum([x**k for x in udivisors(n)])
If n’s prime factorization is:
∏
ω
n= pm
i ,
i
i=1
then
∏
ω
σk∗ (n) = (1 + pm
i
ik
).
i=1
References
[R440]
Examples
sympy.ntheory.factor_.core(n, t=2)
Calculate core(n, t) = coret (n) of a positive integer n
core_2(n) is equal to the squarefree part of n
If n’s prime factorization is:
∏
ω
n= pm
i ,
i
i=1
then
∏
ω
mod t
coret (n) = pm
i
i
.
i=1
References
[R441]
Examples
sympy.ntheory.factor_.digits(n, b=10)
Return a list of the digits of n in base b. The first element in the list is b (or -b if n is
negative).
Examples
class sympy.ntheory.factor_.primenu
Calculate the number of distinct prime factors for a positive integer n.
If n’s prime factorization is:
∏
k
n= pm
i ,
i
i=1
See also:
factorint (page 319)
References
[R442]
Examples
class sympy.ntheory.factor_.primeomega
Calculate the number of prime factors counting multiplicities for a positive integer n.
If n’s prime factorization is:
∏
k
n= pm
i ,
i
i=1
See also:
factorint (page 319)
References
[R443]
Examples
sympy.ntheory.modular.symmetric_residue(a, m)
Return the residual mod m such that it is within half of the modulus.
If the moduli are not co-prime, you may receive an incorrect result if you use
check=False:
Note: the order of gf_crt’s arguments is reversed relative to crt, and that
solve_congruence takes residue, modulus pairs.
Programmer’s note: rather than checking that all pairs of moduli share no GCD (an
O(n**2) test) and rather than factoring all moduli and seeing that there is no factor in
common, a check that the result gives the indicated residuals is performed – an O(n)
operation.
See also:
solve_congruence (page 330)
sympy.polys.galoistools.gf_crt (page 981) low level crt routine used by this rou-
tine
sympy.ntheory.modular.crt1(m)
First part of Chinese Remainder Theorem, for multiple application.
Examples
Examples
sympy.ntheory.modular.solve_congruence(*remainder_modulus_pairs, **hint)
Compute the integer n that has the residual ai when it is divided by mi where the ai
and mi are given as pairs to this function: ((a1, m1), (a2, m2), …). If there is no solution,
return None. Otherwise return n and its modulus.
The mi values need not be co-prime. If it is known that the moduli are not co-prime then
the hint check can be set to False (default=True) and the check for a quicker solution
via crt() (valid when the moduli are co-prime) will be skipped.
If the hint symmetric is True (default is False), the value of n will be within 1/2 of the
modulus, possibly negative.
See also:
crt (page 329) high level routine implementing the Chinese Remainder Theorem
Examples
If you prefer to work with all remainder in one list and all moduli in another, send the
arguments like this:
The moduli need not be co-prime; in this case there may or may not be a solution:
The symmetric flag will make the result be within 1/2 of the modulus:
sympy.ntheory.multinomial.binomial_coefficients(n)
Return a dictionary containing pairs (k1, k2) : Ck n where Ck n are binomial coefficients and
n = k1 + k2. Examples ========
See also:
binomial_coefficients_list (page 331), multinomial_coefficients (page 331)
sympy.ntheory.multinomial.binomial_coefficients_list(n)
Return a list of binomial coefficients as rows of the Pascal’s triangle.
See also:
binomial_coefficients (page 331), multinomial_coefficients (page 331)
Examples
sympy.ntheory.multinomial.multinomial_coefficients(m, n)
Return a dictionary containing pairs {(k1,k2,..,km) : C_kn} where C_kn are multino-
mial coefficients such that n=k1+k2+..+km.
For example:
Code contributed to Sage by Yann Laigle-Chapuy, copied with permission of the author.
See also:
binomial_coefficients_list (page 331), binomial_coefficients (page 331)
sympy.ntheory.multinomial.multinomial_coefficients_iterator(m, n, _tu-
ple=<class
’tuple’>)
multinomial coefficient iterator
This routine has been optimized for m large with respect to n by taking advantage of the
fact that when the monomial tuples t are stripped of zeros, their coefficient is the same
as that of the monomial tuples from multinomial_coefficients(n, n). Therefore, the
latter coefficients are precomputed to save memory and time.
Examples
sympy.ntheory.partitions_.npartitions(n, verbose=False)
Calculate the partition function P(n), i.e. the number of ways that n can be written as a
sum of positive integers.
P(n) is computed using the Hardy-Ramanujan-Rademacher formula [R444].
The correctness of this implementation has been tested through 10**10.
References
[R444]
Examples
sympy.ntheory.primetest.mr(n, bases)
Perform a Miller-Rabin strong pseudoprime test on n using a given list of
bases/witnesses.
References
• Richard Crandall & Carl Pomerance (2005), “Prime Numbers: A Computational Per-
spective”, Springer, 2nd edition, 135-138
A list of thresholds and the bases they require are here: http://en.wikipedia.org/wiki/
Miller%E2%80%93Rabin_primality_test#Deterministic_variants_of_the_test
Examples
sympy.ntheory.primetest.isprime(n)
Test if n is a prime number (True) or not (False). For n < 2^64 the answer is definitive;
larger n values have a small probability of actually being pseudoprimes.
Negative numbers (e.g. -2) are not considered prime.
The first step is looking for trivial factors, which if found enables a quick return. Next,
if the sieve is large enough, use bisection search on the sieve. For small numbers, a
set of deterministic Miller-Rabin tests are performed with bases that are known to have
no counterexamples in their range. Finally if the number is larger than 2^64, a strong
BPSW test is performed. While this is a probable prime test and we believe counterex-
amples exist, there are no known counterexamples.
See also:
References
• http://en.wikipedia.org/wiki/Strong_pseudoprime
Examples
sympy.ntheory.residue_ntheory.n_order(a, n)
Returns the order of a modulo n.
The order of a modulo n is the smallest integer k such that a**k leaves a remainder of 1
with n.
Examples
sympy.ntheory.residue_ntheory.is_primitive_root(a, p)
Returns True if a is a primitive root of p
a is said to be the primitive root of p if gcd(a, p) == 1 and totient(p) is the smallest
positive number s.t.
a**totient(p) cong 1 mod(p)
Examples
sympy.ntheory.residue_ntheory.primitive_root(p)
Returns the smallest primitive root or None
Parameters p : positive integer
References
[R445], [R446]
Examples
sympy.ntheory.residue_ntheory.sqrt_mod(a, p, all_roots=False)
Find a root of x**2 = a mod p
Parameters a : integer
p : positive integer
all_roots : if True the list of roots is returned or None
Notes
If there is no root it is returned None; else the returned root is less or equal to p // 2;
in general is not the smallest one. It is returned p // 2 only if it is the only root.
Use all_roots only when it is expected that all the roots fit in memory; otherwise use
sqrt_mod_iter.
Examples
sympy.ntheory.residue_ntheory.quadratic_residues(p)
Returns the list of quadratic residues.
Examples
sympy.ntheory.residue_ntheory.nthroot_mod(a, n, p, all_roots=False)
Find the solutions to x**n = a mod p
Parameters a : integer
n : positive integer
p : positive integer
all_roots : if False returns the smallest root, else the list of roots
Examples
sympy.ntheory.residue_ntheory.is_nthpow_residue(a, n, m)
Returns True if x**n == a (mod m) has solutions.
References
[R447]
sympy.ntheory.residue_ntheory.is_quad_residue(a, p)
Returns True if a (mod p) is in the set of squares mod p, i.e a % p in set([i**2 % p for i in
range(p)]). If p is an odd prime, an iterative method is used to make the determination:
See also:
legendre_symbol (page 336), jacobi_symbol (page 336)
sympy.ntheory.residue_ntheory.legendre_symbol(a, p)
Returns the Legendre symbol (a/p).
For an integer a and an odd prime p, the Legendre symbol is defined as
( ) 0 if p divides a
a
= 1 if a is a quadratic residue modulo p
p
−1 if a is a quadratic nonresidue modulo p
Parameters a : integer
p : odd prime
See also:
is_quad_residue (page 336), jacobi_symbol (page 336)
Examples
sympy.ntheory.residue_ntheory.jacobi_symbol(m, n)
Returns the Jacobi symbol (m/n).
For any integer m and any positive odd integer n the Jacobi symbol is defined as the
product of the Legendre symbols corresponding to the prime factors of n:
( ) ( )α1 ( )α2 ( )αk
m m m m αk
= ... k where n = pα1 α2
1 p2 ...pk
n p1 p2 p
( )
Like the Legendre symbol, if the Jacobi symbol mn = −1 then m is a quadratic nonresidue
modulo n.
( )
But, unlike the Legendre symbol, if the Jacobi symbol mn = 1 then m may or may not be
a quadratic residue modulo n.
Parameters m : integer
n : odd positive integer
See also:
is_quad_residue (page 336), legendre_symbol (page 336)
Examples
>>> L = legendre_symbol
>>> S(45).factors()
{3: 2, 5: 1}
>>> jacobi_symbol(7, 45) == L(7, 3)**2 * L(7, 5)**1
True
sympy.ntheory.residue_ntheory.discrete_log(n, a, b, order=None,
prime_order=None)
Compute the discrete logarithm of a to the base b modulo n.
This is a recursive function to reduce the discrete logarithm problem in cyclic groups of
composite order to the problem in cyclic groups of prime order.
It employs different algorithms depending on the problem (subgroup order size, prime
order or not):
• Trial multiplication
• Baby-step giant-step
• Pollard’s Rho
• Pohlig-Hellman
References
[R448], [R449]
Examples
sympy.ntheory.continued_fraction.continued_fraction_convergents(cf)
Return an iterator over the convergents of a continued fraction (cf).
The parameter should be an iterable returning successive partial quotients of the con-
tinued fraction, such as might be returned by continued_fraction_iterator. In computing
the convergents, the continued fraction need not be strictly in canonical form (all inte-
gers, all but the first positive). Rational and negative elements may be present in the
expansion.
See also:
continued_fraction_iterator (page 338)
Examples
>>> it = continued_fraction_convergents(continued_fraction_iterator(pi))
>>> for n in range(7):
... print(next(it))
3
22/7
333/106
355/113
103993/33102
104348/33215
208341/66317
sympy.ntheory.continued_fraction.continued_fraction_iterator(x)
Return continued fraction expansion of x as iterator.
References
[R450]
Examples
sympy.ntheory.continued_fraction.continued_fraction_periodic(p, q, d=0)
Find the periodic continued fraction expansion of a quadratic irrational.
Compute
√
the continued fraction expansion of a rational or a quadratic irrational number,
i.e. q , where p, q and d ≥ 0 are integers.
p+ d
Returns the continued fraction representation (canonical form) as a list of integers, op-
tionally ending (for quadratic irrationals) with repeating block as the last term of this
list.
Parameters p : int
the rational part of the number’s numerator
q : int
the denominator of the number
d : int, optional
the irrational part (discriminator) of the number’s numerator
See also:
continued_fraction_iterator (page 338), continued_fraction_reduce (page 340)
References
[R451], [R452]
Examples
>>> continued_fraction_periodic(1, 2, 5)
[[1]]
If the discriminator is zero or a perfect square then the number will be a rational number:
>>> continued_fraction_periodic(4, 3, 0)
[1, 3]
>>> continued_fraction_periodic(4, 3, 49)
[3, 1, 2]
sympy.ntheory.continued_fraction.continued_fraction_reduce(cf)
Reduce a continued fraction to a rational or quadratic irrational.
Compute the rational or quadratic irrational number from its terminating or periodic
continued fraction expansion. The continued fraction expansion (cf) should be supplied
as a terminating iterator supplying the terms of the expansion. For terminating contin-
ued fractions, this is equivalent to list(continued_fraction_convergents(cf))[-1],
only a little more efficient. If the expansion has a repeating part, a list of the repeat-
ing terms should be returned as the last element from the iterator. This is the format
returned by continued_fraction_periodic.
For quadratic irrationals, returns the largest solution found, which is generally the one
sought, if the fraction is in canonical form (all terms positive except possibly the first).
See also:
continued_fraction_periodic (page 339)
Examples
class sympy.ntheory.mobius
Möbius function maps natural number to {-1, 0, 1}
It is defined as follows:
1) 1 if n = 1.
References
[R453], [R454]
Examples
sympy.ntheory.egyptian_fraction.egyptian_fraction(r, algorithm=’Greedy’)
Return the list of denominators of an Egyptian fraction expansion [R455] of the said
rational r.
Parameters r : Rational
a positive rational number.
algorithm : { “Greedy”, “Graham Jewett”, “Takenouchi”, “Golomb” }, op-
tional
Denotes the algorithm to be used (the default is “Greedy”).
See also:
sympy.core.numbers.Rational (page 145)
Notes
References
Examples
Warning: This module is intended for educational purposes only. Do not use the functions
in this module for real cryptographic applications. If you wish to encrypt real data, we
recommend using something like the cryptography module.
Encryption is the process of hiding a message and a cipher is a means of doing so. Included
in this module are both block and stream ciphers:
• Shift cipher
• Affine cipher
• substitution ciphers
• Vigenere’s cipher
• Hill’s cipher
• Bifid ciphers
• RSA
• Kid RSA
• linear-feedback shift registers (for stream ciphers)
• ElGamal encryption
In a substitution cipher “units” (not necessarily single characters) of plaintext are replaced
with ciphertext according to a regular system.
A transposition cipher is a method of encryption by which the positions held by “units” of
plaintext are replaced by a permutation of the plaintext. That is, the order of the units is
changed using a bijective function on the position of the characters to perform the encryption.
A monoalphabetic cipher uses fixed substitution over the entire message, whereas a polyal-
phabetic cipher uses a number of substitutions at different times in the message.
sympy.crypto.crypto.AZ(s=None)
Return the letters of s in uppercase. In case more than one string is passed, each of
them will be processed and a list of upper case strings will be returned.
See also:
check_and_join (page 344)
Examples
omitting characters in key that are not in symbols. A ValueError is raised if a) there are
duplicate characters in symbols or b) there are characters in key that are not in symbols.
Examples
Examples
sympy.crypto.crypto.cycle_list(k, n)
Returns the elements of the list range(n) shifted to the left by k (so the list starts with
k (mod n)).
Examples
Notes
The shift cipher is also called the Caesar cipher, after Julius Caesar, who, according
to Suetonius, used it with a shift of three to protect messages of military significance.
Caesar’s nephew Augustus reportedly used a similar cipher, but with a right shift of 1.
ALGORITHM:
INPUT:
key: an integer (the secret key)
msg: plaintext of upper-case letters
OUTPUT:
ct: ciphertext of upper-case letters
STEPS:
0. Number the letters of the alphabet from 0, …, N
1. Compute from the string msg a list L1 of corresponding integers.
2. Compute from the list L1 a new list L2, given by adding (k mod 26) to
each element in L1.
3. Compute from the list L2 a string ct of corresponding letters.
Examples
There is also a convenience function that does this with the original key:
>>> decipher_shift(ct, 1)
'GONAVYBEATARMY'
Examples
>>> decipher_shift(ct, 1)
'GONAVYBEATARMY'
Notes
This is a straightforward generalization of the shift cipher with the added complexity of
requiring 2 characters to be deciphered in order to recover the key.
ALGORITHM:
INPUT:
msg: string of characters that appear in symbols
a, b: a pair integers, with gcd(a, N) = 1 (the secret key)
symbols: string of characters (default = uppercase letters). When no
symbols are given, msg is converted to upper case letters and all other
charactes are ignored.
OUTPUT:
ct: string of characters (the ciphertext message)
STEPS:
0. Number the letters of the alphabet from 0, …, N
1. Compute from the string msg a list L1 of corresponding integers.
2. Compute from the list L1 a new list L2, given by replacing x by a*x + b
(mod N), for each element x in L1.
3. Compute from the list L2 a string ct of corresponding letters.
Examples
Notes
This is a more general than the affine cipher in that the key can only be recovered by
determining the mapping for each symbol. Though in practice, once a few symbols are
recognized the mappings for other characters can be quickly guessed.
Examples
In the special case where old and new are a permutation of order 2 (representing a
transposition of characters) their order is immaterial:
The substitution cipher, in general, is a method whereby “units” (not necessarily single
characters) of plaintext are replaced with ciphertext according to a regular system.
Notes
The Vigenère cipher is named after Blaise de Vigenère, a sixteenth century diplomat
and cryptographer, by a historical accident. Vigenère actually invented a different and
more complicated cipher. The so-called Vigenère cipher was actually invented by Giovan
Batista Belaso in 1553.
This cipher was used in the 1800’s, for example, during the American Civil War. The
Confederacy used a brass cipher disk to implement the Vigenère cipher (now on display
in the NSA Museum in Fort Meade) [R78].
The Vigenère cipher is a generalization of the shift cipher. Whereas the shift cipher
shifts each letter by the same amount (that amount being the key of the shift cipher) the
Vigenère cipher shifts a letter by an amount determined by the key (which is a word or
phrase known only to the sender and receiver).
For example, if the key was a single letter, such as “C”, then the so-called Vigenere cipher
is actually a shift cipher with a shift of 2 (since “C” is the 2nd letter of the alphabet, if
you start counting at 0). If the key was a word with two letters, such as “CA”, then the so-
called Vigenère cipher will shift letters in even positions by 2 and letters in odd positions
are left alone (shifted by 0, since “A” is the 0th letter, if you start counting at 0).
ALGORITHM:
INPUT:
msg: string of characters that appear in symbols (the plaintext)
key: a string of characters that appear in symbols (the secret key)
symbols: a string of letters defining the alphabet
OUTPUT:
ct: string of characters (the ciphertext message)
STEPS:
0. Number the letters of the alphabet from 0, …, N
1. Compute from the string key a list L1 of corresponding integers. Let n1
= len(L1).
2. Compute from the string msg a list L2 of corresponding integers. Let n2
= len(L2).
3. Break L2 up sequentially into sublists of size n1; the last sublist may be
smaller than n1
4. For each of these sublists L of L2, compute a new list C given by C[i] =
L[i] + L1[i] (mod N) to the i-th element in the sublist, for each i.
5. Assemble these lists C by concatenation into a new list of length n2.
6. Compute from the new list a string ct of corresponding letters.
Once it is known that the key is, say, n characters long, frequency analysis can be applied
to every n-th letter of the ciphertext to determine the plaintext. This method is called
Kasiski examination (although it was first discovered by Babbage). If they key is as long
as the message and is comprised of randomly selected characters – a one-time pad – the
message is theoretically unbreakable.
The cipher Vigenère actually discovered is an “auto-key” cipher described as follows.
ALGORITHM:
INPUT:
key: a string of letters (the secret key)
msg: string of letters (the plaintext message)
OUTPUT:
ct: string of upper-case letters (the ciphertext message)
STEPS:
0. Number the letters of the alphabet from 0, …, N
1. Compute from the string msg a list L2 of corresponding integers. Let n2
= len(L2).
2. Let n1 be the length of the key. Append to the string key the first n2 -
n1 characters of the plaintext message. Compute from this string (also
of length n2) a list L1 of integers corresponding to the letter numbers in
the first step.
3. Compute a new list C given by C[i] = L1[i] + L2[i] (mod N).
4. Compute from the new list a string ct of letters corresponding to the
new integers.
To decipher the auto-key ciphertext, the key is used to decipher the first n1 characters
and then those characters become the key to decipher the next n1 characters, etc…:
References
[R78], [R79]
Examples
Section 1 of the Kryptos sculpture at the CIA headquarters uses this cipher and also
changes the order of the the alphabet [R79]. Here is the first line of that section of the
sculpture:
Examples
Notes
The Hill cipher [R80], invented by Lester S. Hill in the 1920’s [R81], was the first poly-
graphic cipher in which it was practical (though barely) to operate on more than three
symbols at once. The following discussion assumes an elementary knowledge of matri-
ces.
First, each letter is first encoded as a number starting with 0. Suppose your message
msg consists of n capital letters, with no spaces. This may be regarded an n-tuple M of
elements of Z26 (if the letters are those of the English alphabet). A key in the Hill cipher
is a kxk matrix K, all of whose entries are in Z26 , such that the matrix K is invertible (i.e.,
the linear transformation K : ZN k
→ ZNk
is one-to-one).
ALGORITHM:
INPUT:
STEPS:
0. Number the letters of the alphabet from 0, …, N
1. Compute from the string msg a list L of corresponding integers. Let n =
len(L).
2. Break the list L up into t = ceiling(n/k) sublists L_1, …, L_t of size k
(with the last list “padded” to ensure its size is k).
3. Compute new list C_1, …, C_t given by C[i] = K*L_i (arithmetic is done
mod N), for each i.
4. Concatenate these into a list C = C_1 + ... + C_t.
5. Compute from C a string ct of corresponding letters. This has length
k*t.
References
[R80], [R81]
sympy.crypto.crypto.decipher_hill(msg, key, symbols=None)
Deciphering is the same as enciphering but using the inverse of the key matrix.
Examples
When the length of the plaintext (stripped of invalid characters) is not a multiple of the
key dimension, extra characters will appear at the end of the enciphered and deciphered
text. In order to decipher the text, those characters must be included in the text to be
deciphered. In the following, the key has a dimension of 4 but the text is 2 short of being
a multiple of 4 so two characters will be added.
If the last two characters of the ciphertext were ignored in either case, the wrong plain-
text would be recovered:
Examples
When entering the text or ciphertext, spaces are ignored so it can be formatted as de-
sired. Re-entering the ciphertext from the preceding, putting 4 characters per line and
padding with an extra J, does not cause problems for the deciphering:
>>> decipher_bifid('''
... IEILH
... HFSTS
... FQYEJ''', key, alp)
'MEETMEONMONDAY'
And if the key used to decrypt the message is not exact, the original text will not be
perfectly obtained:
sympy.crypto.crypto.bifid5_square(key=None)
5x5 Polybius square.
Produce the Polybius square for the 5 × 5 Bifid cipher.
Examples
sympy.crypto.crypto.encipher_bifid5(msg, key)
Performs the Bifid cipher encryption on plaintext msg, and returns the ciphertext.
This is the version of the Bifid cipher that uses the 5 × 5 Polybius square. The letter
“J” is ignored so it must be replaced with something else (traditionally an “I”) before
encryption.
See also:
decipher_bifid5 (page 355), encipher_bifid (page 352)
Notes
The Bifid cipher was invented around 1901 by Felix Delastelle. It is a fractional substi-
tution cipher, where letters are replaced by pairs of symbols from a smaller alphabet.
The cipher uses a 5 × 5 square filled with some ordering of the alphabet, except that
“J” is replaced with “I” (this is a so-called Polybius square; there is a 6 × 6 analog if you
add back in “J” and also append onto the usual 26 letter alphabet, the digits 0, 1, …, 9).
According to Helen Gaines’ book Cryptanalysis, this type of cipher was used in the field
by the German Army during World War I.
ALGORITHM: (5x5 case)
INPUT:
msg: plaintext string; converted to upper case and filtered of anything
but all letters except J.
key: short string for key; non-alphabetic letters, J and duplicated char-
acters are ignored and then, if the length is less than 25 characters, it
is padded with other letters of the alphabet (in alphabetical order).
OUTPUT:
ciphertext (all caps, no spaces)
STEPS:
0. Create the 5 × 5 Polybius square S associated to key as follows:
a) moving from left-to-right, top-to-bottom, place the letters of the key
into a 5 × 5 matrix,
b) if the key has less than 25 letters, add the letters of the alphabet not
in the key until the 5 × 5 square is filled.
1. Create a list P of pairs of numbers which are the coordinates in the Poly-
bius square of the letters in msg.
2. Let L1 be the list of all first coordinates of P (length of L1 = n), let L2 be
the list of all second coordinates of P (so the length of L2 is also n).
3. Let L be the concatenation of L1 and L2 (length L = 2*n), except that
consecutive numbers are paired (L[2*i], L[2*i + 1]). You can regard
L as a list of pairs of length n.
4. Let C be the list of all letters which are of the form S[i, j], for all (i,
j) in L. As a string, this is the ciphertext of msg.
Examples
sympy.crypto.crypto.decipher_bifid5(msg, key)
Return the Bifid cipher decryption of msg.
This is the version of the Bifid cipher that uses the 5 × 5 Polybius square; the letter “J” is
ignored unless a key of length 25 is used.
INPUT:
msg: ciphertext string
key: short string for key; duplicated characters are ignored and if the length is
less then 25 characters, it will be padded with other letters from the alphabet
omitting “J”. Non-alphabetic characters are ignored.
OUTPUT:
plaintext from Bifid5 cipher (all caps, no spaces)
Examples
sympy.crypto.crypto.bifid5_square(key=None)
5x5 Polybius square.
Produce the Polybius square for the 5 × 5 Bifid cipher.
Examples
sympy.crypto.crypto.encipher_bifid6(msg, key)
Performs the Bifid cipher encryption on plaintext msg, and returns the ciphertext.
This is the version of the Bifid cipher that uses the 6 × 6 Polybius square.
INPUT:
msg: plaintext string (digits okay)
key: short string for key (digits okay). If key is less than 36 characters long,
the square will be filled with letters A through Z and digits 0 through 9.
OUTPUT:
ciphertext from Bifid cipher (all caps, no spaces)
See also:
decipher_bifid6 (page 356), encipher_bifid (page 352)
sympy.crypto.crypto.decipher_bifid6(msg, key)
Performs the Bifid cipher decryption on ciphertext msg, and returns the plaintext.
This is the version of the Bifid cipher that uses the 6 × 6 Polybius square.
INPUT:
msg: ciphertext string (digits okay); converted to upper case
key: short string for key (digits okay). If key is less than 36 characters long, the
square will be filled with letters A through Z and digits 0 through 9. All letters
are converted to uppercase.
OUTPUT:
plaintext from Bifid cipher (all caps, no spaces)
Examples
sympy.crypto.crypto.bifid6_square(key=None)
6x6 Polybius square.
Produces the Polybius square for the 6 × 6 Bifid cipher. Assumes alphabet of symbols is
“A”, …, “Z”, “0”, …, “9”.
Examples
sympy.crypto.crypto.rsa_public_key(p, q, e)
Return the RSA public key pair, (n, e), where n is a product of two primes and e is relatively
prime (coprime) to the Euler totient ϕ(n). False is returned if any assumption is violated.
Examples
sympy.crypto.crypto.rsa_private_key(p, q, e)
Return the RSA private key, (n, d), where n is a product of two primes and d is the inverse
of e (mod ϕ(n)). False is returned if any assumption is violated.
Examples
sympy.crypto.crypto.encipher_rsa(i, key)
Return encryption of i by computing ie (mod n), where key is the public key (n, e).
Examples
sympy.crypto.crypto.decipher_rsa(i, key)
Return decyption of i by computing id (mod n), where key is the private key (n, d).
Examples
sympy.crypto.crypto.kid_rsa_public_key(a, b, A, B)
Kid RSA is a version of RSA useful to teach grade school children since it does not involve
exponentiation.
Alice wants to talk to Bob. Bob generates keys as follows. Key generation:
• Select positive integers a, b, A, B at random.
• Compute M = ab − 1, e = AM + a, d = BM + b, n = (ed − 1)//M .
• The public key is (n, e). Bob sends these to Alice.
• The private key is (n, d), which Bob keeps secret.
Encryption: If p is the plaintext message then the ciphertext is c = pe (mod n).
Decryption: If c is the ciphertext message then the plaintext is p = cd (mod n).
Examples
sympy.crypto.crypto.kid_rsa_private_key(a, b, A, B)
Compute M = ab − 1, e = AM + a, d = BM + b, n = (ed − 1)/M . The private key is d, which
Bob keeps secret.
Examples
sympy.crypto.crypto.encipher_kid_rsa(msg, key)
Here msg is the plaintext and key is the public key.
Examples
sympy.crypto.crypto.decipher_kid_rsa(msg, key)
Here msg is the plaintext and key is the private key.
Examples
References
[R82]
Examples
References
[R83]
Examples
sympy.crypto.crypto.lfsr_sequence(key, fill, n)
This function creates an lfsr sequence.
INPUT:
key: a list of finite field elements, [c0 , c1 , . . . , ck ].
fill: the list of the initial terms of the lfsr sequence, [x0 , x1 , . . . , xk ].
n: number of terms of the sequence that the function returns.
OUTPUT:
The lfsr sequence defined by xn+1 = ck xn + . . . + c0 xn−k , for n ≤ k.
Notes
1 ∑
N
C(k) = C(k, a) = lim (−1)an +an+k .
N →∞ N
n=1
1 ∑
P
C(k) = (−1)an +an+k .
P n=1
• low autocorrelation:
{
1, k = 0,
C(k) =
̸ 0.
ϵ, k =
(For sequences satisfying these first two properties, it is known that ϵ = −1/P must
hold.)
• proportional runs property: In each period, half the runs have length 1, one-fourth
have length 2, etc. Moreover, there are as many runs of 1’s as there are of 0’s.
References
[G83]
Examples
sympy.crypto.crypto.lfsr_autocorrelation(L, P, k)
This function computes the LFSR autocorrelation function.
INPUT:
L: is a periodic sequence of elements of GF (2). L must have length larger than
P.
P: the period of L
k: an integer (0 < k < p)
OUTPUT:
the k-th value of the autocorrelation of the LFSR L
Examples
sympy.crypto.crypto.lfsr_connection_polynomial(s)
This function computes the LFSR connection polynomial.
INPUT:
s: a sequence of elements of even length, with entries in a finite field
OUTPUT:
C(x): the connection polynomial of a minimal LFSR yielding s.
This implements the algorithm in section 3 of J. L. Massey’s article [M84].
References
[M84]
Examples
sympy.crypto.crypto.elgamal_public_key(key)
Return three number tuple as public key.
Parameters key : Tuple (p, r, e) generated by elgamal_private_key
Returns (p, r, e = r**d mod p) : d is a random number in private key.
Examples
sympy.crypto.crypto.elgamal_private_key(digit=10, seed=None)
Return three number tuple as private key.
Elgamal encryption is based on the mathmatical problem called the Discrete Logarithm
Problem (DLP). For example,
ab ≡ c (mod p)
In general, if a and b are known, ct is easily calculated. If b is unknown, it is hard to use
a and ct to get b.
Parameters digit : minimum number of binary digits for key
Notes
For testing purposes, the seed parameter may be set to control the output of this routine.
See sympy.utilities.randtest._randrange.
Examples
Notes
For testing purposes, the seed parameter may be set to control the output of this routine.
See sympy.utilities.randtest._randrange.
Examples
sympy.crypto.crypto.decipher_elgamal(msg, key)
Decrypt message with private key
msg = (c1 , c2 )
key = (p, r, d)
According to extended Eucliden theorem, ucd1 + pn = 1
u ≡ 1/c1 d (mod p)
uc2 ≡ 1
cd
c2 ≡ 1
c
r ad 2
(mod p)
1
1
r ad
mea ≡ 1
r ad
mrda ≡ m (mod p)
Examples
sympy.crypto.crypto.dh_public_key(key)
Return three number tuple as public key.
This is the tuple that Alice sends to Bob.
Parameters key: Tuple (p, g, a) generated by ‘‘dh_private_key‘‘
Returns (p, g, g^a mod p) : p, g and a as in Parameters
Examples
sympy.crypto.crypto.dh_private_key(digit=10, seed=None)
Return three integer tuple as private key.
Diffie-Hellman key exchange is based on the mathematical problem called the Discrete
Logarithm Problem (see ElGamal).
Diffie-Hellman key exchange is divided into the following steps:
• Alice and Bob agree on a base that consist of a prime p and a primitive root of p
called g
• Alice choses a number a and Bob choses a number b where a and b are random
numbers in range [2, p). These are their private keys.
• Alice then publicly sends Bob g a (mod p) while Bob sends Alice g b (mod p)
• They both raise the received value to their secretly chosen number (a or b) and now
have both as their shared key g ab (mod p)
Notes
For testing purposes, the seed parameter may be set to control the output of this routine.
See sympy.utilities.randtest._randrange.
Examples
sympy.crypto.crypto.dh_shared_key(key, b)
Return an integer that is the shared key.
This is what Bob and Alice can both calculate using the public keys they received from
each other and their private keys.
Parameters key: Tuple (p, g, x) generated by ‘‘dh_public_key‘‘
b: Random number in the range of 2 to p - 1
(Chosen by second key exchange member (Bob))
Returns shared key (int)
Examples
Notes
For testing purposes, the seed parameter may be set to control the output of this routine.
See sympy.utilities.randtest._randrange.
Examples
sympy.crypto.crypto.decipher_elgamal(msg, key)
Decrypt message with private key
msg = (c1 , c2 )
key = (p, r, d)
According to extended Eucliden theorem, ucd1 + pn = 1
u ≡ 1/c1 d (mod p)
uc2 ≡ 1
cd
c2 ≡ 1
c
r ad 2
(mod p)
1
1
r ad
mea ≡ 1
r ad
mrda ≡ m (mod p)
Examples
The center stage, in recurrence solving and summations, play hypergeometric terms. For-
mally these are sequences annihilated by first order linear recurrence operators. In simple
words if we are given term a(n) then it is hypergeometric if its consecutive term ratio is a
rational function in n.
To check if a sequence is of this type you can use the is_hypergeometric method which is
available in Basic class. Here is simple example involving a polynomial:
Of course polynomials are hypergeometric but are there any more complicated sequences of
this type? Here are some trivial examples:
>>> factorial(n).is_hypergeometric(n)
True
>>> binomial(n, k).is_hypergeometric(n)
True
>>> rf(n, k).is_hypergeometric(n)
True
>>> ff(n, k).is_hypergeometric(n)
True
>>> gamma(n).is_hypergeometric(n)
True
>>> (2**n).is_hypergeometric(n)
True
We see that all species used in summations and other parts of concrete mathematics are
hypergeometric. Note also that binomial coefficients and both rising and falling factorials
are hypergeometric in both their arguments:
To say more, all previously shown examples are valid for integer linear arguments:
>>> factorial(2*n).is_hypergeometric(n)
True
>>> binomial(3*n+1, k).is_hypergeometric(n)
True
>>> rf(n+1, k-1).is_hypergeometric(n)
True
>>> ff(n-1, k+1).is_hypergeometric(n)
True
>>> gamma(5*n).is_hypergeometric(n)
True
>>> (2**(n-7)).is_hypergeometric(n)
True
>>> factorial(n**2).is_hypergeometric(n)
False
>>> (2**(n**3 + 1)).is_hypergeometric(n)
False
If not only the knowledge of being hypergeometric or not is needed, you can use hypersimp()
function. It will try to simplify combinatorial expression and if the term given is hypergeo-
metric it will return a quotient of polynomials of minimal degree. Otherwise is will return
N one to say that sequence is not hypergeometric:
>>> hypersimp(factorial(2*n), n)
2*(n + 1)*(2*n + 1)
>>> hypersimp(factorial(n**2), n)
class sympy.concrete.summations.Sum
Represents unevaluated summation.
Sum represents a finite or infinite series, with the first argument being the general form
of terms in the series, and the second argument being (dummy_variable, start, end),
with dummy_variable taking all integer values from start through end. In accordance
with long-standing mathematical convention, the end term is included in the summation.
See also:
summation (page 379), Product, product
References
Examples
Here are examples to do summation with symbolic indices. You can use either Function
of IndexedBase classes:
>>> f = Function('f')
>>> Sum(f(n), (n, 0, 3)).doit()
f(0) + f(1) + f(2) + f(3)
>>> Sum(f(n), (n, 0, oo)).doit()
Sum(f(n), (n, 0, oo))
>>> f = IndexedBase('f')
>>> Sum(f[n]**2, (n, 0, 3)).doit()
f[0]**2 + f[1]**2 + f[2]**2 + f[3]**2
An example showing that the symbolic result of a summation is still valid for seemingly
nonsensical values of the limits. Then the Karr convention allows us to give a perfectly
valid interpretation to those sums by interchanging the limits according to the above
rules:
Finite Sums
For finite sums (and sums with symbolic limits assumed to be finite) we follow the sum-
mation convention described by Karr [1], especially definition 3 of section 1.4. The sum:
∑
f (i)
m≤i<n
with the upper limit value f (n) excluded. The sum over an empty set is zero if and only
if m = n:
∑
f (i) = 0 for m = n
m≤i<n
Finally, for all other sums over empty sets we assume the following definition:
∑ ∑
f (i) = − f (i) for m > n
m≤i<n n≤i<m
It is important to note that Karr defines all sums with the upper limit being exclusive.
This is in contrast to the usual mathematical notation, but does not affect the summation
convention. Indeed we have:
∑ ∑
n−1
f (i) = f (i)
m≤i<n i=m
where the difference in notation is intentional to emphasize the meaning, with limits
typeset on the top being inclusive.
euler_maclaurin(m=0, n=0, eps=0, eval_integral=True)
Return an Euler-Maclaurin approximation of self, where m is the number of leading
terms to sum directly and n is the number of terms in the tail.
With m = n = 0, this is simply the corresponding integral plus a first-order endpoint
correction.
Returns (s, e) where s is the Euler-Maclaurin approximation and e is the estimated
error (taken to be the magnitude of the first omitted term in the tail):
With a nonzero eps specified, the summation is ended as soon as the remainder term
is less than the epsilon.
eval_zeta_function(f, limits)
Check whether the function matches with the zeta function. If it matches, then
return a P iecewise expression because zeta function does not converge unless s > 1
and q > 0
is_absolutely_convergent()
Checks for the absolute convergence of an infinite series.
Same as checking convergence of absolute value of sequence_term of an infinite
series.
See also:
Sum.is_convergent (page 372)
References
[R40]
Examples
is_convergent()
Checks for the convergence of a Sum.
We divide the study of convergence of infinite sums and products in two parts.
First Part: One part is the question whether all the terms are well defined, i.e., they
are finite in a sum and also non-zero in a product. Zero is the analogy of (minus)
infinity in products as e−∞ = 0.
Second Part: The second part is the question of convergence after infinities, and
zeros in products, have been omitted assuming that their number is finite. This
means that we only consider the tail of the sum or product, starting from some point
after which all terms are well defined.
For example, in a sum of the form:
∑ 1
n2 + an + b
1≤i<∞
where a and b are numbers. The routine will return true, even if there are infinities
in the term sequence (at most two). An analogous product would be:
∏ 1
e n2 +an+b
1≤i<∞
References
[R41]
Examples
reverse_order(*indices)
Reverse the order of a limit in a Sum.
See also:
index, reorder_limit, reorder
References
[R42]
Examples
While one should prefer variable names when specifying which limits to reverse, the
index counting notation comes in handy in case there are several symbols with the
same name.
Usage
References
Examples
>>> Pe = Pe.rewrite(gamma)
>>> Pe
pi**2*gamma(n + 1 + pi/2)*gamma(n - pi/2 + 1)/(2*gamma(1 + pi/2)*gamma(-pi/2 +␣
,→1)*gamma(n + 1)**2)
>>> Pe = simplify(Pe)
(continues on next page)
Products with the lower limit being larger than the upper one:
An example showing that the symbolic result of a product is still valid for seemingly
nonsensical values of the limits. Then the Karr convention allows us to give a perfectly
valid interpretation to those products by interchanging the limits according to the above
rules:
Finite Products
For finite products (and products with symbolic limits assumed to be finite) we follow
the analogue of the summation convention described by Karr [1], especially definition 3
of section 1.4. The product:
∏
f (i)
m≤i<n
with the upper limit value f (n) excluded. The product over an empty set is one if and
only if m = n:
∏
f (i) = 1 for m = n
m≤i<n
Finally, for all other products over empty sets we assume the following definition:
∏ 1
f (i) = ∏ for m > n
m≤i<n n≤i<m f (i)
It is important to note that above we define all products with the upper limit being ex-
clusive. This is in contrast to the usual mathematical notation, but does not affect the
product convention. Indeed we have:
∏ ∏
n−1
f (i) = f (i)
m≤i<n i=m
where the difference in notation is intentional to emphasize the meaning, with limits
typeset on the top being inclusive.
is_convergent()
See docs of Sum.is_convergent() for explanation of convergence in SymPy.
The infinite product:
∏
f (i)
1≤i<∞
as n increases without bound. The product converges to a non-zero value if and only
if the sum:
∑
log f (n)
1≤i<∞
converges.
References
[R46]
Examples
>>> from sympy import Interval, S, Product, Symbol, cos, pi, exp, oo
>>> n = Symbol('n', integer=True)
>>> Product(n/(n + 1), (n, 1, oo)).is_convergent()
False
>>> Product(1/n**2, (n, 1, oo)).is_convergent()
False
>>> Product(cos(pi/n), (n, 1, oo)).is_convergent()
True
>>> Product(exp(-n**2), (n, 1, oo)).is_convergent()
False
reverse_order(*indices)
Reverse the order of a limit in a Product.
See also:
index, reorder_limit, reorder
References
[R47]
Examples
While one should prefer variable names when specifying which limits to reverse, the
index counting notation comes in handy in case there are several symbols with the
same name.
Usage
b
____
\ `
summation(f, (i, a, b)) = ) f
/___,
i = a
If it cannot compute the sum, it returns an unevaluated Sum object. Repeated sums can
be computed by introducing additional symbols tuples:
See also:
Sum (page 369), Product, product
sympy.concrete.products.product(*args, **kwargs)
Compute the product.
The notation for symbols is similar to the notation used in Sum or Integral. product(f, (i,
a, b)) computes the product of f with respect to i from a to b, i.e.,
b
_____
product(f(n), (i, a, b)) = | | f(n)
| |
i = a
sympy.concrete.gosper.gosper_normal(f, g, n, polys=True)
Compute the Gosper’s normal form of f and g.
Given relatively prime univariate polynomials f and g, rewrite their quotient to a normal
form defined as follows:
f (n) A(n)C(n + 1)
=Z·
g(n) B(n)C(n)
where Z is an arbitrary constant and A, B, C are monic polynomials in n with the following
properties:
1. gcd(A(n), B(n + h)) = 1∀h ∈ N
2. gcd(B(n), C(n + 1)) = 1
3. gcd(A(n), C(n)) = 1
This normal form, or rational factorization in other words, is a crucial step in Gosper’s
algorithm and in solving of difference equations. It can be also used to decide if two
hypergeometric terms are similar or not.
This procedure will return a tuple containing elements of this factorization in the form
(Z*A, B, C).
Examples
sympy.concrete.gosper.gosper_term(f, n)
Compute Gosper’s hypergeometric term for f.
Suppose f is a hypergeometric term such that:
∑
n−1
sn = fk
k=0
Examples
sympy.concrete.gosper.gosper_sum(f, k)
Gosper’s hypergeometric summation algorithm.
Given a hypergeometric term f such that:
∑
n−1
sn = fk
k=0
and f (n) doesn’t depend on n, returns gn − g(0) where gn+1 − gn = fn , or None if sn can not
be expressed in closed form as a sum of hypergeometric terms.
References
[R48]
Examples
The discrete module in SymPy implements methods to compute discrete transforms and
convolutions of finite sequences.
This module contains functions which operate on discrete sequences.
Transforms - fft, ifft, ntt, intt, fwht, ifwht, mobius_transform, in-
verse_mobius_transform
Convolutions - convolution, convolution_fft, convolution_ntt, convolution_fwht,
convolution_subset, covering_product, intersecting_product
Since the discrete transforms can be used to reduce the computational complexity of the
discrete convolutions, the convolutions module makes use of the transforms module for
efficient computation (notable for long input sequences).
5.6.1 Transforms
This section lists the methods which implement the basic transforms for discrete sequences.
sympy.discrete.transforms.fft(seq, dps=None)
Performs the Discrete Fourier Transform (DFT) in the complex domain.
The sequence is automatically padded to the right with zeros, as the radix-2 FFT requires
the number of sample points to be a power of 2.
This method should be used with default arguments only for short sequences as the
complexity of expressions increases with the size of the sequence.
Parameters seq : iterable
The sequence on which DFT is to be applied.
dps : Integer
Specifies the number of decimal digits for precision.
References
[R86], [R87]
Examples
sympy.discrete.transforms.ifft(seq, dps=None)
Performs the Discrete Fourier Transform (DFT) in the complex domain.
The sequence is automatically padded to the right with zeros, as the radix-2 FFT requires
the number of sample points to be a power of 2.
This method should be used with default arguments only for short sequences as the
complexity of expressions increases with the size of the sequence.
Parameters seq : iterable
The sequence on which DFT is to be applied.
dps : Integer
Specifies the number of decimal digits for precision.
References
[R88], [R89]
Examples
sympy.discrete.transforms.ntt(seq, prime)
Performs the Number Theoretic Transform (NTT), which specializes the Discrete Fourier
Transform (DFT) over quotient ring Z/pZ for prime p instead of complex numbers C.
The sequence is automatically padded to the right with zeros, as the radix-2 NTT requires
the number of sample points to be a power of 2.
Parameters seq : iterable
The sequence on which DFT is to be applied.
prime : Integer
Prime modulus of the form (m2k + 1) to be used for performing NTT
on the sequence.
References
Examples
sympy.discrete.transforms.intt(seq, prime)
Performs the Number Theoretic Transform (NTT), which specializes the Discrete Fourier
Transform (DFT) over quotient ring Z/pZ for prime p instead of complex numbers C.
The sequence is automatically padded to the right with zeros, as the radix-2 NTT requires
the number of sample points to be a power of 2.
Parameters seq : iterable
The sequence on which DFT is to be applied.
prime : Integer
Prime modulus of the form (m2k + 1) to be used for performing NTT
on the sequence.
References
Examples
sympy.discrete.transforms.fwht(seq)
Performs the Walsh Hadamard Transform (WHT), and uses Hadamard ordering for the
sequence.
The sequence is automatically padded to the right with zeros, as the radix-2 FWHT re-
quires the number of sample points to be a power of 2.
Parameters seq : iterable
The sequence on which WHT is to be applied.
References
[R96], [R97]
Examples
sympy.discrete.transforms.ifwht(seq)
Performs the Walsh Hadamard Transform (WHT), and uses Hadamard ordering for the
sequence.
The sequence is automatically padded to the right with zeros, as the radix-2 FWHT re-
quires the number of sample points to be a power of 2.
Parameters seq : iterable
The sequence on which WHT is to be applied.
References
[R98], [R99]
Examples
Möbius Transform
sympy.discrete.transforms.mobius_transform(seq, subset=True)
Performs the Möbius Transform for subset lattice with indices of sequence as bitmasks.
The indices of each argument, considered as bit strings, correspond to subsets of a finite
set.
The sequence is automatically padded to the right with zeros, as the definition of sub-
set/superset based on bitmasks (indices) requires the size of sequence to be a power of
2.
Parameters seq : iterable
The sequence on which Möbius Transform is to be applied.
subset : bool
Specifies if Möbius Transform is applied by enumerating subsets or
supersets of the given set.
References
Examples
sympy.discrete.transforms.inverse_mobius_transform(seq, subset=True)
Performs the Möbius Transform for subset lattice with indices of sequence as bitmasks.
The indices of each argument, considered as bit strings, correspond to subsets of a finite
set.
The sequence is automatically padded to the right with zeros, as the definition of sub-
set/superset based on bitmasks (indices) requires the size of sequence to be a power of
2.
Parameters seq : iterable
The sequence on which Möbius Transform is to be applied.
subset : bool
Specifies if Möbius Transform is applied by enumerating subsets or
supersets of the given set.
References
Examples
5.6.2 Convolutions
This section lists the methods which implement the basic convolutions for discrete sequences.
Convolution
This is a general method for calculating the convolution of discrete sequences, which inter-
nally calls one of the methods convolution_fft, convolution_ntt, convolution_fwht, or
convolution_subset.
sympy.discrete.convolutions.convolution(a, b, cycle=0, dps=None, prime=None,
dyadic=None, subset=None)
Performs convolution by determining the type of desired convolution using hints.
Exactly one of dps, prime, dyadic, subset arguments should be specified explicitly for
identifying the type of convolution, and the argument cycle can be specified optionally.
For the default arguments, linear convolution is performed using FFT.
Parameters a, b : iterables
The sequences for which convolution is performed.
cycle : Integer
Specifies the length for doing cyclic convolution.
dps : Integer
Specifies the number of decimal digits for precision for performing
FFT on the sequence.
prime : Integer
Prime modulus of the form (m2k + 1) to be used for performing NTT
on the sequence.
dyadic : bool
Identifies the convolution type as dyadic (bitwise-XOR) convolution,
which is performed using FWHT.
subset : bool
Identifies the convolution type as subset convolution.
Examples
sympy.discrete.convolutions.convolution_fft(a, b, dps=None)
Performs linear convolution using Fast Fourier Transform.
Parameters a, b : iterables
The sequences for which convolution is performed.
dps : Integer
Specifies the number of decimal digits for precision.
References
[R106], [R107]
Examples
sympy.discrete.convolutions.convolution_ntt(a, b, prime)
Performs linear convolution using Number Theoretic Transform.
Parameters a, b : iterables
The sequences for which convolution is performed.
prime : Integer
Prime modulus of the form (m2k + 1) to be used for performing NTT
on the sequence.
References
[R108], [R109]
Examples
sympy.discrete.convolutions.convolution_fwht(a, b)
Performs dyadic (bitwise-XOR) convolution using Fast Walsh Hadamard Transform.
The convolution is automatically padded to the right with zeros, as the radix-2 FWHT
requires the number of sample points to be a power of 2.
Parameters a, b : iterables
The sequences for which convolution is performed.
References
[R110], [R111]
Examples
Subset Convolution
sympy.discrete.convolutions.convolution_subset(a, b)
Performs Subset Convolution of given sequences.
The indices of each argument, considered as bit strings, correspond to subsets of a finite
set.
The sequence is automatically padded to the right with zeros, as the definition of subset
based on bitmasks (indices) requires the size of sequence to be a power of 2.
Parameters a, b : iterables
The sequences for which convolution is performed.
References
[R112]
Examples
Covering Product
sympy.discrete.convolutions.covering_product(a, b)
Returns the covering product of given sequences.
The indices of each argument, considered as bit strings, correspond to subsets of a finite
set.
The covering product of given sequences is a sequence which contains the sum of prod-
ucts of the elements of the given sequences grouped by the bitwise-OR of the corre-
sponding indices.
The sequence is automatically padded to the right with zeros, as the definition of subset
based on bitmasks (indices) requires the size of sequence to be a power of 2.
Parameters a, b : iterables
The sequences for which covering product is to be obtained.
References
[R113]
Examples
Intersecting Product
sympy.discrete.convolutions.intersecting_product(a, b)
Returns the intersecting product of given sequences.
The indices of each argument, considered as bit strings, correspond to subsets of a finite
set.
The intersecting product of given sequences is the sequence which contains the sum
of products of the elements of the given sequences grouped by the bitwise-AND of the
corresponding indices.
The sequence is automatically padded to the right with zeros, as the definition of subset
based on bitmasks (indices) requires the size of sequence to be a power of 2.
Parameters a, b : iterables
References
[R114]
Examples
5.7.1 Basics
>>> N(sqrt(2)*pi, 5)
4.4429
>>> N(sqrt(2)*pi, 50)
4.4428829381583662470158809900606936986146216893757
If the expression contains symbols or for some other reason cannot be evaluated numerically,
calling .evalf() or N() returns the original expression, or in some cases a partially evalu-
ated expression. For example, when the expression is a polynomial in expanded form, the
coefficients are evaluated:
>>> x = Symbol('x')
>>> (pi*x**2 + x/3).evalf()
3.14159265358979*x**2 + 0.333333333333333*x
You can also use the standard Python functions float(), complex() to convert SymPy ex-
pressions to regular Python numbers:
>>> float(pi)
3.1415926535...
>>> complex(pi+E*I)
(3.1415926535...+2.7182818284...j)
If these functions are used, failure to evaluate the expression to an explicit number (for ex-
ample if the expression contains symbols) will raise an exception.
There is essentially no upper precision limit. The following command, for example, computes
the first 100,000 digits of π/e:
Floating-point numbers in SymPy are instances of the class Float. A Float can be created
with a custom precision as second argument:
>>> Float(0.1)
0.100000000000000
>>> Float(0.1, 10)
0.1000000000
>>> Float(0.125, 30)
0.125000000000000000000000000000
>>> Float(0.1, 30)
0.100000000000000005551115123126
As the last example shows, some Python floats are only accurate to about 15 digits as inputs,
while others (those that have a denominator that is a power of 2, like 0.125 = 1/8) are exact.
To create a Float from a high-precision decimal number, it is better to pass a string, Rational,
or evalf a Rational:
The precision of a number determines 1) the precision to use when performing arithmetic
with the number, and 2) the number of digits to display when printing the number. When two
numbers with different precision are used together in an arithmetic operation, the higher of
the precisions is used for the result. The product of 0.1 +/- 0.001 and 3.1415 +/- 0.0001 has
an uncertainty of about 0.003 and yet 5 digits of precision are shown.
So the displayed precision should not be used as a model of error propagation or significance
arithmetic; rather, this scheme is employed to ensure stability of numerical algorithms.
N and evalf can be used to change the precision of existing floating-point numbers:
>>> N(3.5)
3.50000000000000
>>> N(3.5, 5)
3.5000
>>> N(3.5, 30)
3.50000000000000000000000000000
When the input to N or evalf is a complicated expression, numerical error propagation be-
comes a concern. As an example, consider
√ the 100’th Fibonacci number and the excellent
(but not exact) approximation φ100 / 5 where φ is the golden ratio. With ordinary floating-
point arithmetic, subtracting these numbers from each other erroneously results in a com-
plete cancellation:
N and evalf keep track of errors and automatically increase the precision used internally in
order to obtain a correct result:
Unfortunately, numerical evaluation cannot tell an expression that is exactly zero apart from
one that is merely very small. The working precision is therefore capped, by default to around
100 digits. If we try with the 1000’th Fibonacci number, the following happens:
The lack of digits in the returned number indicates that N failed to achieve full accuracy. The
result indicates that the magnitude of the expression is something less than 10^84, but that
is not a particularly good answer. To force a higher working precision, the maxn keyword
argument can be used:
Normally, maxn can be set very high (thousands of digits), but be aware that this may cause
significant slowdown in extreme cases. Alternatively, the strict=True option can be set to
force an exception instead of silently returning a value with less than the requested accuracy:
<BLANKLINE>
from zero. Try simplifying the input, using chop=True, or providing a higher maxn for␣
,→evalf
If we add a term so that the Fibonacci approximation becomes exact (the full form of Binet’s
formula), we get an expression that is exactly zero, but N does not know this:
In situations where such cancellations are known to occur, the chop options is useful. This
basically replaces very small numbers in the real or imaginary portions of a number with
exact zeros:
In situations where you wish to remove meaningless digits, re-evaluation or the use of the
round method are useful:
If you are dealing with a numeric expression that contains no floats, it can be evaluated to
arbitrary precision. To round the result relative to a given decimal, the round method is
useful:
Sums (in particular, infinite series) and integrals can be used like regular closed-form expres-
sions, and support arbitrary-precision evaluation:
By default, the tanh-sinh quadrature algorithm is used to evaluate integrals. This algorithm
is very efficient and robust for smooth integrands (and even integrals with endpoint singulari-
ties), but may struggle with integrals that are highly oscillatory or have mid-interval disconti-
nuities. In many cases, evalf/N will correctly estimate the error. With the following integral,
the result is accurate but only good to four digits:
>>> f = abs(sin(x))
>>> Integral(abs(sin(x)), (x, 0, 4)).evalf()
2.346
It can be dealt with much more efficiently by telling evalf or N to use an oscillatory quadrature
algorithm:
of variables:
Infinite series use direct summation if the series converges quickly enough. Otherwise, ex-
trapolation methods (generally the Euler-Maclaurin formula but also Richardson extrapola-
tion) are used to speed up convergence. This allows high-precision evaluation of slowly con-
vergent series:
>>> var('k')
k
>>> Sum(1/k**2, (k, 1, oo)).evalf()
1.64493406684823
>>> zeta(2).evalf()
1.64493406684823
>>> Sum(1/k-log(1+1/k), (k, 1, oo)).evalf()
0.577215664901533
>>> Sum(1/k-log(1+1/k), (k, 1, oo)).evalf(50)
0.57721566490153286060651209008240243104215933593992
>>> EulerGamma.evalf(50)
0.57721566490153286060651209008240243104215933593992
The Euler-Maclaurin formula is also used for finite series, allowing them to be approximated
quickly without evaluating all terms:
Note that evalf makes some assumptions that are not always optimal. For fine-tuned con-
trol over numerical summation, it might be worthwhile to manually use the method Sum.
euler_maclaurin.
Special optimizations are used for rational hypergeometric series (where the term is a product
of polynomials, powers, factorials, binomial coefficients and the like). N/evalf sum series of
this type very rapidly to high precision. For example, this Ramanujan formula for pi can be
summed to 10,000 digits in a fraction of a second with a simple command:
>>> f = factorial
>>> n = Symbol('n', integer=True)
>>> R = 9801/sqrt(8)/Sum(f(4*n)*(1103+26390*n)/f(n)**4/396**(4*n),
... (n, 0, oo)) #doctest: +SKIP
>>> N(R, 10000) #doctest: +SKIP
3.141592653589793238462643383279502884197169399375105820974944592307816406286208
(continues on next page)
The function nsimplify attempts to find a formula that is numerically equal to the given
input. This feature can be used to guess an exact formula for an approximate floating-point
input, or to guess a simpler formula for a complicated symbolic input. The algorithm used
by nsimplify is capable of identifying simple fractions, simple algebraic expressions, linear
combinations of given constants, and certain elementary functional transformations of any of
the preceding.
Optionally, nsimplify can be passed a list of constants to include (e.g. pi) and a minimum
numerical tolerance. Here are some elementary examples:
>>> nsimplify(0.1)
1/10
>>> nsimplify(6.28, [pi], tolerance=0.01)
2*pi
>>> nsimplify(pi, tolerance=0.01)
22/7
>>> nsimplify(pi, tolerance=0.001)
355
---
113
>>> nsimplify(0.33333, tolerance=1e-4)
1/3
>>> nsimplify(2.0**(1/3.), tolerance=0.001)
635
---
504
>>> nsimplify(2.0**(1/3.), tolerance=0.001, full=True)
3 ___
\/ 2
Several submodules in SymPy allow one to generate directly compilable and executable code
in a variety of different programming languages from SymPy expressions. In addition, there
are functions that generate Python importable objects that can evaluate SymPy expressions
very efficiently.
We will start with a brief introduction to the components that make up the code generation
capabilities of SymPy.
5.8.1 Introduction
expression
|
code printers
|
code generators
|
autowrap
sympy.utilities.autowrap (page 1431) uses codegen, and codegen uses the code printers.
sympy.utilities.autowrap (page 1431) does everything: it lets you go from SymPy expres-
sion to numerical function in the same Python process in one step. Codegen is actual code
generation, i.e., to compile and use later, or to include in some larger project.
The code printers translate the SymPy objects into actual code, like abs(x) -> fabs(x) (for
C).
The code printers don’t print optimal code in many cases. An example of this is powers in C.
x**2 prints as pow(x, 2) instead of x*x. Other optimizations (like mathematical simplifica-
tions) should happen before the code printers.
Currently, sympy.simplify.cse_main.cse() (page 1205) is not applied automatically any-
where in this chain. It ideally happens at the codegen level, or somewhere above it.
We will iterate through the levels below.
The following three lines will be used to setup each example:
This is where the meat of code generation is; the translation of SymPy actually more like a
lightweight version of codegen for Python, and Python (sympy.printing.pycode.pycode()
(page 1077)), and sympy.printing.lambdarepr.lambdarepr() (page 1072), which supports
many libraries (like NumPy), and theano (sympy.printing.theanocode.theano_function()
(page 1071)). The code printers are special cases of the other prints in SymPy (str printer,
pretty printer, etc.).
An important distinction is that the code printer has to deal with assignments (using the
sympy.codegen.ast.Assignment (page 414) object). This serves as building blocks for the
code printers and hence the codegen module. An example that shows the use of Assignment
in C code:
To generate code with some math functions provided by e.g. the C99 standard we need to
import functions from sympy.codegen.cfunctions (page 429):
The various printers also tend to support Indexed objects well. With contract=True these
expressions will be turned into loops, whereas contract=False will just print the assignment
expression that should be looped over:
>>> len_y = 5
>>> mat_1 = IndexedBase('mat_1', shape=(len_y,))
>>> mat_2 = IndexedBase('mat_2', shape=(len_y,))
>>> Dy = IndexedBase('Dy', shape=(len_y-1,))
>>> i = Idx('i', len_y-1)
>>> eq = Eq(Dy[i], (mat_1[i+1] - mat_1[i]) / (mat_2[i+1] - mat_2[i]))
>>> print(jscode(eq.rhs, assign_to=eq.lhs, contract=False))
Dy[i] = (mat_1[i + 1] - mat_1[i])/(mat_2[i + 1] - mat_2[i]);
>>> Res = IndexedBase('Res', shape=(len_y,))
>>> j = Idx('j', len_y)
>>> eq = Eq(Res[j], mat_1[j]*mat_2[j])
>>> print(jscode(eq.rhs, assign_to=eq.lhs, contract=True))
for (var j=0; j<5; j++){
Res[j] = 0;
}
for (var j=0; j<5; j++){
for (var j=0; j<5; j++){
Res[j] = Res[j] + mat_1[j]*mat_2[j];
}
}
>>> print(jscode(eq.rhs, assign_to=eq.lhs, contract=False))
Res[j] = mat_1[j]*mat_2[j];
Custom printing can be defined for certain types by passing a dictionary of “type” : “function”
to the user_functions kwarg. Alternatively, the dictionary value can be a list of tuples i.e.,
[(argument_test, cfunction_string)]. This can be used to call a custom Octave function:
>>> custom_functions = {
... ”f”: ”existing_octave_fcn”,
... ”g”: [(lambda x: x.is_Matrix, ”my_mat_fcn”),
... (lambda x: not x.is_Matrix, ”my_fcn”)]
... }
>>> mat = Matrix([[1, x]])
>>> octave_code(f(x) + g(x) + g(mat), user_functions=custom_functions)
existing_octave_fcn(x) + my_fcn(x) + my_mat_fcn([1 x])
>>> x_ = Function('x')
>>> expr = x_(n*T) * sin((t - n*T) / T)
>>> expr = expr / ((-T*n + t) / T)
>>> expr
⎛-T⋅n + t⎞
T⋅x(T⋅n)⋅sin⎜────────⎟
⎝ T ⎠
──────────────────────
-T⋅n + t
We can go through a common expression in different languages we support and see how it
works:
This module deals with creating compilable code from SymPy expressions. This is lower
level than autowrap, as it doesn’t actually attempt to compile the code, but higher level than
the printers, as it generates compilable files (including header files), rather than just code
snippets.
The user friendly functions, here, are codegen and make_routine. codegen takes a list of
(variable, expression) pairs and a language (C, F95, and Octave/Matlab are supported).
It returns, as strings, a code file and a header file (for relevant languages). The variables are
created as functions that return the value of the expression as output.
Note: The codegen callable is not in the sympy namespace automatically, to use it you must
For instance:
Various flags to codegen let you modify things. The project name for preprocessor instructions
can be varied using project. Variables listed as global variables in arg global_vars will not
show up as function arguments.
language is a case-insensitive string that indicates the source code language. Currently, C,
F95 and Octave are supported. Octave generates code compatible with both Octave and
Matlab.
header when True, a header is written on top of each source file. empty when True, empty
lines are used to structure the code. With argument_sequence a sequence of arguments for
the routine can be defined in a preferred order.
prefix defines a prefix for the names of the files that contain the source code. If omitted, the
name of the first name_expr tuple is used.
to_files when True, the code will be written to one or more files with the given prefix.
Here is an example:
The method make_routine creates a Routine object, which represents an evaluation rou-
tine for a set of expressions. This is only good for internal use by the CodeGen objects, as
>>> routine = make_routine('fcn', [x*y, Eq(a, 1), Eq(r, x + r), Matrix([[x, 2]])])
>>> [arg.result_var for arg in routine.results]
[result_5397460570204848505]
>>> [arg.expr for arg in routine.results]
[x⋅y]
>>> [arg.name for arg in routine.arguments]
[x, y, a, r, out_8598435338387848786]
5.8.4 Autowrap
Autowrap automatically generates code, writes it to disk, compiles it, and imports it into
the current session. Main functions of this module are autowrap, binary_function, and
ufuncify.
It also automatically converts expressions containing Indexed objects into summations. The
classes IndexedBase, Indexed and Idx represent a matrix element M[i, j]. See Tensor Module
(page 1399) for more on this.
autowrap creates a wrapper using f2py or Cython and creates a numerical function.
Note: The autowrap callable is not in the sympy namespace automatically, to use it you must
first import autowrap from sympy.utilities.autowrap
The callable returned from autowrap() is a binary Python function, not a SymPy object. For
example:
The various flags available with autowrap() help to modify the services provided by the
method. The argument tempdir tells autowrap to compile the code in a specific directory,
and leave the files intact when finished. For instance:
Checking the Fortran source code in the directory specified reveals this:
end subroutine
yields:
subroutine autofunc(y, x, m, omega)
implicit none
INTEGER*4, intent(in) :: m
REAL*8, intent(in) :: omega
REAL*8, intent(out), dimension(1:m) :: y
REAL*8, intent(in), dimension(1:m) :: x
INTEGER*4 :: i
end subroutine
The argument verbose is boolean, optional and if True, autowrap will not mute the command
line backends. This can be helpful for debugging.
The argument language and backend are used to change defaults: Fortran and f2py to C and
Cython. The argument helpers is used to define auxiliary expressions needed for the main
expression. If the main expression needs to call a specialized function it should be put in the
helpers iterable. Autowrap will then make sure that the compiled main expression can link to
the helper routine. Items should be tuples with (<function_name>, <sympy_expression>,
<arguments>). It is mandatory to supply an argument sequence to helper routines.
Another method available at the autowrap level is binary_function. It returns a sympy
function. The advantage is that we can have very fast functions as compared to SymPy speeds.
This is because we will be using compiled functions with Sympy attributes and methods. An
illustration:
>>> from sympy.utilities.autowrap import binary_function
>>> from sympy.physics.hydrogen import R_nl
>>> psi_nl = R_nl(1, 0, a, r)
>>> f = binary_function('f', psi_nl)
>>> f(a, r).evalf(3, subs={a: 1, r: 2})
0.766
While NumPy operations are very efficient for vectorized data but they sometimes incur un-
necessary costs when chained together. Consider the following operation
>>> x = get_numpy_array(...) # doctest: +SKIP
>>> y = sin(x) / x
The operators sin and / call routines that execute tight for loops in C. The resulting compu-
tation looks something like this
for(int i = 0; i < n; i++)
{
temp[i] = sin(x[i]);
}
(continues on next page)
Statically compiled projects like NumPy are unable to take advantage of such optimizations.
Fortunately, SymPy is able to generate efficient low-level C or Fortran code. It can then
depend on projects like Cython or f2py to compile and reconnect that code back up to Python.
Fortunately this process is well automated and a SymPy user wishing to make use of this code
generation should call the ufuncify function.
ufuncify is the third method available with Autowrap module. It basically implies ‘Universal
functions’ and follows an ideology set by NumPy. The main point of ufuncify as compared to
autowrap is that it allows arrays as arguments and can operate in an element-by-element fash-
ion. The core operation done element-wise is in accordance to Numpy’s array broadcasting
rules. See this for more.
>>> from sympy import *
>>> from sympy.abc import x
>>> expr = sin(x)/x
This function f consumes and returns a NumPy array. Generally ufuncify performs at least
as well as lambdify. If the expression is complicated then ufuncify often significantly out-
performs the NumPy backed solution. Jensen has a good blog post on this topic.
Let us see an example for some quantitative analysis:
>>> from sympy.physics.hydrogen import R_nl
>>> expr = R_nl(3, 1, x, 6)
>>> expr
-2⋅x
8⋅x⋅(-4⋅x + 4)⋅e
────────────────────
3
The lambdify function translates SymPy expressions into Python functions, leveraging a vari-
ety of numerical libraries. By default lambdify relies on implementations in the math standard
library. Naturally, Raw Python is faster than SymPy. However it also supports mpmath and
most notably, numpy. Using the NumPy library gives the generated function access to power-
ful vectorized ufuncs that are backed by compiled C code.
Let us compare the speeds:
0.004707066000264604
The options available with ufuncify are more or less the same as those available with au-
towrap.
There are other facilities available with SymPy to do efficient numeric computation. See this
(page 443) page for a comparison among them.
Classes and functions useful for rewriting expressions for optimized code generation. Some
languages (or standards thereof), e.g. C99, offer specialized math functions for better per-
formance and/or precision.
Using the optimize function in this module, together with a collection of rules (represented
as instances of Optimization), one can rewrite the expressions for this purpose:
The optims_c99 imported above is tuple containing the following instances (which may be
imported from sympy.codegen.rewriting):
• expm1_opt
• log1p_opt
• exp2_opt
• log2_opt
• log2const_opt
Examples
sympy.codegen.rewriting.create_expand_pow_optimization(limit)
Creates an instance of ReplaceOptim (page 410) for expanding Pow.
The requirements for expansions are that the base needs to be a symbol and the exponent
needs to be an integer (and be less than or equal to limit).
Parameters limit : int
The highest power which is expanded into multiplication.
Examples
sympy.codegen.rewriting.optimize(expr, optimizations)
Apply optimizations to an expression.
Parameters expr : expression
optimizations : iterable of Optimization instances
Examples
Examples
Examples
*Basic*
|--->AssignmentBase
| |--->Assignment
| |--->AugmentedAssignment
| |--->AddAugmentedAssignment
| |--->SubAugmentedAssignment
| |--->MulAugmentedAssignment
| |--->DivAugmentedAssignment
| |--->ModAugmentedAssignment
|
|--->CodeBlock
|
(continues on next page)
Predefined types
A number of Type instances are provided in the sympy.codegen.ast module for convenience.
Perhaps the two most common ones for code-generation (of numeric codes) are float32 and
float64 (known as single and double precision respectively). There are also precision generic
versions of Types (for which the codeprinters selects the underlying data type at time of
printing): real, integer, complex_, bool_.
The other Type instances defined are:
• intc: Integer type used by C’s “int”.
• intp: Integer type used by C’s “unsigned”.
• int8, int16, int32, int64: n-bit integers.
• uint8, uint16, uint32, uint64: n-bit unsigned integers.
• float80: known as “extended precision” on modern x86/amd64 hardware.
• complex64: Complex number represented by two float32 numbers
• complex128: Complex number represented by two float64 numbers
It is possible to construct simple algorithms using the AST nodes. Let’s construct a loop
applying Newton’s method:
If we want to generate Fortran code for the same while loop we simple call fcode:
There is a function constructing a loop (or a complete function) like this in sympy.codegen.
algorithms (page 441).
class sympy.codegen.ast.Assignment
Represents variable assignment for code generation.
Parameters lhs : Expr
Sympy object representing the lhs of the expression. These should
be singular objects, such as one would use in writing code. Notable
types include Symbol, MatrixSymbol, MatrixElement, and Indexed.
Types that subclass these types are also supported.
rhs : Expr
Sympy object representing the rhs of the expression. This can be any
type, provided its shape corresponds to that of the lhs. For example,
Examples
class sympy.codegen.ast.AssignmentBase
Abstract base class for Assignment and AugmentedAssignment.
Attributes:
Examples
class sympy.codegen.ast.AugmentedAssignment
Base class for augmented assignments.
Attributes:
binop [str] Symbol for binary operation being applied in the assignment, such as “+”,
“*”, etc.
class sympy.codegen.ast.BreakToken
Represents ‘break’ in C/Python (‘exit’ in Fortran).
Use the premade instance break_ or instantiate manually.
Examples
class sympy.codegen.ast.CodeBlock
Represents a block of code
For now only assignments are supported. This restriction will be lifted in the future.
Useful attributes on this object are:
left_hand_sides: Tuple of left-hand sides of assignments, in order.
left_hand_sides: Tuple of right-hand sides of assignments, in order.
free_symbols: Free symbols of the expressions in the right-hand sides which do
not appear in the left-hand side of an assignment.
Useful methods on this object are:
topological_sort: Class method. Return a CodeBlock with assignments sorted so that
variables are assigned before they are used.
cse: Return a new CodeBlock with common subexpressions eliminated and pulled out
as assignments.
Example
Examples
>>> c = CodeBlock(
... Assignment(x, 1),
... Assignment(y, sin(x) + 1),
... Assignment(z, sin(x) - 1),
... )
...
>>> c.cse()
CodeBlock(
Assignment(x, 1),
Assignment(x0, sin(x)),
Assignment(y, x0 + 1),
Assignment(z, x0 - 1)
)
classmethod topological_sort(assignments)
Return a CodeBlock with topologically sorted assignments so that variables are as-
signed before they are used.
The existing order of assignments is preserved as much as possible.
This function assumes that variables are assigned to only once.
This is a class constructor so that the default constructor for CodeBlock can error
when variables are used before they are assigned.
Example
>>> assignments = [
... Assignment(x, y + z),
... Assignment(y, z + 1),
... Assignment(z, 2),
... ]
>>> CodeBlock.topological_sort(assignments)
CodeBlock(
Assignment(z, 2),
Assignment(y, z + 1),
Assignment(x, y + z)
)
class sympy.codegen.ast.Comment
Represents a comment.
class sympy.codegen.ast.ComplexType
Represents a complex floating point number.
class sympy.codegen.ast.ContinueToken
Represents ‘continue’ in C/Python (‘cycle’ in Fortran)
Examples
class sympy.codegen.ast.Declaration
Represents a variable declaration
Parameters variable : Variable
Examples
class sympy.codegen.ast.Element
Element in (a possibly N-dimensional) array.
Examples
class sympy.codegen.ast.FloatBaseType
Represents a floating point number type.
cast_nocheck
alias of sympy.core.numbers.Float (page 143)
class sympy.codegen.ast.FloatType
Represents a floating point type with fixed bit width.
Base 2 & one sign bit is assumed.
Examples
cast_nocheck(value)
Casts without checking if out of bounds or subnormal.
decimal_dig
Number of digits needed to store & load without loss.
Number of decimal digits needed to guarantee that two consecutive conversions
(float -> text -> float) to be idempotent. This is useful when one do not want to loose
precision due to rounding errors when storing a floating point value as text.
dig
Number of decimal digits that are guaranteed to be preserved in text.
When converting text -> float -> text, you are guaranteed that at least dig number
of digits are preserved with respect to rounding or overflow.
eps
Difference between 1.0 and the next representable value.
max
Maximum value representable.
max_exponent
The largest positive number n, such that 2**(n - 1) is a representable finite value.
min_exponent
The lowest negative number n, such that 2**(n - 1) is a valid normalized number.
tiny
The minimum positive normalized value.
class sympy.codegen.ast.For
Represents a ‘for-loop’ in the code.
Expressions are of the form:
“for target in iter: body…”
Examples
class sympy.codegen.ast.FunctionCall
Represents a call to a function in the code.
Parameters name : str
function_args : Tuple
Examples
class sympy.codegen.ast.FunctionDefinition
Represents a function definition in the code.
Parameters return_type : Type
name : str
parameters: iterable of Variable instances
body : CodeBlock or iterable
attrs : iterable of Attribute instances
Examples
class sympy.codegen.ast.FunctionPrototype
Represents a function prototype
Allows the user to generate forward declaration in e.g. C/C++.
Parameters return_type : Type
name : str
parameters: iterable of Variable instances
attrs : iterable of Attribute instances
Examples
class sympy.codegen.ast.IntBaseType
Integer base type, contains no size information.
class sympy.codegen.ast.Node
Subclass of Token, carrying the attribute ‘attrs’ (Tuple)
Examples
()
>>> n2.attr_params('dimension') # get the parameters of attribute (by name)
(5, 3)
>>> n2.attr_params(pointer_const) is None
True
attr_params(looking_for)
Returns the parameters of the Attribute with name looking_for in self.attrs
class sympy.codegen.ast.NoneToken
The AST equivalence of Python’s NoneType
The corresponding instance of Python’s None is none.
Examples
class sympy.codegen.ast.Pointer
Represents a pointer. See Variable.
Examples
class sympy.codegen.ast.Print
Represents print command in the code.
Parameters formatstring : str
*args : Basic instances (or convertible to such through sympify)
Examples
class sympy.codegen.ast.QuotedString
Represents a string which should be printed with quotes.
class sympy.codegen.ast.Return
Represents a return command in the code.
class sympy.codegen.ast.Scope
Represents a scope in the code.
Parameters body : CodeBlock or iterable
When passed an iterable it is used to instantiate a CodeBlock.
class sympy.codegen.ast.SignedIntType
Represents a signed integer type.
class sympy.codegen.ast.Stream
Represents a stream.
There are two predefined Stream instances stdout & stderr.
Parameters name : str
Examples
class sympy.codegen.ast.String
SymPy object representing a string.
Atomic object which is not an expression (as opposed to Symbol).
Parameters text : str
Examples
class sympy.codegen.ast.Token
Base class for the AST types.
Defining fields are set in __slots__. Attributes (defined in __slots__) are only allowed
to contain instances of Basic (unless atomic, see String). The arguments to __new__()
correspond to the attributes in the order defined in __slots__`. The ``defaults class
attribute is a dictionary mapping attribute names to their default values.
Subclasses should not need to override the __new__() method. They may define a class
or static method named _construct_<attr> for each attribute to process the value
passed to __new__(). Attributes listed in the class attribute not_in_args are not passed
to sympy.Basic.
kwargs(exclude=(), apply=None)
Get instance’s attributes as dict of keyword arguments.
Parameters exclude : collection of str
Collection of keywords to exclude.
apply : callable, optional
Function to apply to all values.
class sympy.codegen.ast.Type
Represents a type.
The naming is a super-set of NumPy naming. Type has a classmethod from_expr which
offer type deduction. It also has a method cast_check which casts the argument to its
type, possibly raising an exception if rounding error is not within tolerances, or if the
value is not representable by the underlying data type (e.g. unsigned integers).
Parameters name : str
Name of the type, e.g. object, int16, float16 (where the latter two
would use the Type sub-classes IntType and FloatType respectively).
If a Type instance is given, the said instance is returned.
References
[R25]
Examples
Examples
classmethod from_expr(expr)
Deduces type from an expression or a Symbol.
Parameters expr : number or SymPy object
The type will be deduced from type or properties.
Raises ValueError when type deduction fails.
Examples
class sympy.codegen.ast.UnsignedIntType
Represents an unsigned integer type.
class sympy.codegen.ast.Variable
Represents a variable
Parameters symbol : Symbol
type : Type (optional)
Type of the variable.
Examples
One may also construct a Variable instance with the type deduced from assumptions
about the symbol using the deduced classmethod:
as_Declaration(**kwargs)
Convenience method for creating a Declaration instance.
If the variable of the Declaration need to wrap a modified variable keyword argu-
ments may be passed (overriding e.g. the value of the Variable instance).
Examples
Examples
class sympy.codegen.ast.While
Represents a ‘for-loop’ in the code.
Expressions are of the form:
“while condition: body…”
Examples
Examples
This module contains SymPy functions mathcin corresponding to special math functions in
the C standard library (since C99, also available in C++11).
The functions defined in this module allows the user to express functions such as expm1 as a
SymPy function for symbolic manipulation.
class sympy.codegen.cfunctions.Cbrt
Represents the cube root function.
The reason why one would use Cbrt(x) over cbrt(x) is that the latter is internally rep-
resented as Pow(x, Rational(1, 3)) which may not be what one wants when doing
code-generation.
See also:
Sqrt (page 430)
Examples
fdiff(argindex=1)
Returns the first derivative of this function.
class sympy.codegen.cfunctions.Sqrt
Represents the square root function.
The reason why one would use Sqrt(x) over sqrt(x) is that the latter is internally
represented as Pow(x, S.Half) which may not be what one wants when doing code-
generation.
See also:
Cbrt (page 429)
Examples
fdiff(argindex=1)
Returns the first derivative of this function.
class sympy.codegen.cfunctions.exp2
Represents the exponential function with base two.
The benefit of using exp2(x) over 2**x is that the latter is not as efficient under finite
precision arithmetic.
See also:
log2 (page 432)
Examples
fdiff(argindex=1)
Returns the first derivative of this function.
class sympy.codegen.cfunctions.expm1
Represents the exponential function minus one.
The benefit of using expm1(x) over exp(x) - 1 is that the latter is prone to cancellation
under finite precision arithmetic when x is close to zero.
See also:
log1p (page 432)
Examples
fdiff(argindex=1)
Returns the first derivative of this function.
class sympy.codegen.cfunctions.fma
Represents “fused multiply add”.
The benefit of using fma(x, y, z) over x*y + z is that, under finite precision arith-
metic, the former is supported by special instructions on some CPUs.
Examples
fdiff(argindex=1)
Returns the first derivative of this function.
class sympy.codegen.cfunctions.hypot
Represents the hypotenuse function.
The hypotenuse function is provided by e.g. the math library in the C99 standard, hence
one may want to represent the function symbolically when doing code-generation.
Examples
fdiff(argindex=1)
Returns the first derivative of this function.
class sympy.codegen.cfunctions.log10
Represents the logarithm function with base ten.
See also:
log2 (page 432)
Examples
fdiff(argindex=1)
Returns the first derivative of this function.
class sympy.codegen.cfunctions.log1p
Represents the natural logarithm of a number plus one.
The benefit of using log1p(x) over log(x + 1) is that the latter is prone to cancellation
under finite precision arithmetic when x is close to zero.
See also:
expm1 (page 430)
Examples
fdiff(argindex=1)
Returns the first derivative of this function.
class sympy.codegen.cfunctions.log2
Represents the logarithm function with base two.
The benefit of using log2(x) over log(x)/log(2) is that the latter is not as efficient
under finite precision arithmetic.
See also:
exp2 (page 430), log10 (page 431)
Examples
fdiff(argindex=1)
Returns the first derivative of this function.
Examples
class sympy.codegen.cnodes.PostDecrement
Represents the post-decrement operator
class sympy.codegen.cnodes.PostIncrement
Represents the post-increment operator
class sympy.codegen.cnodes.PreDecrement
Represents the pre-decrement operator
Examples
class sympy.codegen.cnodes.PreIncrement
Represents the pre-increment operator
sympy.codegen.cnodes.alignof(arg)
Generate of FunctionCall instance for calling ‘alignof’
class sympy.codegen.cnodes.goto
Represents goto in C
sympy.codegen.cnodes.sizeof(arg)
Generate of FunctionCall instance for calling ‘sizeof’
Examples
class sympy.codegen.cnodes.struct
Represents a struct in C
class sympy.codegen.cnodes.union
Represents a union in C
Examples
class sympy.codegen.fnodes.Do
Represents a Do loop in in Fortran
Examples
class sympy.codegen.fnodes.Extent
Represents a dimension extent.
Examples
class sympy.codegen.fnodes.FortranReturn
AST node explicitly mapped to a fortran “return”.
Because a return statement in fortran is different from C, and in order to aid reuse of
our codegen ASTs the ordinary .codegen.ast.Return is interpreted as assignment to
the result variable of the function. If one for some reason needs to generate a fortran
RETURN statement, this node should be used.
Examples
class sympy.codegen.fnodes.GoTo
Represents a goto statement in Fortran
Examples
class sympy.codegen.fnodes.ImpliedDoLoop
Represents an implied do loop in Fortran
Examples
class sympy.codegen.fnodes.Module
Represents a module in Fortran
Examples
class sympy.codegen.fnodes.Program
Represents a ‘program’ block in Fortran
Examples
class sympy.codegen.fnodes.Subroutine
Represents a subroutine in Fortran
Examples
class sympy.codegen.fnodes.SubroutineCall
Represents a call to a subroutine in Fortran
Examples
sympy.codegen.fnodes.allocated(array)
Creates an AST node for a function call to Fortran’s “allocated(…)”
Examples
Examples
sympy.codegen.fnodes.bind_C(name=None)
Creates an Attribute bind_C with a name
Parameters name : str
Examples
class sympy.codegen.fnodes.cmplx
Fortran complex conversion function.
sympy.codegen.fnodes.dimension(*args)
Creates a ‘dimension’ Attribute with (up to 7) extents.
Examples
class sympy.codegen.fnodes.dsign
Fortran sign intrinsic for double precision arguments.
class sympy.codegen.fnodes.isign
Fortran sign intrinsic for integer arguments.
class sympy.codegen.fnodes.kind
Fortran kind function.
sympy.codegen.fnodes.lbound(array, dim=None, kind=None)
Creates an AST node for a function call to Fortran’s “lbound(…)”
Parameters array : Symbol or String
dim : expr
kind : expr
Examples
class sympy.codegen.fnodes.literal_dp
Fortran double precision real literal
Attributes
is_irrational
is_rational
class sympy.codegen.fnodes.literal_sp
Fortran single precision real literal
Attributes
is_irrational
is_rational
class sympy.codegen.fnodes.merge
Fortran merge function
sympy.codegen.fnodes.reshape(source, shape, pad=None, order=None)
Creates an AST node for a function call to Fortran’s “reshape(…)”
Parameters source : Symbol or String
shape : ArrayExpr
sympy.codegen.fnodes.shape(source, kind=None)
Creates an AST node for a function call to Fortran’s “shape(…)”
Examples
Examples
class sympy.codegen.fnodes.use
Represents a use statement in Fortran
Examples
class sympy.codegen.fnodes.use_rename
Represents a renaming in a use statement in Fortran
Examples
References
[R26]
Examples
Examples
sympy.codegen.pyutils.render_as_module(content)
Renders python code as a module (with the required imports)
sympy.codegen.cutils.render_as_source_file(content, Printer=<class
’sympy.printing.ccode.C99CodePrinter’>,
settings=None)
Renders a C source file (with required #include statements)
Symbolic computer algebra systems like SymPy facilitate the construction and manipulation
of mathematical expressions. Unfortunately when it comes time to evaluate these expressions
on numerical data, symbolic systems often have poor performance.
Fortunately SymPy offers a number of easy-to-use hooks into other numeric systems, allowing
you to create mathematical expressions in SymPy and then ship them off to the numeric
system of your choice. This page documents many of the options available including the math
library, the popular array computing package numpy, code generation in Fortran or C, and
the use of the array compiler Theano.
5.9.1 Subs/evalf
Subs is the slowest but simplest option. It runs at SymPy speeds. The .subs(...).evalf()
method can substitute a numeric value for a symbolic one and then evaluate the result within
SymPy.
This method is slow. You should use this method production only if performance is not an
issue. You can expect .subs to take tens of microseconds. It can be useful while prototyping
or if you just want to see a value once.
5.9.2 Lambdify
The lambdify function translates SymPy expressions into Python functions, leveraging a va-
riety of numerical libraries. It is used as follows:
Here lambdify makes a function that computes f(x) = sin(x)/x. By default lambdify relies
on implementations in the math standard library. This numerical evaluation takes on the order
of hundreds of nanoseconds, roughly two orders of magnitude faster than the .subs method.
This is the speed difference between SymPy and raw Python.
Lambdify can leverage a variety of numerical backends. By default it uses the math library.
However it also supports mpmath and most notably, numpy. Using the numpy library gives the
generated function access to powerful vectorized ufuncs that are backed by compiled C code.
If you have array-based data this can confer a considerable speedup, on the order of 10 nano-
seconds per element. Unfortunately numpy incurs some start-up time and introduces an
overhead of a few microseconds.
5.9.3 uFuncify
5.9.4 Theano
SymPy has a strong connection with Theano, a mathematical array compiler. SymPy expres-
sions can be easily translated to Theano graphs and then compiled using the Theano compiler
chain.
If array broadcasting or types are desired then Theano requires this extra information
Theano has a more sophisticated code generation system than SymPy’s C/Fortran code print-
ers. Among other things it handles common sub-expressions and compilation onto the GPU.
Theano also supports SymPy Matrix and Matrix Expression objects.
The options here were listed in order from slowest and least dependencies to fastest and
most dependencies. For example, if you have Theano installed then that will often be the best
choice. If you don’t have Theano but do have f2py then you should use ufuncify.
All functions support the methods documented below, inherited from sympy.core.function.
Function (page 189).
class sympy.core.function.Function
Base class for applied mathematical functions.
It also serves as a constructor for undefined function classes.
Examples
First example shows how to use Function as a constructor for undefined function classes:
>>> from sympy import Function, Symbol
>>> x = Symbol('x')
>>> f = Function('f')
>>> g = Function('g')(x)
>>> f
f
>>> f(x)
f(x)
>>> g
g(x)
>>> f(x).diff(x)
Derivative(f(x), x)
>>> g.diff(x)
Derivative(g(x), x)
Note that assumptions on a function are unrelated to the assumptions on the variable it is
called on. If you want to add a relationship, subclass Function and define the appropriate
_eval_is_assumption methods.
In the following example Function is used as a base class for my_func that represents a
mathematical function my_func. Suppose that it is well known, that my_func(0) is 1 and
my_func at infinity goes to 0, so we want those two simplifications to occur automatically.
Suppose also that my_func(x) is real exactly when x is real. Here is an implementation
that honours those requirements:
>>> from sympy import Function, S, oo, I, sin
>>> class my_func(Function):
...
... @classmethod
... def eval(cls, x):
... if x.is_Number:
... if x is S.Zero:
... return S.One
... elif x is S.Infinity:
... return S.Zero
...
(continues on next page)
In order for my_func to become useful, several other methods would need to be im-
plemented. See source code of some of the already implemented functions for more
complete examples.
Also, if the function can take more than one argument, then nargs must be defined, e.g.
if my_func can take one or two arguments then,
as_base_exp()
Returns the method as the 2-tuple (base, exponent).
fdiff(argindex=1)
Returns the first derivative of the function.
is_commutative
Returns whether the function is commutative.
5.10.1 Contents
Elementary
This module implements elementary functions such as trigonometric, hyperbolic, and sqrt, as
well as functions like Abs, Max, Min etc.
sympy.functions.elementary.complexes
re
class sympy.functions.elementary.complexes.re
Returns real part of expression. This function performs only elementary analysis and so
it will fail to decompose properly more complicated expressions. If completely simplified
result is needed then use Basic.as_real_imag() or perform complex expansion on instance
of this function.
See also:
im (page 448)
Examples
as_real_imag(deep=True, **hints)
Returns the real number with a zero imaginary part.
im
class sympy.functions.elementary.complexes.im
Returns imaginary part of expression. This function performs only elementary analysis
and so it will fail to decompose properly more complicated expressions. If completely
simplified result is needed then use Basic.as_real_imag() or perform complex expansion
on instance of this function.
See also:
re (page 447)
Examples
as_real_imag(deep=True, **hints)
Return the imaginary part with a zero real part.
Examples
sign
class sympy.functions.elementary.complexes.sign
Returns the complex sign of an expression:
If the expression is real the sign will be:
• 1 if expression is positive
• 0 if expression is equal to zero
• -1 if expression is negative
If the expression is imaginary the sign will be:
• I if im(expression) is positive
• -I if im(expression) is negative
Otherwise an unevaluated expression will be returned. When evaluated, the result (in
general) will be cos(arg(expr)) + I*sin(arg(expr)).
See also:
Abs (page 449), conjugate (page 451)
Examples
>>> sign(-1)
-1
>>> sign(0)
0
>>> sign(-3*I)
-I
>>> sign(1 + I)
sign(1 + I)
>>> _.evalf()
0.707106781186548 + 0.707106781186548*I
Abs
class sympy.functions.elementary.complexes.Abs
Return the absolute value of the argument.
This is an extension of the built-in function abs() to accept symbolic values. If you pass
a SymPy expression to the built-in abs(), it will pass it automatically to Abs().
See also:
sign (page 449), conjugate (page 451)
Examples
Note that the Python built-in will return either an Expr or int depending on the argument:
>>> type(abs(-1))
<... 'int'>
>>> type(abs(S.NegativeOne))
<class 'sympy.core.numbers.One'>
Examples
arg
class sympy.functions.elementary.complexes.arg
Returns the argument (in radians) of a complex number. For a positive number, the
argument is always 0.
Examples
conjugate
class sympy.functions.elementary.complexes.conjugate
Returns the complexconjugate Ref[1] of an argument. In mathematics, the complex conju-
gate of a complex number is given by changing the sign of the imaginary part.
Thus, the conjugate of the complex number a + ib (where a and b are real numbers) is
a − ib
See also:
sign (page 449), Abs (page 449)
References
[R153]
Examples
polar_lift
class sympy.functions.elementary.complexes.polar_lift
Lift argument to the Riemann surface of the logarithm, using the standard branch.
>>> polar_lift(4*x)
4*polar_lift(x)
>>> polar_lift(4*p)
4*p
See also:
sympy.functions.elementary.exponential.exp_polar, periodic_argument
(page 452)
periodic_argument
class sympy.functions.elementary.complexes.periodic_argument
Represent the argument on a quotient of the Riemann surface of the logarithm. That is,
given a period P, always return a value in (-P/2, P/2], by using exp(P*I) == 1.
See also:
sympy.functions.elementary.exponential.exp_polar
polar_lift (page 451) Lift argument to the Riemann surface of the logarithm
principal_branch (page 452)
principal_branch
class sympy.functions.elementary.complexes.principal_branch
Represent a polar number reduced to its principal branch on a quotient of the Riemann
surface of the logarithm.
This is a function of two arguments. The first argument is a polar number z, and the
second one a positive real number of infinity, p. The result is “z mod exp_polar(I*p)”.
See also:
sympy.functions.elementary.exponential.exp_polar
polar_lift (page 451) Lift argument to the Riemann surface of the logarithm
periodic_argument (page 452)
sympy.functions.elementary.trigonometric
Trigonometric Functions
sin
class sympy.functions.elementary.trigonometric.sin
The sine function.
Returns the sine of x (measured in radians).
See also:
csc (page 456), cos (page 453), sec (page 456), tan (page 454), cot (page 455), asin
(page 458), acsc (page 461), acos (page 458), asec (page 460), atan (page 459), acot
(page 460), atan2 (page 462)
Notes
This function will evaluate automatically in the case x/pi is some rational number [R157].
For example, if x is a multiple of pi, pi/2, pi/3, pi/4 and pi/6.
References
Examples
Members
cos
class sympy.functions.elementary.trigonometric.cos
The cosine function.
Returns the cosine of x (measured in radians).
See also:
sin (page 453), csc (page 456), sec (page 456), tan (page 454), cot (page 455), asin
(page 458), acsc (page 461), acos (page 458), asec (page 460), atan (page 459), acot
(page 460), atan2 (page 462)
Notes
References
Examples
Members
tan
class sympy.functions.elementary.trigonometric.tan
The tangent function.
Returns the tangent of x (measured in radians).
See also:
sin (page 453), csc (page 456), cos (page 453), sec (page 456), cot (page 455), asin
(page 458), acsc (page 461), acos (page 458), asec (page 460), atan (page 459), acot
(page 460), atan2 (page 462)
Notes
References
Examples
Members
cot
class sympy.functions.elementary.trigonometric.cot
The cotangent function.
Returns the cotangent of x (measured in radians).
See also:
sin (page 453), csc (page 456), cos (page 453), sec (page 456), tan (page 454), asin
(page 458), acsc (page 461), acos (page 458), asec (page 460), atan (page 459), acot
(page 460), atan2 (page 462)
Notes
References
Examples
Members
sec
class sympy.functions.elementary.trigonometric.sec
The secant function.
Returns the secant of x (measured in radians).
See also:
sin (page 453), csc (page 456), cos (page 453), tan (page 454), cot (page 455), asin
(page 458), acsc (page 461), acos (page 458), asec (page 460), atan (page 459), acot
(page 460), atan2 (page 462)
Notes
References
Examples
Members
csc
class sympy.functions.elementary.trigonometric.csc
The cosecant function.
Returns the cosecant of x (measured in radians).
See also:
sin (page 453), cos (page 453), sec (page 456), tan (page 454), cot (page 455), asin
(page 458), acsc (page 461), acos (page 458), asec (page 460), atan (page 459), acot
(page 460), atan2 (page 462)
Notes
References
Examples
Members
sinc
class sympy.functions.elementary.trigonometric.sinc
Represents unnormalized sinc function
References
[R173]
Examples
• Automated Evaluation
>>> sinc(0)
1
>>> sinc(oo)
0
• Differentiation
>>> sinc(x).diff()
(x*cos(x) - sin(x))/x**2
• Series Expansion
>>> sinc(x).series()
1 - x**2/6 + x**4/120 + O(x**6)
>>> sinc(x).rewrite(jn)
jn(0, x)
Members
Trigonometric Inverses
asin
class sympy.functions.elementary.trigonometric.asin
The inverse sine function.
Returns the arcsine of x in radians.
See also:
sin (page 453), csc (page 456), cos (page 453), sec (page 456), tan (page 454), cot
(page 455), acsc (page 461), acos (page 458), asec (page 460), atan (page 459), acot
(page 460), atan2 (page 462)
Notes
asin(x) will evaluate automatically in the cases oo, -oo, 0, 1, -1 and for some instances
when the result is a rational multiple of pi (see the eval class method).
References
Examples
Members
acos
class sympy.functions.elementary.trigonometric.acos
The inverse cosine function.
Returns the arc cosine of x (measured in radians).
See also:
sin (page 453), csc (page 456), cos (page 453), sec (page 456), tan (page 454), cot
(page 455), asin (page 458), acsc (page 461), asec (page 460), atan (page 459), acot
(page 460), atan2 (page 462)
Notes
References
Examples
Members
atan
class sympy.functions.elementary.trigonometric.atan
The inverse tangent function.
Returns the arc tangent of x (measured in radians).
See also:
sin (page 453), csc (page 456), cos (page 453), sec (page 456), tan (page 454), cot
(page 455), asin (page 458), acsc (page 461), acos (page 458), asec (page 460), acot
(page 460), atan2 (page 462)
Notes
References
Examples
Members
acot
class sympy.functions.elementary.trigonometric.acot
The inverse cotangent function.
Returns the arc cotangent of x (measured in radians).
See also:
sin (page 453), csc (page 456), cos (page 453), sec (page 456), tan (page 454), cot
(page 455), asin (page 458), acsc (page 461), acos (page 458), asec (page 460), atan
(page 459), atan2 (page 462)
References
asec
class sympy.functions.elementary.trigonometric.asec
The inverse secant function.
Returns the arc secant of x (measured in radians).
See also:
sin (page 453), csc (page 456), cos (page 453), sec (page 456), tan (page 454), cot
(page 455), asin (page 458), acsc (page 461), acos (page 458), atan (page 459), acot
(page 460), atan2 (page 462)
Notes
At x = 0, for positive branch cut, the limit evaluates to zoo. For negative branch cut,
the limit
√
lim −i ∗ (log(− 1 − z 2 + 1)/z)
z→0
References
Examples
Members
acsc
class sympy.functions.elementary.trigonometric.acsc
The inverse cosecant function.
Returns the arc cosecant of x (measured in radians).
See also:
sin (page 453), csc (page 456), cos (page 453), sec (page 456), tan (page 454), cot
(page 455), asin (page 458), acos (page 458), asec (page 460), atan (page 459), acot
(page 460), atan2 (page 462)
Notes
References
Examples
Members
atan2
class sympy.functions.elementary.trigonometric.atan2
The function atan2(y, x) computes atan(y/x) taking two arguments y and x. Signs of
both y and x are considered to determine the appropriate quadrant of atan(y/x). The
range is (−π, π]. The complete definition reads as follows:
( )
arctan xy x>0
(y)
arctan + π y ≥ 0, x < 0
arctan ( y ) − π
x
x y < 0, x < 0
atan2(y, x) =
π
+ 2
y > 0, x = 0
− π
2
y < 0, x = 0
undefined y = 0, x = 0
Attention: Note the role reversal of both arguments. The y-coordinate is the first argu-
ment and the x-coordinate the second.
See also:
sin (page 453), csc (page 456), cos (page 453), sec (page 456), tan (page 454), cot
(page 455), asin (page 458), acsc (page 461), acos (page 458), asec (page 460), atan
(page 459), acot (page 460)
References
Examples
Going counter-clock wise around the origin we find the following angles:
>>> from sympy import atan2
>>> atan2(0, 1)
0
>>> atan2(1, 1)
pi/4
>>> atan2(1, 0)
pi/2
>>> atan2(1, -1)
3*pi/4
>>> atan2(0, -1)
pi
>>> atan2(-1, -1)
(continues on next page)
which are all correct. Compare this to the results of the ordinary atan function for the
point (x, y) = (−1, 1)
where only the atan2 function reurns what we expect. We can differentiate the function
with respect to both arguments:
but note that this form is undefined on the negative real axis.
Members
sympy.functions.elementary.hyperbolic
Hyperbolic Functions
HyperbolicFunction
class sympy.functions.elementary.hyperbolic.HyperbolicFunction
Base class for hyperbolic functions.
See also:
sinh (page 464), cosh (page 464), tanh (page 464), coth (page 464)
Members
sinh
class sympy.functions.elementary.hyperbolic.sinh
−x
The hyperbolic sine function, e −e
x
2 .
• sinh(x) -> Returns the hyperbolic sine of x
See also:
cosh (page 464), tanh (page 464), asinh (page 465)
Members
cosh
class sympy.functions.elementary.hyperbolic.cosh
x −x
The hyperbolic cosine function, e +e
2 .
• cosh(x) -> Returns the hyperbolic cosine of x
See also:
sinh (page 464), tanh (page 464), acosh (page 465)
Members
tanh
class sympy.functions.elementary.hyperbolic.tanh
sinh(x)
The hyperbolic tangent function, cosh(x) .
• tanh(x) -> Returns the hyperbolic tangent of x
See also:
sinh (page 464), cosh (page 464), atanh (page 466)
Members
coth
class sympy.functions.elementary.hyperbolic.coth
The hyperbolic cotangent function, cosh(x)
sinh(x) .
Members
sech
class sympy.functions.elementary.hyperbolic.sech
2
The hyperbolic secant function, ex +e −x
Members
csch
class sympy.functions.elementary.hyperbolic.csch
2
The hyperbolic cosecant function, ex −e −x
Members
Hyperbolic Inverses
asinh
class sympy.functions.elementary.hyperbolic.asinh
The inverse hyperbolic sine function.
• asinh(x) -> Returns the inverse hyperbolic sine of x
See also:
acosh (page 465), atanh (page 466), sinh (page 464)
Members
acosh
class sympy.functions.elementary.hyperbolic.acosh
The inverse hyperbolic cosine function.
• acosh(x) -> Returns the inverse hyperbolic cosine of x
See also:
asinh (page 465), atanh (page 466), cosh (page 464)
Members
atanh
class sympy.functions.elementary.hyperbolic.atanh
The inverse hyperbolic tangent function.
• atanh(x) -> Returns the inverse hyperbolic tangent of x
See also:
asinh (page 465), acosh (page 465), tanh (page 464)
Members
acoth
class sympy.functions.elementary.hyperbolic.acoth
The inverse hyperbolic cotangent function.
• acoth(x) -> Returns the inverse hyperbolic cotangent of x
Members
asech
class sympy.functions.elementary.hyperbolic.asech
The inverse hyperbolic secant function.
• asech(x) -> Returns the inverse hyperbolic secant of x
See also:
asinh (page 465), atanh (page 466), cosh (page 464), acoth (page 466)
References
Examples
Members
acsch
class sympy.functions.elementary.hyperbolic.acsch
The inverse hyperbolic cosecant function.
• acsch(x) -> Returns the inverse hyperbolic cosecant of x
References
Examples
Members
sympy.functions.elementary.integers
ceiling
class sympy.functions.elementary.integers.ceiling
Ceiling is a univariate function which returns the smallest integer value not less than its
argument. This implementation generalizes ceiling to complex numbers by taking the
ceiling of the real and imaginary parts separately.
See also:
sympy.functions.elementary.integers.floor (page 468)
References
[R202], [R203]
Examples
Members
floor
class sympy.functions.elementary.integers.floor
Floor is a univariate function which returns the largest integer value not greater than
its argument. This implementation generalizes floor to complex numbers by taking the
floor of the real and imaginary parts separately.
See also:
sympy.functions.elementary.integers.ceiling (page 467)
References
[R204], [R205]
Examples
Members
RoundFunction
class sympy.functions.elementary.integers.RoundFunction
The base class for rounding functions.
frac
class sympy.functions.elementary.integers.frac
Represents the fractional part of x
For real numbers it is defined [R206] as
x − ⌊x⌋
See also:
sympy.functions.elementary.integers.floor (page 468), sympy.functions.
elementary.integers.ceiling (page 467)
References
[R206], [R207]
Examples
rewrite as floor
>>> x = Symbol('x')
>>> frac(x).rewrite(floor)
x - floor(x)
sympy.functions.elementary.exponential
exp
class sympy.functions.elementary.exponential.exp
The exponential function, ex .
See also:
log (page 470)
Members
LambertW
class sympy.functions.elementary.exponential.LambertW
The Lambert W function W (z) is defined as the inverse function of w exp(w) [R208].
In other words, the value of W (z) is such that z = W (z) exp(W (z)) for any complex number
z. The Lambert W function is a multivalued function with infinitely many branches Wk (z),
indexed by k ∈ Z. Each branch gives a different solution w of the equation z = w exp(w).
The Lambert W function has two partially real branches: the principal branch (k = 0) is
real for real z > −1/e, and the k = −1 branch is real for −1/e < z < 0. All branches except
k = 0 have a logarithmic singularity at z = 0.
References
[R208]
Examples
Members
log
class sympy.functions.elementary.exponential.log
The natural logarithm function ln(x) or log(x). Logarithms are taken with the natural
base, e. To get a logarithm of a different base b, use log(x, b), which is essentially
short-hand for log(x)/log(b).
See also:
exp (page 470)
Members
sympy.functions.elementary.piecewise
ExprCondPair
class sympy.functions.elementary.piecewise.ExprCondPair
Represents an expression, condition pair.
Members
Piecewise
class sympy.functions.elementary.piecewise.Piecewise
Represents a piecewise function.
Usage:
Piecewise( (expr,cond), (expr,cond), … )
• Each argument is a 2-tuple defining an expression and condition
• The conds are evaluated in turn returning the first that is True. If any of
the evaluated conds are not determined explicitly False, e.g. x < 1, the
function is returned in symbolic form.
• If the function is evaluated at a place where all conditions are False, nan
will be returned.
• Pairs where the cond is explicitly False, will be removed.
See also:
piecewise_fold (page 472), ITE
Examples
The folded version of this results in a Piecewise whose expressions are Booleans:
>>> folded_cond = piecewise_fold(cond); folded_cond
Piecewise((2 < y, x < 0), (3 < y, True))
When a Boolean containing Piecewise (like cond) or a Piecewise with Boolean expres-
sions (like folded_cond) is used as a condition, it is converted to an equivalent ITE object:
>>> piecewise_fold(_)
Piecewise((1, ((x >= 0) | (y > 2)) & ((y > 3) | (x < 0))))
Members
sympy.functions.elementary.piecewise.piecewise_fold(expr)
Takes an expression containing a piecewise function and returns the expression in piece-
wise form. In addition, any ITE conditions are rewritten in negation normal form and
simplified.
See also:
Piecewise (page 471)
Examples
sympy.functions.elementary.miscellaneous
IdentityFunction
class sympy.functions.elementary.miscellaneous.IdentityFunction
The identity function
Examples
Members
Min
class sympy.functions.elementary.miscellaneous.Min
Return, if possible, the minimum value of the list. It is named Min and not min to avoid
conflicts with the built-in function min.
See also:
Examples
Members
Max
class sympy.functions.elementary.miscellaneous.Max
Return, if possible, the maximum value of the list.
When number of arguments is equal one, then return this argument.
When number of arguments is equal two, then return, if possible, the value from (a, b)
that is >= the other.
In common case, when the length of list greater than 2, the task is more complicated.
Return only the arguments, which are greater than others, if it is possible to determine
directional relation.
If is not possible to determine such a relation, return a partially evaluated result.
Assumptions are used to make the decision too.
Also, only comparable arguments are permitted.
It is named Max and not max to avoid conflicts with the built-in function max.
See also:
References
[R209], [R210]
Examples
• Algorithm
The task can be considered as searching of supremums in the directed complete partial
orders [R209].
The source values are sequentially allocated by the isolated subsets in which supremums
are searched and result as Max arguments.
If the resulted supremum is single, then it is returned.
The isolated subsets are the sets of values which are only the comparable with each
other in the current set. E.g. natural numbers are comparable with each other, but not
comparable with the x symbol. Another example: the symbol x with negative assumption
is comparable with a natural number.
Also there are “least” elements, which are comparable with all others, and have a zero
property (maximum or minimum for all elements). E.g. oo. In case of it the allocation
operation is terminated and only this value is returned.
Assumption:
• if A > B > C then A > C
• if A == B then B can be removed
Members
root
References
• http://en.wikipedia.org/wiki/Square_root
• http://en.wikipedia.org/wiki/Real_root
• http://en.wikipedia.org/wiki/Root_of_unity
• http://en.wikipedia.org/wiki/Principal_value
• http://mathworld.wolfram.com/CubeRoot.html
Examples
>>> root(x, 2)
sqrt(x)
>>> root(x, 3)
x**(1/3)
>>> root(x, n)
x**(1/n)
To get all n n-th roots you can use the rootof function. The following examples show the
roots of unity for n equal 2, 3 and 4:
>>> from sympy import rootof, I
SymPy, like other symbolic algebra systems, returns the complex root of negative num-
bers. This is the principal root and differs from the text-book result that one might be
expecting. For example, the cube root of -8 does not come back as -2:
>>> root(-8, 3)
2*(-1)**(1/3)
The real_root function can be used to either make the principal result real (or simply to
return the real root directly):
Alternatively, the n//2-th n-th root of a negative number can be computed with root:
sqrt
sympy.functions.elementary.miscellaneous.sqrt(arg, evaluate=None)
The square root function
sqrt(x) -> Returns the principal square root of x.
The parameter evaluate determines if the expression should be evaluated. If None, its
value is taken from global_evaluate
See also:
sympy.polys.rootoftools.rootof (page 895), root (page 475), real_root (page 478)
References
[R211], [R212]
Examples
>>> sqrt(x)
sqrt(x)
>>> sqrt(x)**2
x
>>> sqrt(x**2)
sqrt(x**2)
This is because the two are not equal to each other in general. For example, consider x
== -1:
This is because sqrt computes the principal square root, so the square may put the ar-
gument in a different branch. This identity does hold if x is positive:
You can force this simplification by using the powdenest() function with the force option
set to True:
To get both branches of the square root you can use the rootof function:
cbrt
sympy.functions.elementary.miscellaneous.cbrt(arg, evaluate=None)
This function computes the principal cube root of arg, so it’s just a shortcut for arg ∗
∗Rational(1, 3).
The parameter evaluate determines if the expression should be evaluated. If None, its
value is taken from global_evaluate.
See also:
sympy.polys.rootoftools.rootof (page 895), root (page 475), real_root (page 478)
References
• http://en.wikipedia.org/wiki/Cube_root
• http://en.wikipedia.org/wiki/Principal_value
Examples
>>> cbrt(x)
x**(1/3)
>>> cbrt(x)**3
x
>>> cbrt(x**3)
(x**3)**(1/3)
This is because the two are not equal to each other in general. For example, consider
x == −1:
This is because cbrt computes the principal cube root, this identity does hold if x is
positive:
real_root
Examples
>>> real_root(-8, 3)
-2
>>> root(-8, 3)
2*(-1)**(1/3)
>>> real_root(_)
-2
If one creates a non-principal root and applies real_root, the result will not be real (so
use with caution):
>>> root(-8, 3, 2)
-2*(-1)**(2/3)
>>> real_root(_)
-2*(-1)**(2/3)
Combinatorial
bell
class sympy.functions.combinatorial.numbers.bell
Bell numbers / Bell polynomials
The Bell numbers satisfy B0 = 1 and
∑(
n−1
n−1
)
Bn = Bk .
k
k=0
∑(
n−1
n−1
)
Bn (x) = x Bk−1 (x).
k−1
k=1
The second kind of Bell polynomials (are sometimes called “partial” Bell polynomials or
incomplete Bell polynomials) are defined as
∑ ( x )j 1 ( x )j 2 ( )jn−k+1
n! 1 2 xn−k+1
Bn,k (x1 , x2 , . . . xn−k+1 ) = ··· .
j1 +j2 +j2 +···=k
j1 !j2 ! · · · jn−k+1 ! 1! 2! (n − k + 1)!
j1 +2j2 +3j2 +···=n
See also:
bernoulli (page 480), catalan (page 483), euler (page 484), fibonacci (page 488),
harmonic (page 490), lucas (page 492), genocchi (page 493), partition (page 493),
tribonacci (page 489)
Notes
Not to be confused with Bernoulli numbers and Bernoulli polynomials, which use the
same notation.
References
Examples
bernoulli
class sympy.functions.combinatorial.numbers.bernoulli
Bernoulli numbers / Bernoulli polynomials
The Bernoulli numbers are a sequence of rational numbers defined by B0 = 1 and the
recursive relation (n > 0):
n (
∑ )
n+1
0= Bk
k
k=0
x
They are also commonly defined by their exponential generating function, which is ex −1 .
For odd indices > 1, the Bernoulli numbers are zero.
The Bernoulli polynomials satisfy the analogous formula:
n ( )
∑ n
Bn (x) = Bk xn−k
k
k=0
where:
{
n+3
3 n ≡ 0 or 2 (mod 6)
A(n) =
− n+3
6 n ≡ 4 (mod 6)
and:
[n/6] ( )
∑ n+3
S(n) = Bn−6k
n − 6k
k=1
This formula is similar to the sum given in the definition, but cuts 2/3 of the terms. For
Bernoulli polynomials, we use the formula in the definition.
• bernoulli(n) gives the nth Bernoulli number, Bn
• bernoulli(n, x) gives the nth Bernoulli polynomial in x, Bn (x)
See also:
bell (page 479), catalan (page 483), euler (page 484), fibonacci (page 488), harmonic
(page 490), lucas (page 492), genocchi (page 493), partition (page 493), tribonacci
(page 489)
References
Examples
binomial
class sympy.functions.combinatorial.factorials.binomial
Implementation of the binomial coefficient. It can be defined in two ways depending on
its desired interpretation:
( ) ( )
n n! n f f (n, k)
= or =
k k!(n − k)! k k!
First, in a strict combinatorial sense it defines the number of ways we can choose k
elements from a set of n elements. In this case both arguments are nonnegative integers
and binomial is computed using an efficient algorithm based on prime factorization.
The other definition is generalization for arbitrary n, however k must also be nonnegative.
This case is very useful when evaluating summations.
For the sake of convenience for negative integer k this function will return zero no matter
what valued is the other argument.
References
[R122]
Examples
>>> binomial(15, 8)
6435
>>> N = -4
>>> [binomial(N, i) for i in range(1 - N)]
[1, -4, 10, -20, 35]
>>> binomial(n, 3)
binomial(n, 3)
catalan
class sympy.functions.combinatorial.numbers.catalan
Catalan numbers
The nth catalan number is given by:
( )
1 2n
Cn =
n+1 n
References
Examples
>>> catalan(n)
catalan(n)
Catalan numbers can be transformed into several other, identical expressions involving
other mathematical functions
>>> catalan(n).rewrite(binomial)
binomial(2*n, n)/(n + 1)
>>> catalan(n).rewrite(gamma)
4**n*gamma(n + 1/2)/(sqrt(pi)*gamma(n + 2))
>>> catalan(n).rewrite(hyper)
hyper((-n + 1, -n), (2,), 1)
For some non-integer values of n we can get closed form expressions by rewriting in
terms of gamma functions:
>>> catalan(Rational(1,2)).rewrite(gamma)
8/(3*pi)
We can differentiate the Catalan numbers C(n) interpreted as a continuous real function
in n:
>>> diff(catalan(n), n)
(polygamma(0, n + 1/2) - polygamma(0, n + 2) + log(4))*catalan(n)
As a more advanced example consider the following ratio between consecutive numbers:
>>> catalan(I).rewrite(gamma)
4**I*gamma(1/2 + I)/(sqrt(pi)*gamma(2 + I))
>>> catalan(I).evalf(20)
0.39764993382373624267 - 0.020884341620842555705*I
euler
class sympy.functions.combinatorial.numbers.euler
Euler numbers / Euler polynomials
The Euler numbers are given by:
∑∑
2n+1 k ( )
k (−1)j (k − 2j)2n+1
E2n = I
j 2k I k k
k=1 j=0
E2n+1 = 0
Euler numbers and Euler polynomials are related by
( )
1
En = 2 n En .
2
However, numerical evaluation of the Euler polynomial is computed more efficiently (and
more accurately) using the mpmath library.
• euler(n) gives the nth Euler number, En .
• euler(n, x) gives the nth Euler polynomial, En (x).
See also:
bell (page 479), bernoulli (page 480), catalan (page 483), fibonacci (page 488),
harmonic (page 490), lucas (page 492), genocchi (page 493), partition (page 493),
tribonacci (page 489)
References
Examples
>>> x = Symbol(”x”)
>>> euler(n, x)
euler(n, x)
>>> euler(0, x)
1
>>> euler(1, x)
x - 1/2
>>> euler(2, x)
x**2 - x
>>> euler(3, x)
x**3 - 3*x**2/2 + 1/4
>>> euler(4, x)
x**4 - 2*x**3 + x
factorial
class sympy.functions.combinatorial.factorials.factorial
Implementation of factorial function over nonnegative integers. By convention (con-
sistent with the gamma function and the binomial coefficients), factorial of a negative
integer is complex infinity.
The factorial is very important in combinatorics where it gives the number of ways in
which n objects can be permuted. It also arises in calculus, probability, number theory,
etc.
There is strict relation of factorial with gamma function. In fact n! = gamma(n + 1) for
nonnegative integers. Rewrite of this kind is very useful in case of combinatorial simpli-
fication.
Computation of the factorial is done using two algorithms. For small arguments a pre-
computed look up table is used. However for bigger input algorithm Prime-Swing is used.
It is the fastest algorithm known and computes n! via prime factorization of special class
of numbers, called here the ‘Swing Numbers’.
See also:
factorial2 (page 487), RisingFactorial (page 494), FallingFactorial (page 488)
Examples
>>> factorial(0)
1
>>> factorial(7)
5040
>>> factorial(-2)
zoo
>>> factorial(n)
factorial(n)
>>> factorial(2*n)
factorial(2*n)
>>> factorial(S(1)/2)
factorial(1/2)
subfactorial
class sympy.functions.combinatorial.factorials.subfactorial
The subfactorial counts the derangements of n items and is defined for non-negative
integers as:
1 n=0
!n = 0 n=1
(n − 1)(!(n − 1)+!(n − 2)) n > 1
!x = Γ(x + 1, −1)/e
which is valid for non-negative integers x. The above formula is not very useful incase
of non-integers. Γ(x + 1, −1) is single-valued only for integral arguments x, elsewhere on
the positive real axis it has an infinite number of branches none of which are real.
See also:
sympy.functions.combinatorial.factorials.factorial (page 485), sympy.
utilities.iterables.generate_derangements (page 1461), sympy.functions.
special.gamma_functions.uppergamma (page 510)
References
[R132], [R133]
Examples
class sympy.functions.combinatorial.factorials.factorial2
The double factorial n!!, not to be confused with (n!)!
The double factorial is defined for nonnegative integers and for odd negative integers
as:
1 n=0
n(n − 2)(n − 4) · · · 1 n positive odd
n!! =
n(n − 2)(n − 4) · · · 2
n positive even
(n + 2)!!/(n + 2) n negative odd
See also:
factorial (page 485), RisingFactorial (page 494), FallingFactorial (page 488)
References
[R134]
Examples
FallingFactorial
class sympy.functions.combinatorial.factorials.FallingFactorial
Falling factorial (related to rising factorial) is a double valued function arising in concrete
mathematics, hypergeometric functions and series expansions. It is defined by
f f (x, k) = x · (x − 1) · · · (x − k + 1)
where x can be arbitrary expression and k is an integer. For more information check
“Concrete mathematics” by Graham, pp. 66 or visit http://mathworld.wolfram.com/
FallingFactorial.html page.
When x is a Poly instance of degree >= 1 with single variable, f f (x, k) = x(y) · x(y −
1) · · · x(y −k +1), where y is the variable of x. This is as described in Peter Paule, “Greatest
Factorial Factorization and Symbolic Summation”, Journal of Symbolic Computation, vol.
20, pp. 235-268, 1995.
>>> from sympy import ff, factorial, rf, gamma, polygamma, binomial, symbols, Poly
>>> from sympy.abc import x, k
>>> n, m = symbols('n m', integer=True)
>>> ff(x, 0)
1
>>> ff(5, 5)
120
>>> ff(x, 5) == x*(x-1)*(x-2)*(x-3)*(x-4)
True
>>> ff(Poly(x**2, x), 2)
Poly(x**4 - 2*x**3 + x**2, x, domain='ZZ')
>>> ff(n, n)
factorial(n)
Rewrite
See also:
factorial (page 485), factorial2 (page 487), RisingFactorial (page 494)
References
[R135]
fibonacci
class sympy.functions.combinatorial.numbers.fibonacci
Fibonacci numbers / Fibonacci polynomials
The Fibonacci numbers are the integer sequence defined by the initial terms F0 = 0,
F1 = 1 and the two-term recurrence relation Fn = Fn−1 + Fn−2 . This definition extended
to arbitrary real and complex arguments using the formula
ϕz − cos(πz)ϕ−z
Fz = √
5
The Fibonacci polynomials are defined by F1 (x) = 1, F2 (x) = x, and Fn (x) = x ∗ Fn−1 (x) +
Fn−2 (x) for n > 2. For all positive integers n, Fn (1) = Fn .
• fibonacci(n) gives the nth Fibonacci number, Fn
• fibonacci(n, x) gives the nth Fibonacci polynomial in x, Fn (x)
See also:
bell (page 479), bernoulli (page 480), catalan (page 483), euler (page 484), harmonic
(page 490), lucas (page 492), genocchi (page 493), partition (page 493), tribonacci
(page 489)
References
[R136], [R137]
Examples
tribonacci
class sympy.functions.combinatorial.numbers.tribonacci
Tribonacci numbers / Tribonacci polynomials
The Fibonacci numbers are the integer sequence defined by the initial terms T0 = 0,
T1 = 1, T2 = 1 and the three-term recurrence relation Tn = Tn−1 + Tn−2 + Tn−3 .
The Tribonacci polynomials are defined by T0 (x) = 0, T1 (x) = 1, T2 (x) = x2 , and Tn (x) =
x2 Tn−1 (x) + xTn−2 (x) + Tn−3 (x) for n > 2. For all positive integers n, Tn (1) = Tn .
• tribonacci(n) gives the nth Tribonacci number, Tn
• tribonacci(n, x) gives the nth Tribonacci polynomial in x, Tn (x)
See also:
bell (page 479), bernoulli (page 480), catalan (page 483), euler (page 484), fi-
bonacci (page 488), harmonic (page 490), lucas (page 492), genocchi (page 493), par-
tition (page 493)
References
Examples
harmonic
class sympy.functions.combinatorial.numbers.harmonic
Harmonic numbers
1 1
The nth harmonic number is given by Hn = 1 + 2 + 3 + . . . + n1 .
More generally:
∑
n
1
Hn,m =
km
k=1
References
Examples
>>> harmonic(n).rewrite(Sum)
Sum(1/_k, (_k, 1, n))
We can evaluate harmonic numbers for all integral and positive rational arguments:
>>> H = harmonic(1/S(3))
>>> H
harmonic(1/3)
>>> He = expand_func(H)
>>> He
-log(6) - sqrt(3)*pi/6 + 2*Sum(log(sin(_k*pi/3))*cos(2*_k*pi/3), (_k, 1, 1))
+ 3*Sum(1/(3*_k + 1), (_k, 0, 0))
>>> He.doit()
-log(6) - sqrt(3)*pi/6 - log(sqrt(3)/2) + 3
>>> H = harmonic(25/S(7))
>>> He = simplify(expand_func(H).doit())
>>> He
log(sin(pi/7)**(-2*cos(pi/7))*sin(2*pi/7)**(2*cos(16*pi/7))*cos(pi/14)**(-
,→2*sin(pi/14))/14)
+ pi*tan(pi/14)/2 + 30247/9900
>>> He.n(40)
1.983697455232980674869851942390639915940
>>> harmonic(25/S(7)).n(40)
1.983697455232980674869851942390639915940
>>> harmonic(n).rewrite(digamma)
polygamma(0, n + 1) + EulerGamma
>>> harmonic(n).rewrite(polygamma)
polygamma(0, n + 1) + EulerGamma
>>> harmonic(n,3).rewrite(polygamma)
polygamma(2, n + 1)/2 - polygamma(2, 1)/2
>>> harmonic(n,m).rewrite(polygamma)
(-1)**m*(polygamma(m - 1, 1) - polygamma(m - 1, n + 1))/factorial(m - 1)
>>> expand_func(harmonic(n+4))
harmonic(n) + 1/(n + 4) + 1/(n + 3) + 1/(n + 2) + 1/(n + 1)
>>> expand_func(harmonic(n-4))
harmonic(n) - 1/(n - 1) - 1/(n - 2) - 1/(n - 3) - 1/n
lucas
class sympy.functions.combinatorial.numbers.lucas
Lucas numbers
Lucas numbers satisfy a recurrence relation similar to that of the Fibonacci sequence,
in which each term is the sum of the preceding two. They are generated by choosing the
initial values L0 = 2 and L1 = 1.
• lucas(n) gives the nth Lucas number
See also:
bell (page 479), bernoulli (page 480), catalan (page 483), euler (page 484), fi-
bonacci (page 488), harmonic (page 490), genocchi (page 493), partition (page 493),
tribonacci (page 489)
References
[R144], [R145]
Examples
genocchi
class sympy.functions.combinatorial.numbers.genocchi
Genocchi numbers
The Genocchi numbers are a sequence of integers Gn that satisfy the relation:
∞
∑
2t Gn tn
=
et + 1 n=1 n!
See also:
bell (page 479), bernoulli (page 480), catalan (page 483), euler (page 484), fi-
bonacci (page 488), harmonic (page 490), lucas (page 492), partition (page 493),
tribonacci (page 489)
References
[R146], [R147]
Examples
partition
class sympy.functions.combinatorial.numbers.partition
Partition numbers
The Partition numbers are a sequence of integers pn that represent the number of dis-
tinct ways of representing n as a sum of natural numbers (with order irrelevant). The
generating function for pn is given by:
∞
∑ ∞
∏
pn xn = (1 − xk )−1
n=0 k=1
See also:
bell (page 479), bernoulli (page 480), catalan (page 483), euler (page 484), fi-
bonacci (page 488), harmonic (page 490), lucas (page 492), genocchi (page 493), tri-
bonacci (page 489)
References
[R148], [R149]
Examples
MultiFactorial
class sympy.functions.combinatorial.factorials.MultiFactorial
RisingFactorial
class sympy.functions.combinatorial.factorials.RisingFactorial
Rising factorial (also called Pochhammer symbol) is a double valued function arising in
concrete mathematics, hypergeometric functions and series expansions. It is defined by:
rf (x, k) = x · (x + 1) · · · (x + k − 1)
where x can be arbitrary expression and k is an integer. For more information check
“Concrete mathematics” by Graham, pp. 66 or visit http://mathworld.wolfram.com/
RisingFactorial.html page.
When x is a Poly instance of degree >= 1 with a single variable, rf (x, k) = x(y) · x(y +
1) · · · x(y +k −1), where y is the variable of x. This is as described in Peter Paule, “Greatest
Factorial Factorization and Symbolic Summation”, Journal of Symbolic Computation, vol.
20, pp. 235-268, 1995.
See also:
factorial (page 485), factorial2 (page 487), FallingFactorial (page 488)
References
[R150]
Examples
>>> from sympy import rf, symbols, factorial, ff, binomial, Poly
>>> from sympy.abc import x
>>> n, k = symbols('n k', integer=True)
>>> rf(x, 0)
(continues on next page)
Rewrite
stirling
References
[R151], [R152]
Examples
Second kind:
Enumeration
Three functions are available. Each of them attempts to efficiently compute a given combi-
natorial quantity for a given set or multiset which can be entered as an integer, sequence or
multiset (dictionary with elements as keys and multiplicities as values). The k parameter indi-
cates the number of elements to pick (or the number of partitions to make). When k is None,
the sum of the enumeration for all k (from 0 through the number of items represented by n)
is returned. A replacement parameter is recognized for combinations and permutations; this
indicates that any item may appear with multiplicity as high as the number of items in the
original set.
nC
nP
nT
Note that the integer for n indicates identical items for nT but indicates n different items for
nC and nP.
Special
DiracDelta
class sympy.functions.special.delta_functions.DiracDelta
The DiracDelta function and its derivatives.
DiracDelta is not an ordinary function. It can be rigorously defined either as a distribu-
tion or as a measure.
DiracDelta only makes sense in definite integrals, and in particular, integrals of the form
Integral(f(x)*DiracDelta(x - x0), (x, a, b)), where it equals f(x0) if a <= x0
<= b and 0 otherwise. Formally, DiracDelta acts in some ways like a function that is 0
everywhere except at 0, but in many ways it also does not. It can often be useful to
treat DiracDelta in formal ways, building up and manipulating expressions with delta
functions (which may eventually be integrated), but care must be taken to not treat it
as a real function. SymPy’s oo is similar. It only truly makes sense formally in certain
contexts (such as integration limits), but SymPy allows its use everywhere, and it tries
to be consistent with operations on it (like 1/oo), but it is easy to get into trouble and get
wrong results if oo is treated too much like a number. Similarly, if DiracDelta is treated
too much like a function, it is easy to get wrong or nonsensical results.
DiracDelta function has the following properties:
1) diff(Heaviside(x), x) = DiracDelta(x)
2) integrate(DiracDelta(x - a)*f(x),(x, -oo, oo)) = f(a) and inte-
grate(DiracDelta(x - a)*f(x),(x, a - e, a + e)) = f(a)
3) DiracDelta(x) = 0 for all x != 0
4) DiracDelta(g(x)) = Sum_i(DiracDelta(x - x_i)/abs(g'(x_i))) Where x_i-s
are the roots of g
5) DiracDelta(-x) = DiracDelta(x)
Derivatives of k-th order of DiracDelta have the following property:
6) DiracDelta(x, k) = 0, for all x != 0
7) DiracDelta(-x, k) = -DiracDelta(x, k) for odd k
8) DiracDelta(-x, k) = DiracDelta(x, k) for even k
See also:
Heaviside (page 501), simplify, is_simple (page 500), sympy.functions.special.
tensor_functions.KroneckerDelta (page 576)
References
[R213]
Examples
>>> DiracDelta(x)
DiracDelta(x)
>>> DiracDelta(1)
0
>>> DiracDelta(-1)
0
>>> DiracDelta(pi)
0
>>> DiracDelta(x - 4).subs(x, 4)
DiracDelta(0)
>>> diff(DiracDelta(x))
DiracDelta(x, 1)
>>> diff(DiracDelta(x - 1),x,2)
DiracDelta(x - 1, 2)
>>> diff(DiracDelta(x**2 - 1),x,2)
2*(2*x**2*DiracDelta(x**2 - 1, 2) + DiracDelta(x**2 - 1, 1))
>>> DiracDelta(3*x).is_simple(x)
True
>>> DiracDelta(x**2).is_simple(x)
False
>>> DiracDelta((x**2 - 1)*y).expand(diracdelta=True, wrt=x)
DiracDelta(x - 1)/(2*Abs(y)) + DiracDelta(x + 1)/(2*Abs(y))
Examples
>>> DiracDelta(x)
DiracDelta(x)
>>> DiracDelta(-x, 1)
-DiracDelta(x, 1)
>>> DiracDelta(1)
0
>>> DiracDelta(5, 1)
0
>>> DiracDelta(0)
DiracDelta(0)
>>> DiracDelta(-1)
0
>>> DiracDelta(S.NaN)
nan
>>> DiracDelta(x).eval(1)
0
fdiff(argindex=1)
Returns the first derivative of a DiracDelta Function.
The difference between diff() and fdiff() is:- diff() is the user-level function
and fdiff() is an object method. fdiff() is just a convenience method available in
the Function class. It returns the derivative of the function without considering the
chain rule. diff(function, x) calls Function._eval_derivative which in turn
calls fdiff() internally to compute the derivative of the function.
Examples
>>> DiracDelta(x).fdiff()
DiracDelta(x, 1)
is_simple(self, x)
Tells whether the argument(args[0]) of DiracDelta is a linear expression in x.
x can be:
• a symbol
See also:
simplify, Diracdelta
Examples
>>> DiracDelta(x*y).is_simple(x)
True
>>> DiracDelta(x*y).is_simple(y)
True
>>> DiracDelta(cos(x)).is_simple(x)
False
Heaviside
class sympy.functions.special.delta_functions.Heaviside
Heaviside Piecewise function
Heaviside function has the following properties [R214]:
1) diff(Heaviside(x),x) = DiracDelta(x) ( 0, if x < 0
2) Heaviside(x) = < ( undefined if x==0 [1] ( 1, if x > 0
3) Max(0,x).diff(x) = Heaviside(x)
To specify the value of Heaviside at x=0, a second argument can be given. Omit this 2nd
argument or pass None to recover the default behavior.
>>> from sympy import Heaviside, S
>>> from sympy.abc import x
>>> Heaviside(9)
1
>>> Heaviside(-9)
0
>>> Heaviside(0)
Heaviside(0)
>>> Heaviside(0, S.Half)
1/2
>>> (Heaviside(x) + 1).replace(Heaviside(x), Heaviside(x, 1))
Heaviside(x, 1) + 1
See also:
DiracDelta (page 498)
References
[R215], [R216]
classmethod eval(arg, H0=None)
Returns a simplified form or a value of Heaviside depending on the argument passed
by the Heaviside object.
The eval() method is automatically called when the Heaviside class is about to
be instantiated and it returns either some simplified instance or the unevaluated
instance depending on the argument passed. In other words, eval() method is not
needed to be called explicitly, it is being called and evaluated once the object is
called.
Examples
>>> Heaviside(x)
Heaviside(x)
>>> Heaviside(19)
1
>>> Heaviside(0)
Heaviside(0)
>>> Heaviside(0, 1)
1
>>> Heaviside(-5)
0
>>> Heaviside(S.NaN)
nan
>>> Heaviside(x).eval(100)
1
fdiff(argindex=1)
Returns the first derivative of a Heaviside Function.
Examples
>>> Heaviside(x).fdiff()
DiracDelta(x)
>>> diff(Heaviside(x)).fdiff()
DiracDelta(x, 1)
Singularity Function
class sympy.functions.special.singularity_functions.SingularityFunction
The Singularity functions are a class of discontinuous functions. They take a variable, an
offset and an exponent as arguments. These functions are represented using Macaulay
brackets as :
SingularityFunction(x, a, n) := <x - a>^n
The singularity function will automatically evaluate to Derivative(DiracDelta(x - a),
x, -n - 1) if n < 0 and (x - a)**n*Heaviside(x - a) if n >= 0.
See also:
DiracDelta, Heaviside
Examples
>>> expr.rewrite(Heaviside)
(x - 4)**5*Heaviside(x - 4) + DiracDelta(x + 3) - DiracDelta(x, 1)
>>> expr.rewrite(DiracDelta)
(x - 4)**5*Heaviside(x - 4) + DiracDelta(x + 3) - DiracDelta(x, 1)
>>> expr.rewrite('HeavisideDiracDelta')
(x - 4)**5*Heaviside(x - 4) + DiracDelta(x + 3) - DiracDelta(x, 1)
Reference
Examples
fdiff(argindex=1)
Returns the first derivative of a DiracDelta Function.
The difference between diff() and fdiff() is:- diff() is the user-level function
and fdiff() is an object method. fdiff() is just a convenience method available in
the Function class. It returns the derivative of the function without considering the
chain rule. diff(function, x) calls Function._eval_derivative which in turn
calls fdiff() internally to compute the derivative of the function.
class sympy.functions.special.gamma_functions.gamma
The gamma function
∫ ∞
Γ(x) := tx−1 e−t dt.
0
The gamma function implements the function which passes through the values of the
factorial function, i.e. Γ(n) = (n − 1)! when n is an integer. More general, Γ(z) is defined
in the whole complex plane except at the negative integers where there are simple poles.
See also:
References
Examples
>>> gamma(1)
1
>>> gamma(4)
6
>>> gamma(S(3)/2)
sqrt(pi)/2
We can numerically evaluate the gamma function to arbitrary precision on the whole
complex plane:
>>> gamma(pi).evalf(40)
2.288037795340032417959588909060233922890
>>> gamma(1+I).evalf(20)
0.49801566811835604271 - 0.15494982830181068512*I
class sympy.functions.special.gamma_functions.loggamma
The loggamma function implements the logarithm of the gamma function i.e, log Γ(x).
See also:
References
Examples
Several special values are known. For numerical integral arguments we have:
We can numerically evaluate the gamma function to arbitrary precision on the whole
complex plane:
>>> from sympy import I
>>> loggamma(5).evalf(30)
3.17805383034794561964694160130
>>> loggamma(I).evalf(20)
-0.65092319930185633889 - 1.8724366472624298171*I
class sympy.functions.special.gamma_functions.polygamma
The function polygamma(n, z) returns log(gamma(z)).diff(n + 1).
It is a meromorphic function on C and defined as the (n+1)-th derivative of the logarithm
of the gamma function:
n+1
d
ψ (n) (z) := log Γ(z).
dz n+1
See also:
References
Examples
>>> n = Symbol(”n”)
>>> diff(polygamma(n, x), x)
polygamma(n + 1, x)
>>> diff(polygamma(n, x), x, 2)
polygamma(n + 2, x)
sympy.functions.special.gamma_functions.digamma(x)
The digamma function is the first derivative of the loggamma function i.e,
d Γ′ (z)
ψ(x) := log Γ(z) =
dz Γ(z)
References
References
where γ(s, x) is the lower incomplete gamma function, lowergamma (page 511). This can
be shown to be the same as
( )
xs s
Γ(s, x) = Γ(s) − 1 F1 −x ,
s s+1
where 1 F1 is the (confluent) hypergeometric function.
The upper incomplete gamma function is also essentially equivalent to the generalized
exponential integral:
∫ ∞ −xt
e
En (x) = dt = xn−1 Γ(1 − n, x).
1 tn
See also:
References
Examples
class sympy.functions.special.gamma_functions.lowergamma
The lower incomplete gamma function.
It can be defined as the meromorphic continuation of
∫ x
γ(s, x) := ts−1 e−t dt = Γ(s) − Γ(s, x).
0
References
Examples
class sympy.functions.special.beta_functions.beta
The beta integral is called the Eulerian integral of the first kind by Legendre:
∫ 1
B(x, y) := tx−1 (1 − t)y−1 dt.
0
Beta function or Euler’s first integral is closely associated with gamma function. The
Beta function often used in probability theory and mathematical statistics. It satisfies
properties like:
1
B(a, 1) =
a
B(a, b) = B(b, a)
Γ(a)Γ(b)
B(a, b) =
Γ(a + b)
Therefore for integral values of a and b:
(a − 1)!(b − 1)!
B=
(a + b − 1)!
See also:
References
Examples
We can numerically evaluate the gamma function to arbitrary precision on the whole
complex plane:
>>> beta(1+I,1+I).evalf(20)
-0.2112723729365330143 - 0.7655283165378005676*I
class sympy.functions.special.error_functions.erf
The Gauss error function. This function is defined as:
∫ x
2
e−t dt.
2
erf(x) = √
π 0
See also:
References
Examples
>>> erf(0)
0
>>> erf(oo)
1
>>> erf(-oo)
-1
>>> erf(I*oo)
oo*I
>>> erf(-I*oo)
-oo*I
In general one can pull out factors of -1 and I from the argument:
>>> erf(-z)
-erf(z)
We can numerically evaluate the error function to arbitrary precision on the whole com-
plex plane:
>>> erf(4).evalf(30)
0.999999984582742099719981147840
>>> erf(-4*I).evalf(30)
-1296959.73071763923152794095062*I
class sympy.functions.special.error_functions.erfc
Complementary Error Function. The function is defined as:
∫ ∞
2
e−t dt
2
erfc(x) = √
π x
See also:
References
Examples
>>> erfc(0)
1
>>> erfc(oo)
0
>>> erfc(-oo)
2
>>> erfc(I*oo)
-oo*I
>>> erfc(-I*oo)
oo*I
It also follows
>>> erfc(-z)
-erfc(z) + 2
>>> erfc(4).evalf(30)
0.0000000154172579002800188521596734869
>>> erfc(4*I).evalf(30)
1.0 - 1296959.73071763923152794095062*I
class sympy.functions.special.error_functions.erfi
Imaginary error function. The function erfi is defined as:
∫ x
2 2
erfi(x) = √ et dt
π 0
See also:
References
Examples
>>> erfi(0)
0
>>> erfi(oo)
oo
>>> erfi(-oo)
-oo
>>> erfi(I*oo)
I
>>> erfi(-I*oo)
-I
In general one can pull out factors of -1 and I from the argument:
>>> erfi(-z)
-erfi(z)
We can numerically evaluate the imaginary error function to arbitrary precision on the
whole complex plane:
>>> erfi(2).evalf(30)
18.5648024145755525987042919132
>>> erfi(-2*I).evalf(30)
-0.995322265018952734162069256367*I
class sympy.functions.special.error_functions.erf2
Two-argument error function. This function is defined as:
∫ y
2
e−t dt
2
erf2(x, y) = √
π x
See also:
References
[R261]
Examples
>>> erf2(0, 0)
0
>>> erf2(x, x)
0
>>> erf2(x, oo)
-erf(x) + 1
>>> erf2(x, -oo)
-erf(x) - 1
>>> erf2(oo, y)
erf(y) - 1
>>> erf2(-oo, y)
erf(y) + 1
class sympy.functions.special.error_functions.erfinv
Inverse Error Function. The erfinv function is defined as:
erf(x) = y ⇒ erfinv(y) = x
See also:
References
[R262], [R263]
Examples
We can numerically evaluate the inverse error function to arbitrary precision on [-1, 1]:
>>> erfinv(0.2).evalf(30)
0.179143454621291692285822705344
class sympy.functions.special.error_functions.erfcinv
Inverse Complementary Error Function. The erfcinv function is defined as:
erfc(x) = y ⇒ erfcinv(y) = x
See also:
References
[R264], [R265]
Examples
>>> erfcinv(1)
0
>>> erfcinv(0)
oo
class sympy.functions.special.error_functions.erf2inv
Two-argument Inverse error function. The erf2inv function is defined as:
erf2(x, w) = y ⇒ erf2inv(x, y) = w
See also:
References
[R266]
Examples
>>> erf2inv(0, 0)
0
>>> erf2inv(1, 0)
1
>>> erf2inv(0, 1)
oo
>>> erf2inv(0, y)
erfinv(y)
>>> erf2inv(oo, y)
erfcinv(-y)
class sympy.functions.special.error_functions.FresnelIntegral
Base class for the Fresnel integrals.
class sympy.functions.special.error_functions.fresnels
Fresnel integral S.
This function is defined by
∫ z
π 2
S(z) = sin t dt.
0 2
It is an entire function.
See also:
References
Examples
>>> fresnels(0)
0
>>> fresnels(oo)
1/2
>>> fresnels(-oo)
-1/2
>>> fresnels(I*oo)
-I/2
>>> fresnels(-I*oo)
I/2
In general one can pull out factors of -1 and i from the argument:
>>> fresnels(-z)
-fresnels(z)
>>> fresnels(I*z)
-I*fresnels(z)
We can numerically evaluate the Fresnel integral to arbitrary precision on the whole
complex plane:
>>> fresnels(2).evalf(30)
0.343415678363698242195300815958
>>> fresnels(-2*I).evalf(30)
0.343415678363698242195300815958*I
class sympy.functions.special.error_functions.fresnelc
Fresnel integral C.
This function is defined by
∫ z
π 2
C(z) = cos t dt.
0 2
It is an entire function.
See also:
References
Examples
>>> fresnelc(0)
0
>>> fresnelc(oo)
1/2
>>> fresnelc(-oo)
-1/2
>>> fresnelc(I*oo)
I/2
>>> fresnelc(-I*oo)
-I/2
In general one can pull out factors of -1 and i from the argument:
>>> fresnelc(-z)
-fresnelc(z)
>>> fresnelc(I*z)
I*fresnelc(z)
We can numerically evaluate the Fresnel integral to arbitrary precision on the whole
complex plane:
>>> fresnelc(2).evalf(30)
0.488253406075340754500223503357
>>> fresnelc(-2*I).evalf(30)
-0.488253406075340754500223503357*I
class sympy.functions.special.error_functions.Ei
The classical exponential integral.
For use in SymPy, this function is defined as
∑∞
xn
Ei(x) = + log(x) + γ,
n=1
n n!
If the integral is interpreted as a Cauchy principal value, this statement holds for x > 0
and Ei(x) as defined above.
See also:
References
Examples
>>> Ei(-1)
Ei(-1)
>>> Ei(-1).n(chop=True)
-0.219383934395520
>>> Ei(polar_lift(-1)).n(chop=True)
-0.21938393439552 + 3.14159265358979*I
>>> Ei(x*exp_polar(2*I*pi))
Ei(x) + 2*I*pi
Differentiation is supported:
>>> Ei(x).diff(x)
exp(x)/x
The exponential integral is related to many other special functions. For example:
class sympy.functions.special.error_functions.expint
Generalized exponential integral.
This function is defined as
References
Examples
At non-postive integer orders, the exponential integral reduces to the exponential func-
tion:
>>> expint(0, z)
exp(-z)/z
>>> expint(-1, z)
exp(-z)/z + exp(-z)/z**2
>>> expint(S(1)/2, z)
sqrt(pi)*erfc(sqrt(z))/sqrt(z)
At positive integer orders it can be rewritten in terms of exponentials and expint(1, z).
Use expand_func() to do this:
sympy.functions.special.error_functions.E1(z)
Classical case of the generalized exponential integral.
This is equivalent to expint(1, z).
See also:
class sympy.functions.special.error_functions.li
The classical logarithmic integral.
For the use in SymPy, this function is defined as
∫ x
1
li(x) = dt .
0 log(t)
See also:
References
Examples
>>> li(0)
0
>>> li(1)
-oo
>>> li(oo)
oo
We can numerically evaluate the logarithmic integral to arbitrary precision on the whole
complex plane (except the singular points):
>>> li(2).evalf(30)
1.04516378011749278484458888919
>>> li(2*I).evalf(30)
1.0652795784357498247001125598 + 3.08346052231061726610939702133*I
class sympy.functions.special.error_functions.Li
The offset logarithmic integral.
For the use in SymPy, this function is defined as
See also:
References
Examples
>>> Li(2)
0
We can numerically evaluate the logarithmic integral to arbitrary precision on the whole
complex plane (except the singular points):
>>> Li(2).evalf(30)
0
>>> Li(4).evalf(30)
1.92242131492155809316615998938
class sympy.functions.special.error_functions.Si
Sine integral.
This function is defined by
∫ z
sin t
Si(z) = dt.
0 t
It is an entire function.
See also:
References
[R290]
Examples
It is unbranched:
>>> from sympy import exp_polar, I, pi
>>> Si(z*exp_polar(2*I*pi))
Si(z)
It can also be expressed in terms of exponential integrals, but beware that the latter is
branched:
>>> from sympy import expint
>>> Si(z).rewrite(expint)
-I*(-expint(1, z*exp_polar(-I*pi/2))/2 +
expint(1, z*exp_polar(I*pi/2))/2) + pi/2
class sympy.functions.special.error_functions.Ci
Cosine integral.
This function is defined for positive x by
∫ x ∫ ∞
cos t − 1 cos t
Ci(x) = γ + log x + dt = − dt,
0 t x t
References
[R291]
Examples
>>> Ci(z).diff(z)
cos(z)/z
The cosine integral behaves somewhat like ordinary cos under multiplication by i:
class sympy.functions.special.error_functions.Shi
Sinh integral.
This function is defined by
∫ z
sinh t
Shi(z) = dt.
0 t
It is an entire function.
See also:
References
[R292]
Examples
>>> Shi(z).diff(z)
sinh(z)/z
It is unbranched:
The sinh integral behaves much like ordinary sinh under multiplication by i:
>>> Shi(I*z)
I*Si(z)
>>> Shi(-z)
-Shi(z)
It can also be expressed in terms of exponential integrals, but beware that the latter is
branched:
class sympy.functions.special.error_functions.Chi
Cosh integral.
This function is defined for positive x by
∫ x
cosh t − 1
Chi(x) = γ + log x + dt,
0 t
References
[R293]
Examples
>>> Chi(z).diff(z)
cosh(z)/z
The cosh integral behaves somewhat like ordinary cosh under multiplication by i:
class sympy.functions.special.bessel.BesselBase
Abstract base class for bessel-type functions.
This class is meant to reduce code duplication. All Bessel type functions can 1) be differ-
entiated, and the derivatives expressed in terms of similar functions and 2) be rewritten
in terms of other bessel-type functions.
Here “bessel-type functions” are assumed to have one complex parameter.
To use this base class, define class attributes _a and _b such that 2*F_n' = -_a*F_{n+1}
+ b*F_{n-1}.
argument
The argument of the bessel-type function.
order
The order of the bessel-type function.
class sympy.functions.special.bessel.besselj
Bessel function of the first kind.
The Bessel J function of order ν is defined to be the function satisfying Bessel’s differ-
ential equation
2
d w dw
z2 2
+z + (z 2 − ν 2 )w = 0,
dz dz
See also:
bessely (page 535), besseli (page 536), besselk (page 536)
References
Examples
Differentiate it:
>>> b.diff(z)
besselj(n - 1, z)/2 - besselj(n + 1, z)/2
>>> b.rewrite(jn)
sqrt(2)*sqrt(z)*jn(n - 1/2, z)/sqrt(pi)
>>> b.order
n
>>> b.argument
z
class sympy.functions.special.bessel.bessely
Bessel function of the second kind.
The Bessel Y function of order ν is defined as
Jµ (z) cos(πµ) − J−µ (z)
Yν (z) = lim ,
µ→ν sin(πµ)
References
[R298]
Examples
class sympy.functions.special.bessel.besseli
Modified Bessel function of the first kind.
The Bessel I function is a solution to the modified Bessel equation
2
d w dw
z2 +z + (z 2 + ν 2 )2 w = 0.
dz 2 dz
It can be defined as
Iν (z) = i−ν Jν (iz),
where Jν (z) is the Bessel function of the first kind.
See also:
besselj (page 534), bessely (page 535), besselk (page 536)
References
[R299]
Examples
class sympy.functions.special.bessel.besselk
Modified Bessel function of the second kind.
The Bessel K function of order ν is defined as
π I−µ (z) − Iµ (z)
Kν (z) = lim ,
µ→ν 2 sin(πµ)
where Iµ (z) is the modified Bessel function of the first kind.
It is a solution of the modified Bessel equation, and linearly independent from Yν .
See also:
besselj (page 534), besseli (page 536), bessely (page 535)
References
[R300]
Examples
class sympy.functions.special.bessel.hankel1
Hankel function of the first kind.
This function is defined as
where Jν (z) is the Bessel function of the first kind, and Yν (z) is the Bessel function of the
second kind.
It is a solution to Bessel’s equation.
See also:
hankel2 (page 537), besselj (page 534), bessely (page 535)
References
[R301]
Examples
class sympy.functions.special.bessel.hankel2
Hankel function of the second kind.
This function is defined as
where Jν (z) is the Bessel function of the first kind, and Yν (z) is the Bessel function of the
second kind.
(1)
It is a solution to Bessel’s equation, and linearly independent from Hν .
See also:
hankel1 (page 537), besselj (page 534), bessely (page 535)
References
[R302]
Examples
class sympy.functions.special.bessel.jn
Spherical Bessel function of the first kind.
This function is a solution to the spherical Bessel equation
2
d w dw
z2 2
+ 2z + (z 2 − ν(ν + 1))w = 0.
dz dz
It can be defined as
√
π
jν (z) = J 1 (z),
2z ν+ 2
where Jν (z) is the Bessel function of the first kind.
The spherical Bessel functions of integral order are calculated using the formula:
References
[R303]
Examples
>>> from sympy import Symbol, jn, sin, cos, expand_func, besselj, bessely
>>> from sympy import simplify
>>> z = Symbol(”z”)
>>> nu = Symbol(”nu”, integer=True)
>>> print(expand_func(jn(0, z)))
sin(z)/z
>>> expand_func(jn(1, z)) == sin(z)/z**2 - cos(z)/z
True
>>> expand_func(jn(3, z))
(-6/z**2 + 15/z**4)*sin(z) + (1/z - 15/z**3)*cos(z)
>>> jn(nu, z).rewrite(besselj)
(continues on next page)
class sympy.functions.special.bessel.yn
Spherical Bessel function of the second kind.
This function is another solution to the spherical Bessel equation, and linearly indepen-
dent from jn . It can be defined as
√
π
yν (z) = Y 1 (z),
2z ν+ 2
See also:
besselj (page 534), bessely (page 535), besselk (page 536), jn (page 538)
References
[R304]
Examples
>>> from sympy import Symbol, yn, sin, cos, expand_func, besselj, bessely
>>> z = Symbol(”z”)
>>> nu = Symbol(”nu”, integer=True)
>>> print(expand_func(yn(0, z)))
-cos(z)/z
>>> expand_func(yn(1, z)) == -cos(z)/z**2-sin(z)/z
True
>>> yn(nu, z).rewrite(besselj)
(-1)**(nu + 1)*sqrt(2)*sqrt(pi)*sqrt(1/z)*besselj(-nu - 1/2, z)/2
>>> yn(nu, z).rewrite(bessely)
sqrt(2)*sqrt(pi)*sqrt(1/z)*bessely(nu + 1/2, z)/2
>>> yn(2, 5.2+0.3j).evalf(20)
0.18525034196069722536 + 0.014895573969924817587*I
Examples
Airy Functions
class sympy.functions.special.bessel.AiryBase
Abstract base class for Airy functions.
This class is meant to reduce code duplication.
class sympy.functions.special.bessel.airyai
The Airy function Ai of the first kind.
The Airy function Ai(z) is defined to be the function satisfying Airy’s differential equation
2
d w(z)
− zw(z) = 0.
dz 2
Equivalently, for real z
∫ ∞ ( )
1 t3
Ai(z) := cos + zt dt.
π 0 3
See also:
References
Examples
>>> airyai(z)
airyai(z)
>>> airyai(0)
3**(1/3)/(3*gamma(2/3))
>>> from sympy import oo
>>> airyai(oo)
0
>>> airyai(-oo)
0
We can numerically evaluate the Airy function to arbitrary precision on the whole com-
plex plane:
>>> airyai(-2).evalf(50)
0.22740742820168557599192443603787379946077222541710
class sympy.functions.special.bessel.airybi
The Airy function Bi of the second kind.
The Airy function Bi(z) is defined to be the function satisfying Airy’s differential equation
2
d w(z)
− zw(z) = 0.
dz 2
Equivalently, for real z
∫ ∞ ( 3 ) ( 3 )
1 t t
Bi(z) := exp − + zt + sin + zt dt.
π 0 3 3
See also:
References
Examples
>>> airybi(z)
airybi(z)
>>> airybi(0)
3**(5/6)/(3*gamma(2/3))
>>> from sympy import oo
>>> airybi(oo)
oo
>>> airybi(-oo)
0
We can numerically evaluate the Airy function to arbitrary precision on the whole com-
plex plane:
>>> airybi(-2).evalf(50)
-0.41230258795639848808323405461146104203453483447240
class sympy.functions.special.bessel.airyaiprime
′
The derivative Ai of the Airy function of the first kind.
′
The Airy function Ai (z) is defined to be the function
′ d Ai(z)
Ai (z) := .
dz
See also:
References
Examples
>>> airyaiprime(z)
airyaiprime(z)
We can numerically evaluate the Airy function to arbitrary precision on the whole com-
plex plane:
>>> airyaiprime(-2).evalf(50)
0.61825902074169104140626429133247528291577794512415
class sympy.functions.special.bessel.airybiprime
′
The derivative Bi of the Airy function of the first kind.
′
The Airy function Bi (z) is defined to be the function
′ d Bi(z)
Bi (z) := .
dz
See also:
References
Examples
>>> airybiprime(z)
airybiprime(z)
>>> airybiprime(0)
3**(1/6)/gamma(1/3)
>>> from sympy import oo
>>> airybiprime(oo)
oo
>>> airybiprime(-oo)
0
We can numerically evaluate the Airy function to arbitrary precision on the whole com-
plex plane:
>>> airybiprime(-2).evalf(50)
0.27879516692116952268509756941098324140300059345163
B-Splines
sympy.functions.special.bsplines.bspline_basis(d, knots, n, x)
The n-th B-spline at x of degree d with knots.
B-Splines are piecewise polynomials of degree d [R321]. They are defined on a set of
knots, which is a sequence of integers or floats.
The 0th degree splines have a value of one on a single interval:
For a given (d, knots) there are len(knots)-d-1 B-splines defined, that are indexed
by n (starting at 0).
Here is an example of a cubic B-spline:
By repeating knot points, you can introduce discontinuities in the B-splines and their
derivatives:
>>> d = 1
>>> knots = [0, 0, 2, 3, 4]
>>> bspline_basis(d, knots, 0, x)
Piecewise((-x/2 + 1, (x >= 0) & (x <= 2)), (0, True))
It is quite time consuming to construct and evaluate B-splines. If you need to evaluate a
B-splines many times, it is best to lambdify them first:
See also:
bsplines_basis_set
References
[R321]
sympy.functions.special.bsplines.bspline_basis_set(d, knots, x)
Return the len(knots)-d-1 B-splines at x of degree d with knots.
This function returns a list of Piecewise polynomials that are the len(knots)-d-1 B-
splines of degree d for the given knots. This function calls bspline_basis(d, knots,
n, x) for different values of n.
See also:
bsplines_basis
Examples
class sympy.functions.special.zeta_functions.zeta
Hurwitz zeta function (or Riemann zeta function).
For Re(a) > 0 and Re(s) > 1, this function is defined as
∞
∑ 1
ζ(s, a) = ,
n=0
(n + a)s
where the standard choice of argument for n + a is used. For fixed a with Re(a) > 0 the
Hurwitz zeta function admits a meromorphic continuation to all of C, it is an unbranched
function with a simple pole at s = 1.
Analytic continuation to other a is possible under some circumstances, but this is not
typically done.
The Hurwitz zeta function is a special case of the Lerch transcendent:
This formula defines an analytic continuation for all possible values of s and a (also
Re(a) < 0), see the documentation of lerchphi (page 550) for a description of the branch-
ing behavior.
If no value is passed for a, by this function assumes a default value of a = 1, yielding the
Riemann zeta function.
See also:
dirichlet_eta (page 549), lerchphi (page 550), polylog (page 549)
References
[R322], [R323]
Examples
For a = 1 the Hurwitz zeta function reduces to the famous Riemann zeta function:
∑∞
1
ζ(s, 1) = ζ(s) = s
.
n=1
n
The Riemann zeta function can also be expressed using the Dirichlet eta function:
The Riemann zeta function at positive even integer and negative odd integer values is
related to the Bernoulli numbers:
>>> zeta(2)
pi**2/6
>>> zeta(4)
pi**4/90
>>> zeta(-1)
-1/12
B2n (2π)2n
ζ(2n) = (−1)n+1
2(2n)!
Bn+1
ζ(−n) = −
n+1
At negative even integers the Riemann zeta function is zero:
>>> zeta(-4)
0
No closed-form expressions are known at positive odd integers, but numerical evaluation
is possible:
>>> zeta(3).n()
1.20205690315959
However the derivative with respect to s has no useful closed form expression:
The Hurwitz zeta function can be expressed in terms of the Lerch transcendent, sympy.
functions.special.lerchphi:
class sympy.functions.special.zeta_functions.dirichlet_eta
Dirichlet eta function.
For Re(s) > 0, this function is defined as
∑∞
(−1)n
η(s) = .
n=1
ns
References
[R324]
Examples
The Dirichlet eta function is closely related to the Riemann zeta function:
class sympy.functions.special.zeta_functions.polylog
Polylogarithm function.
For |z| < 1 and s ∈ C, the polylogarithm is defined by
∑∞
zn
Lis (z) = ,
n=1
ns
where the standard branch of the argument is used for n. It admits an analytic continu-
ation which is branched at z = 1 (notably not on the sheet of initial definition), z = 0 and
z = ∞.
The name polylogarithm comes from the fact that for s = 1, the polylogarithm is related
to the ordinary logarithm (see examples), and that
∫ z
Lis (t)
Lis+1 (z) = dt.
0 t
See also:
zeta (page 547), lerchphi (page 550)
Examples
For z ∈ {0, 1, −1}, the polylogarithm is automatically expressed using other functions:
class sympy.functions.special.zeta_functions.lerchphi
Lerch transcendent (Lerch phi function).
For Re(a) > 0, |z| < 1 and s ∈ C, the Lerch transcendent is defined as
∞
∑ zn
Φ(z, s, a) = ,
n=0
(n + a)s
where the standard branch of the argument is used for n+a, and by analytic continuation
for other values of the parameters.
A commonly used related function is the Lerch zeta function, defined by
using the standard branch for both log x and log log x (a branch of log log x is needed to
evaluate log xs−1 ). This concludes the analytic continuation. The Lerch transcendent is
thus branched at z ∈ {0, 1, ∞} and a ∈ Z≤0 . For fixed z, a outside these branch points, it is
an entire function of s.
See also:
polylog (page 549), zeta (page 547)
References
Examples
The Lerch transcendent is a fairly general function, for this reason it does not automat-
ically evaluate to simpler functions. Use expand_func() to achieve this.
If z = 1, the Lerch transcendent reduces to the Hurwitz zeta function:
>>> from sympy import lerchphi, expand_func
>>> from sympy.abc import z, s, a
>>> expand_func(lerchphi(1, s, a))
zeta(s, a)
More generally, if z is a root of unity, the Lerch transcendent reduces to a sum of Hurwitz
zeta functions:
>>> expand_func(lerchphi(-1, s, a))
2**(-s)*zeta(s, a/2) - 2**(-s)*zeta(s, a/2 + 1/2)
class sympy.functions.special.zeta_functions.stieltjes
Represents Stieltjes constants, γk that occur in Laurent Series expansion of the Riemann
zeta function.
References
[R328]
Examples
>>> stieltjes(0)
EulerGamma
>>> stieltjes(0, 1)
EulerGamma
>>> stieltjes(n, m)
stieltjes(n, m)
>>> stieltjes(-1)
zoo
Hypergeometric Functions
class sympy.functions.special.hyper.hyper
The (generalized) hypergeometric function is defined by a series where the ratios of
successive terms are a rational function of the summation index. When convergent, it is
continued analytically to the largest possible domain.
The hypergeometric function depends on two vectors of parameters, called the numer-
ator parameters ap , and the denominator parameters bq . It also has an argument z. The
series definition is
( ) ∑ ∞
a1 , · · · , ap (a1 )n · · · (ap )n z n
F z = ,
p q
b1 , · · · , b q (b1 )n · · · (bq )n n!
n=0
References
[R329], [R330]
Examples
The parameters must always be iterables, even if they are vectors of length one or zero:
But of course they may be variables (but if they depend on x then you should not expect
much implemented functionality):
>>> hyper((n, a), (n**2,), x)
hyper((n, a), (n**2,), x)
The hypergeometric function generalizes many named special functions. The function
hyperexpand() tries to express a hypergeometric function using named special functions.
For example:
>>> from sympy import hyperexpand
>>> hyperexpand(hyper([], [], x))
exp(x)
More examples:
>>> from sympy import S
>>> hyperexpand(hyper([], [S(1)/2], -x**2/4))
cos(x)
>>> hyperexpand(x*hyper([S(1)/2, S(1)/2], [S(3)/2], x**2))
asin(x)
ap
Numerator parameters of the hypergeometric function.
argument
Argument of the hypergeometric function.
bq
Denominator parameters of the hypergeometric function.
convergence_statement
Return a condition on z under which the series converges.
eta
A quantity related to the convergence of the series.
radius_of_convergence
Compute the radius of convergence of the defining series.
Note that even if this is not oo, the function may still be evaluated outside of the
radius of convergence by analytic continuation. But if this is zero, then the function
is not actually defined anywhere else.
>>> from sympy.functions import hyper
>>> from sympy.abc import z
>>> hyper((1, 2), [3], z).radius_of_convergence
(continues on next page)
class sympy.functions.special.hyper.meijerg
The Meijer G-function is defined by a Mellin-Barnes type integral that resembles an in-
verse Mellin transform. It generalizes the hypergeometric functions.
The Meijer G-function depends on four sets of parameters. There are “numerator pa-
rameters” a1 , . . . , an and an+1 , . . . , ap , and there are “denominator parameters” b1 , . . . , bm
and bm+1 , . . . , bq . Confusingly, it is traditionally denoted as follows (note the position of
m, n, p, q, and how they relate to the lengths of the four parameter vectors):
( )
m,n a1 , · · · , an an+1 , · · · , ap
Gp,q z .
b1 , · · · , bm bm+1 , · · · , bq
However, in sympy the four parameter vectors are always available separately (see ex-
amples), so that there is no need to keep track of the decorating sub- and super-scripts
on the G symbol.
The G function is defined as the following integral:
∫ ∏m ∏n
1 j=1 Γ(bj − s) j=1 Γ(1 − aj + s)
∏ ∏ z s ds,
2πi L qj=m+1 Γ(1 − bj + s) pj=n+1 Γ(aj − s)
where Γ(z) is the gamma function. There are three possible contours which we will not
describe in detail here (see the references). If the integral converges along more than
one of them the definitions agree. The contours all separate the poles of Γ(1 − aj + s) from
the poles of Γ(bk − s), so in particular the G function is undefined if aj − bk ∈ Z>0 for some
j ≤ n and k ≤ m.
The conditions under which one of the contours yields a convergent integral are compli-
cated and we do not state them here, see the references.
Note: Currently the Meijer G-function constructor does not check any convergence con-
ditions.
See also:
hyper (page 552), sympy.simplify.hyperexpand (page 1207)
References
[R331], [R332]
Examples
As with the hypergeometric function, the parameters may be passed as arbitrary iter-
ables. Vectors of length zero and one also have to be passed as iterables. The parameters
need not be constants, but if they depend on the argument then not much implemented
functionality should be expected.
All the subvectors of parameters are available:
>>> from sympy import pprint
>>> g = meijerg([1], [2], [3], [4], x)
>>> pprint(g, use_unicode=False)
__1, 1 /1 2 | \
/__ | | x|
\_|2, 2 \3 4 | /
>>> g.an
(1,)
>>> g.ap
(1, 2)
>>> g.aother
(2,)
>>> g.bm
(3,)
>>> g.bq
(3, 4)
>>> g.bother
(4,)
The Meijer G-function generalizes the hypergeometric functions. In some cases it can be
expressed in terms of hypergeometric functions, using Slater’s theorem. For example:
>>> from sympy import hyperexpand
>>> from sympy.abc import a, b, c
>>> hyperexpand(meijerg([a], [], [c], [b], x), allow_hyper=True)
x**c*gamma(-a + c + 1)*hyper((-a + c + 1,),
(-b + c + 1,), -x)/gamma(-b + c + 1)
Thus the Meijer G-function also subsumes many named functions as special cases. You
can use expand_func or hyperexpand to (try to) rewrite a Meijer G-function in terms of
named special functions. For example:
>>> from sympy import expand_func, S
>>> expand_func(meijerg([[],[]], [[0],[]], -x))
exp(x)
>>> hyperexpand(meijerg([[],[]], [[S(1)/2],[0]], (x/2)**2))
sin(x)/sqrt(pi)
an
First set of numerator parameters.
aother
Second set of numerator parameters.
ap
Combined numerator parameters.
argument
Argument of the Meijer G-function.
bm
First set of denominator parameters.
bother
Second set of denominator parameters.
bq
Combined denominator parameters.
delta
A quantity related to the convergence region of the integral, c.f. references.
get_period()
Return a number P such that G(x*exp(I*P)) == G(x).
>>> from sympy.functions.special.hyper import meijerg
>>> from sympy.abc import z
>>> from sympy import pi, S
integrand(s)
Get the defining integrand D(s).
is_number
Returns true if expression has numeric data only.
nu
A quantity related to the convergence region of the integral, c.f. references.
Elliptic integrals
class sympy.functions.special.elliptic_integrals.elliptic_k
The complete elliptic integral of the first kind, defined by
( )
K(m) = F π2 m
where F (z|m) is the Legendre incomplete elliptic integral of the first kind.
The function K(m) is a single-valued function on the complex plane with branch cut along
the interval (1, ∞).
Note that our notation defines the incomplete elliptic integral in terms of the parameter
m instead of the elliptic modulus (eccentricity) k. In this case, the parameter m is defined
as m = k 2 .
See also:
elliptic_f (page 558)
References
[R333], [R334]
Examples
class sympy.functions.special.elliptic_integrals.elliptic_f
The Legendre incomplete elliptic integral of the first kind, defined by
∫ z
dt
F (z|m) = √
0 2
1 − m sin t
This function reduces to a complete elliptic integral of the first kind, K(m), when z = π/2.
Note that our notation defines the incomplete elliptic integral in terms of the parameter
m instead of the elliptic modulus (eccentricity) k. In this case, the parameter m is defined
as m = k 2 .
See also:
elliptic_k (page 557)
References
[R335], [R336]
Examples
class sympy.functions.special.elliptic_integrals.elliptic_e
Called with two arguments z and m, evaluates the incomplete elliptic integral of the
second kind, defined by
∫ z√
2
E (z|m) = 1 − m sin tdt
0
Called with a single argument m, evaluates the Legendre complete elliptic integral of
the second kind
(π )
E(m) = E 2 m
The function E(m) is a single-valued function on the complex plane with branch cut along
the interval (1, ∞).
Note that our notation defines the incomplete elliptic integral in terms of the parameter
m instead of the elliptic modulus (eccentricity) k. In this case, the parameter m is defined
as m = k 2 .
References
Examples
class sympy.functions.special.elliptic_integrals.elliptic_pi
Called with three arguments n, z and m, evaluates the Legendre incomplete elliptic in-
tegral of the third kind, defined by
∫ z
dt
Π (n; z|m) = ( ) √
0 2 2
1 − n sin t 1 − m sin t
Called with two arguments n and m, evaluates the complete elliptic integral of the third
kind:
( )
Π (n|m) = Π n; π2 m
Note that our notation defines the incomplete elliptic integral in terms of the parameter
m instead of the elliptic modulus (eccentricity) k. In this case, the parameter m is defined
as m = k 2 .
References
Examples
Mathieu Functions
class sympy.functions.special.mathieu_functions.MathieuBase
Abstract base class for Mathieu functions.
This class is meant to reduce code duplication.
class sympy.functions.special.mathieu_functions.mathieus
The Mathieu Sine function S(a, q, z). This function is one solution of the Mathieu differ-
ential equation:
y(x)′′ + (a − 2q cos(2x))y(x) = 0
References
Examples
>>> mathieus(a, q, z)
mathieus(a, q, z)
>>> mathieus(a, 0, z)
sin(sqrt(a)*z)
class sympy.functions.special.mathieu_functions.mathieuc
The Mathieu Cosine function C(a, q, z). This function is one solution of the Mathieu dif-
ferential equation:
y(x)′′ + (a − 2q cos(2x))y(x) = 0
References
Examples
>>> mathieuc(a, q, z)
mathieuc(a, q, z)
>>> mathieuc(a, 0, z)
cos(sqrt(a)*z)
class sympy.functions.special.mathieu_functions.mathieusprime
The derivative S ′ (a, q, z) of the Mathieu Sine function. This function is one solution of the
Mathieu differential equation:
y(x)′′ + (a − 2q cos(2x))y(x) = 0
References
Examples
>>> mathieusprime(a, q, z)
mathieusprime(a, q, z)
>>> mathieusprime(a, 0, z)
sqrt(a)*cos(sqrt(a)*z)
class sympy.functions.special.mathieu_functions.mathieucprime
The derivative C ′ (a, q, z) of the Mathieu Cosine function. This function is one solution of
the Mathieu differential equation:
y(x)′′ + (a − 2q cos(2x))y(x) = 0
References
Examples
>>> mathieucprime(a, q, z)
mathieucprime(a, q, z)
>>> mathieucprime(a, 0, z)
-sqrt(a)*sin(sqrt(a)*z)
Orthogonal Polynomials
Jacobi Polynomials
class sympy.functions.special.polynomials.jacobi
(α,β)
Jacobi polynomial Pn (x)
(α,β)
jacobi(n, alpha, beta, x) gives the nth Jacobi polynomial in x, Pn (x).
The Jacobi polynomials are orthogonal on [−1, 1] with respect to the weight
α β
(1 − x) (1 + x) .
See also:
gegenbauer (page 565), chebyshevt_root (page 568), chebyshevu (page 567), cheby-
shevu_root (page 568), legendre (page 569), assoc_legendre (page 569), her-
mite (page 570), laguerre (page 571), assoc_laguerre (page 572), sympy.polys.
orthopolys.jacobi_poly (page 900), sympy.polys.orthopolys.gegenbauer_poly
(page 899), sympy.polys.orthopolys.chebyshevt_poly (page 899), sympy.polys.
orthopolys.chebyshevu_poly (page 899), sympy.polys.orthopolys.hermite_poly
(page 900), sympy.polys.orthopolys.legendre_poly (page 900), sympy.polys.
orthopolys.laguerre_poly (page 900)
References
Examples
>>> jacobi(0, a, b, x)
1
>>> jacobi(1, a, b, x)
a/2 - b/2 + x*(a/2 + b/2 + 1)
>>> jacobi(2, a, b, x) # doctest:+SKIP
(a**2/8 - a*b/4 - a/8 + b**2/8 - b/8 + x**2*(a**2/8 + a*b/4 + 7*a/8 +
b**2/8 + 7*b/8 + 3/2) + x*(a**2/4 + 3*a/4 - b**2/4 - 3*b/4) - 1/2)
>>> jacobi(n, a, b, x)
jacobi(n, a, b, x)
>>> jacobi(n, a, a, x)
RisingFactorial(a + 1, n)*gegenbauer(n,
a + 1/2, x)/RisingFactorial(2*a + 1, n)
>>> jacobi(n, 0, 0, x)
legendre(n, x)
>>> jacobi(n, a, b, 0)
2**(-n)*gamma(a + n + 1)*hyper((-b - n, -n), (a + 1,), -1)/(factorial(n)*gamma(a␣
,→+ 1))
>>> jacobi(n, a, b, 1)
RisingFactorial(a + 1, n)/factorial(n)
>>> diff(jacobi(n,a,b,x), x)
(a/2 + b/2 + n/2 + 1/2)*jacobi(n - 1, a + 1, b + 1, x)
sympy.functions.special.polynomials.jacobi_normalized(n, a, b, x)
(α,β)
Jacobi polynomial Pn (x)
(α,β)
jacobi_normalized(n, alpha, beta, x) gives the nth Jacobi polynomial in x, Pn (x).
The Jacobi polynomials are orthogonal on [−1, 1] with respect to the weight
α β
(1 − x) (1 + x) .
This functions returns the polynomials normilzed:
∫ 1
(α,β)
Pm (x)Pn(α,β) (x)(1 − x)α (1 + x)β dx = δm,n
−1
See also:
gegenbauer (page 565), chebyshevt_root (page 568), chebyshevu (page 567), cheby-
shevu_root (page 568), legendre (page 569), assoc_legendre (page 569), her-
mite (page 570), laguerre (page 571), assoc_laguerre (page 572), sympy.polys.
orthopolys.jacobi_poly (page 900), sympy.polys.orthopolys.gegenbauer_poly
(page 899), sympy.polys.orthopolys.chebyshevt_poly (page 899), sympy.polys.
orthopolys.chebyshevu_poly (page 899), sympy.polys.orthopolys.hermite_poly
(page 900), sympy.polys.orthopolys.legendre_poly (page 900), sympy.polys.
orthopolys.laguerre_poly (page 900)
References
Examples
>>> jacobi_normalized(n, a, b, x)
jacobi(n, a, b, x)/sqrt(2**(a + b + 1)*gamma(a + n + 1)*gamma(b + n + 1)/((a + b␣
,→+ 2*n + 1)*factorial(n)*gamma(a + b + n + 1)))
Gegenbauer Polynomials
class sympy.functions.special.polynomials.gegenbauer
(α)
Gegenbauer polynomial Cn (x)
(α)
gegenbauer(n, alpha, x) gives the nth Gegenbauer polynomial in x, Cn (x).
The Gegenbauer polynomials are orthogonal on [−1, 1] with respect to the weight
( )α− 21
1 − x2 .
See also:
jacobi (page 563), chebyshevt_root (page 568), chebyshevu (page 567), cheby-
shevu_root (page 568), legendre (page 569), assoc_legendre (page 569), her-
mite (page 570), laguerre (page 571), assoc_laguerre (page 572), sympy.polys.
orthopolys.jacobi_poly (page 900), sympy.polys.orthopolys.gegenbauer_poly
(page 899), sympy.polys.orthopolys.chebyshevt_poly (page 899), sympy.polys.
orthopolys.chebyshevu_poly (page 899), sympy.polys.orthopolys.hermite_poly
(page 900), sympy.polys.orthopolys.legendre_poly (page 900), sympy.polys.
orthopolys.laguerre_poly (page 900)
References
Examples
>>> gegenbauer(n, a, x)
gegenbauer(n, a, x)
>>> gegenbauer(n, a, -x)
(-1)**n*gegenbauer(n, a, x)
>>> gegenbauer(n, a, 0)
2**n*sqrt(pi)*gamma(a + n/2)/(gamma(a)*gamma(-n/2 + 1/2)*gamma(n + 1))
>>> gegenbauer(n, a, 1)
gamma(2*a + n)/(gamma(2*a)*gamma(n + 1))
Chebyshev Polynomials
class sympy.functions.special.polynomials.chebyshevt
Chebyshev polynomial of the first kind, Tn (x)
chebyshevt(n, x) gives the nth Chebyshev polynomial (of the first kind) in x, Tn (x).
The Chebyshev polynomials of the first kind are orthogonal on [−1, 1] with respect to the
1
weight √1−x 2
.
See also:
jacobi (page 563), gegenbauer (page 565), chebyshevt_root (page 568), chebyshevu
(page 567), chebyshevu_root (page 568), legendre (page 569), assoc_legendre
(page 569), hermite (page 570), laguerre (page 571), assoc_laguerre (page 572),
sympy.polys.orthopolys.jacobi_poly (page 900), sympy.polys.orthopolys.
gegenbauer_poly (page 899), sympy.polys.orthopolys.chebyshevt_poly (page 899),
sympy.polys.orthopolys.chebyshevu_poly (page 899), sympy.polys.orthopolys.
hermite_poly (page 900), sympy.polys.orthopolys.legendre_poly (page 900),
sympy.polys.orthopolys.laguerre_poly (page 900)
References
Examples
>>> chebyshevt(n, x)
chebyshevt(n, x)
>>> chebyshevt(n, -x)
(-1)**n*chebyshevt(n, x)
(continues on next page)
>>> chebyshevt(n, 0)
cos(pi*n/2)
>>> chebyshevt(n, -1)
(-1)**n
class sympy.functions.special.polynomials.chebyshevu
Chebyshev polynomial of the second kind, Un (x)
chebyshevu(n, x) gives the nth Chebyshev polynomial of the second kind in x, Un (x).
The Chebyshev
√ polynomials of the second kind are orthogonal on [−1, 1] with respect to
the weight 1 − x2 .
See also:
jacobi (page 563), gegenbauer (page 565), chebyshevt (page 566), chebyshevt_root
(page 568), chebyshevu_root (page 568), legendre (page 569), assoc_legendre
(page 569), hermite (page 570), laguerre (page 571), assoc_laguerre (page 572),
sympy.polys.orthopolys.jacobi_poly (page 900), sympy.polys.orthopolys.
gegenbauer_poly (page 899), sympy.polys.orthopolys.chebyshevt_poly (page 899),
sympy.polys.orthopolys.chebyshevu_poly (page 899), sympy.polys.orthopolys.
hermite_poly (page 900), sympy.polys.orthopolys.legendre_poly (page 900),
sympy.polys.orthopolys.laguerre_poly (page 900)
References
Examples
>>> chebyshevu(n, x)
chebyshevu(n, x)
>>> chebyshevu(n, -x)
(-1)**n*chebyshevu(n, x)
>>> chebyshevu(-n, x)
-chebyshevu(n - 2, x)
>>> chebyshevu(n, 0)
cos(pi*n/2)
>>> chebyshevu(n, 1)
n + 1
class sympy.functions.special.polynomials.chebyshevt_root
chebyshev_root(n, k) returns the kth root (indexed from zero) of the nth Chebyshev poly-
nomial of the first kind; that is, if 0 <= k < n, chebyshevt(n, chebyshevt_root(n, k)) ==
0.
See also:
jacobi (page 563), gegenbauer (page 565), chebyshevt (page 566), chebyshevu
(page 567), chebyshevu_root (page 568), legendre (page 569), assoc_legendre
(page 569), hermite (page 570), laguerre (page 571), assoc_laguerre (page 572),
sympy.polys.orthopolys.jacobi_poly (page 900), sympy.polys.orthopolys.
gegenbauer_poly (page 899), sympy.polys.orthopolys.chebyshevt_poly (page 899),
sympy.polys.orthopolys.chebyshevu_poly (page 899), sympy.polys.orthopolys.
hermite_poly (page 900), sympy.polys.orthopolys.legendre_poly (page 900),
sympy.polys.orthopolys.laguerre_poly (page 900)
Examples
class sympy.functions.special.polynomials.chebyshevu_root
chebyshevu_root(n, k) returns the kth root (indexed from zero) of the nth Chebyshev
polynomial of the second kind; that is, if 0 <= k < n, chebyshevu(n, chebyshevu_root(n,
k)) == 0.
See also:
chebyshevt (page 566), chebyshevt_root (page 568), chebyshevu (page 567), legen-
dre (page 569), assoc_legendre (page 569), hermite (page 570), laguerre (page 571),
assoc_laguerre (page 572), sympy.polys.orthopolys.jacobi_poly (page 900),
sympy.polys.orthopolys.gegenbauer_poly (page 899), sympy.polys.orthopolys.
chebyshevt_poly (page 899), sympy.polys.orthopolys.chebyshevu_poly (page 899),
sympy.polys.orthopolys.hermite_poly (page 900), sympy.polys.orthopolys.
legendre_poly (page 900), sympy.polys.orthopolys.laguerre_poly (page 900)
Examples
Legendre Polynomials
class sympy.functions.special.polynomials.legendre
legendre(n, x) gives the nth Legendre polynomial of x, Pn (x)
The Legendre polynomials are orthogonal on [-1, 1] with respect to the constant weight
1. They satisfy Pn (1) = 1 for all n; further, Pn is odd for odd n and even for even n.
See also:
jacobi (page 563), gegenbauer (page 565), chebyshevt (page 566), chebyshevt_root
(page 568), chebyshevu (page 567), chebyshevu_root (page 568), assoc_legendre
(page 569), hermite (page 570), laguerre (page 571), assoc_laguerre (page 572),
sympy.polys.orthopolys.jacobi_poly (page 900), sympy.polys.orthopolys.
gegenbauer_poly (page 899), sympy.polys.orthopolys.chebyshevt_poly (page 899),
sympy.polys.orthopolys.chebyshevu_poly (page 899), sympy.polys.orthopolys.
hermite_poly (page 900), sympy.polys.orthopolys.legendre_poly (page 900),
sympy.polys.orthopolys.laguerre_poly (page 900)
References
Examples
class sympy.functions.special.polynomials.assoc_legendre
assoc_legendre(n,m, x) gives Pnm (x), where n and m are the degree and order or an ex-
pression which is related to the nth order Legendre polynomial, Pn (x) in the following
manner:
m
m d Pn (x)
Pnm (x) = (−1)m (1 − x2 ) 2
dxm
Associated Legendre polynomial are orthogonal on [-1, 1] with:
• weight = 1 for the same m, and different n.
References
Examples
Hermite Polynomials
class sympy.functions.special.polynomials.hermite
hermite(n, x) gives the nth Hermite polynomial in x, Hn (x)
( )
The Hermite polynomials are orthogonal on (−∞, ∞) with respect to the weight exp −x2 .
See also:
jacobi (page 563), gegenbauer (page 565), chebyshevt (page 566), chebyshevt_root
(page 568), chebyshevu (page 567), chebyshevu_root (page 568), legendre (page 569),
assoc_legendre (page 569), laguerre (page 571), assoc_laguerre (page 572),
sympy.polys.orthopolys.jacobi_poly (page 900), sympy.polys.orthopolys.
gegenbauer_poly (page 899), sympy.polys.orthopolys.chebyshevt_poly (page 899),
sympy.polys.orthopolys.chebyshevu_poly (page 899), sympy.polys.orthopolys.
hermite_poly (page 900), sympy.polys.orthopolys.legendre_poly (page 900),
sympy.polys.orthopolys.laguerre_poly (page 900)
References
Examples
Laguerre Polynomials
class sympy.functions.special.polynomials.laguerre
Returns the nth Laguerre polynomial in x, Ln (x).
Parameters n : int
Degree of Laguerre polynomial. Must be n >= 0.
See also:
jacobi (page 563), gegenbauer (page 565), chebyshevt (page 566), chebyshevt_root
(page 568), chebyshevu (page 567), chebyshevu_root (page 568), legendre (page 569),
assoc_legendre (page 569), hermite (page 570), assoc_laguerre (page 572),
sympy.polys.orthopolys.jacobi_poly (page 900), sympy.polys.orthopolys.
gegenbauer_poly (page 899), sympy.polys.orthopolys.chebyshevt_poly (page 899),
sympy.polys.orthopolys.chebyshevu_poly (page 899), sympy.polys.orthopolys.
hermite_poly (page 900), sympy.polys.orthopolys.legendre_poly (page 900),
sympy.polys.orthopolys.laguerre_poly (page 900)
References
Examples
>>> laguerre(n, x)
laguerre(n, x)
class sympy.functions.special.polynomials.assoc_laguerre
Returns the nth generalized Laguerre polynomial in x, Ln (x).
Parameters n : int
Degree of Laguerre polynomial. Must be n >= 0.
alpha : Expr
Arbitrary expression. For alpha=0 regular Laguerre polynomials will
be generated.
See also:
jacobi (page 563), gegenbauer (page 565), chebyshevt (page 566), chebyshevt_root
(page 568), chebyshevu (page 567), chebyshevu_root (page 568), legendre (page 569),
assoc_legendre (page 569), hermite (page 570), laguerre (page 571), sympy.polys.
orthopolys.jacobi_poly (page 900), sympy.polys.orthopolys.gegenbauer_poly
(page 899), sympy.polys.orthopolys.chebyshevt_poly (page 899), sympy.polys.
orthopolys.chebyshevu_poly (page 899), sympy.polys.orthopolys.hermite_poly
(page 900), sympy.polys.orthopolys.legendre_poly (page 900), sympy.polys.
orthopolys.laguerre_poly (page 900)
References
Examples
>>> assoc_laguerre(n, a, 0)
binomial(a + n, a)
>>> assoc_laguerre(n, a, x)
assoc_laguerre(n, a, x)
>>> assoc_laguerre(n, 0, x)
laguerre(n, x)
Spherical Harmonics
class sympy.functions.special.spherical_harmonics.Ynm
Spherical harmonics defined as
√
m (2n + 1)(n − m)! m
Yn (θ, φ) := exp(imφ)Pn (cos(θ))
4π(n + m)!
Ynm() gives the spherical harmonic function of order n and m in θ and φ, Ynm (θ, φ). The
four parameters are as follows: n ≥ 0 an integer and m an integer such that −n ≤ m ≤ n
holds. The two angles are real-valued with θ ∈ [0, π] and φ ∈ [0, 2π].
See also:
Ynm_c (page 575), Znm (page 575)
References
Examples
For specific integers n and m we can evaluate the harmonics to more useful expressions
>>> simplify(Ynm(0, 0, theta, phi).expand(func=True))
1/(2*sqrt(pi))
To get back the well known expressions in spherical coordinates we use full expansion
>>> from sympy import Ynm, Symbol, expand_func
>>> from sympy.abc import n,m
>>> theta = Symbol(”theta”)
>>> phi = Symbol(”phi”)
See also:
Ynm (page 573), Znm (page 575)
References
See also:
Ynm (page 573), Ynm_c (page 575)
References
Tensor Functions
sympy.functions.special.tensor_functions.Eijk(*args, **kwargs)
Represent the Levi-Civita symbol.
This is just compatibility wrapper to LeviCivita().
See also:
LeviCivita (page 576)
sympy.functions.special.tensor_functions.eval_levicivita(*args)
Evaluate Levi-Civita symbol.
class sympy.functions.special.tensor_functions.LeviCivita
Represent the Levi-Civita symbol.
For even permutations of indices it returns 1, for odd permutations -1, and for everything
else (a repeated index) it returns 0.
Thus it represents an alternating pseudotensor.
See also:
Eijk (page 576)
Examples
class sympy.functions.special.tensor_functions.KroneckerDelta
The discrete, or Kronecker, delta function.
A function that takes in two integers i and j. It returns 0 if i and j are not equal or it
returns 1 if i and j are equal.
Parameters i : Number, Symbol
The first index of the delta function.
j : Number, Symbol
The second index of the delta function.
See also:
eval (page 577), sympy.functions.special.delta_functions.DiracDelta (page 498)
References
[R407]
Examples
Symbolic indices:
classmethod eval(i, j)
Evaluates the discrete delta function.
Examples
>>> KroneckerDelta(i, j)
KroneckerDelta(i, j)
>>> KroneckerDelta(i, i)
1
>>> KroneckerDelta(i, i + 1)
0
>>> KroneckerDelta(i, i + 1 + k)
KroneckerDelta(i, i + k + 1)
# indirect doctest
indices_contain_equal_information
Returns True if indices are either both above or below fermi.
Examples
is_above_fermi
True if Delta can be non-zero above fermi
See also:
is_below_fermi (page 578), is_only_below_fermi (page 579),
is_only_above_fermi (page 579)
Examples
is_below_fermi
True if Delta can be non-zero below fermi
See also:
is_above_fermi (page 578), is_only_above_fermi (page 579),
is_only_below_fermi (page 579)
Examples
is_only_above_fermi
True if Delta is restricted to above fermi
See also:
is_above_fermi (page 578), is_below_fermi (page 578), is_only_below_fermi
(page 579)
Examples
is_only_below_fermi
True if Delta is restricted to below fermi
See also:
is_above_fermi (page 578), is_below_fermi (page 578), is_only_above_fermi
(page 579)
Examples
killable_index
Returns the index which is preferred to substitute in the final expression.
The index to substitute is the index with less information regarding fermi level. If
indices contain same information, ‘a’ is preferred before ‘b’.
See also:
preferred_index (page 580)
Examples
preferred_index
Returns the index which is preferred to keep in the final expression.
The preferred index is the index with more information regarding fermi level. If
indices contain same information, ‘a’ is preferred before ‘b’.
See also:
killable_index (page 579)
Examples
5.11.1 Introduction
The geometry module for SymPy allows one to create two-dimensional geometrical entities,
such as lines and circles, and query for information about these entities. This could include
asking the area of an ellipse, checking for collinearity of a set of points, or finding the intersec-
tion between two lines. The primary use case of the module involves entities with numerical
values, but it is possible to also use symbolic representations.
The following Python session gives one an idea of how to work with some of the geometry
module.
>>> x = Point(0, 0)
>>> y = Point(a, 0)
>>> z = Point(2*a, b)
>>> t = Triangle(x, y, z)
>>> t.area
a*b/2
>>> t.medians[x]
Segment2D(Point2D(0, 0), Point2D(3*a/2, b/2))
References
• The area property of Polygon and Triangle may return a positive or negative value,
depending on whether or not the points are oriented counter-clockwise or clockwise,
respectively. If you always want a positive value be sure to use the abs function.
• Although Polygon can refer to any type of polygon, the code has been written for simple
polygons. Hence, expect potential problems if dealing with complex polygons (overlap-
ping sides).
• Since SymPy is still in its infancy some things may not simplify properly and hence some
things that should return True (e.g., Point.is_collinear) may not actually do so. Sim-
ilarly, attempting to find the intersection of entities that do intersect may result in an
empty result.
When one deals with symbolic entities, it often happens that an assertion cannot be guaran-
teed. For example, consider the following code:
>>> from sympy import *
>>> from sympy.geometry import *
>>> x,y,z = map(Symbol, 'xyz')
>>> p1,p2,p3 = Point(x, y), Point(y, z), Point(2*x*y, y)
>>> Point.is_collinear(p1, p2, p3)
False
Even though the result is currently False, this is not always true. If the quantity z − y − 2 ∗
y ∗ z + 2 ∗ y ∗ ∗2 == 0 then the points will be collinear. It would be really nice to inform the
user of this because such a quantity may be useful to a user for further calculation and, at
the very least, being nice to know. This could be potentially done by returning an object (e.g.,
GeometryResult) that the user could use. This actually would not involve an extensive amount
of work.
Currently a limited subset of the geometry module has been extended to three dimensions,
but it certainly would be a good addition to extend more. This would probably involve a fair
amount of work since many of the algorithms used are specific to two dimensions.
Geometry Visualization
The plotting module is capable of plotting geometric entities. See Plotting Geometric Entities
(page 1104) in the plotting module entry.
Submodules
Entities
class sympy.geometry.entity.GeometryEntity
The base class for all geometrical entities.
This class doesn’t represent any particular geometric entity, it only provides the imple-
mentation of some methods common to all subclasses.
ambient_dimension
What is the dimension of the space that the object is contained in?
bounds
Return a tuple (xmin, ymin, xmax, ymax) representing the bounding rectangle for
the geometric figure.
encloses(o)
Return True if o is inside (not on or outside) the boundaries of self.
The object will be decomposed into Points and individual Entities need only define
an encloses_point method for their class.
See also:
sympy.geometry.ellipse.Ellipse.encloses_point (page 638), sympy.geometry.
polygon.Polygon.encloses_point (page 653)
Examples
intersection(o)
Returns a list of all of the intersections of self with o.
See also:
sympy.geometry.util.intersection (page 587)
Notes
is_similar(other)
Is this geometrical entity similar to another geometrical entity?
Two entities are similar if a uniform scaling (enlarging or shrinking) of one of the
entities will allow one to obtain the other.
See also:
scale (page 586)
Notes
This method is not intended to be used directly but rather through the ares imilar
function found in util.py. An entity is not required to implement this method. If two
different types of entities can be similar, it is only required that one of them be able
to determine this.
parameter_value(other, t)
Return the parameter corresponding to the given point. Evaluating an arbitrary
point of the entity at this parameter value will return the given point.
Examples
reflect(line)
Reflects an object across a line.
Parameters line: Line
Examples
rotate(angle, pt=None)
Rotate angle radians counterclockwise about Point pt.
The default pt is the origin, Point(0, 0)
See also:
scale (page 586), translate (page 586)
Examples
Examples
translate(x=0, y=0)
Shift the object by adding to the x,y-coordinates the values x and y.
See also:
rotate (page 585), scale (page 586)
Examples
Utils
sympy.geometry.util.intersection(*entities, **kwargs)
The intersection of a collection of GeometryEntity instances.
Parameters entities : sequence of GeometryEntity
pairwise (keyword argument) : Can be either True or False
Returns intersection : list of GeometryEntity
Raises NotImplementedError
When unable to calculate intersection.
See also:
sympy.geometry.entity.GeometryEntity.intersection (page 584)
Notes
The intersection of any geometrical entity with itself should return a list with one item:
the entity in question. An intersection requires two or more entities. If only a single en-
tity is given then the function will return an empty list. It is possible for intersection to miss
intersections that one knows exists because the required quantities were not fully sim-
plified internally. Reals should be converted to Rationals, e.g. Rational(str(real_num))
or else failures due to floating point issues may result.
Case 1: When the keyword argument ‘pairwise’ is False (default value): In this case, the
function returns a list of intersections common to all entities.
Case 2: When the keyword argument ‘pairwise’ is True: In this case, the functions re-
turns a list intersections that occur between any pair of entities.
Examples
sympy.geometry.util.convex_hull(*args, **kwargs)
The convex hull surrounding the Points contained in the list of entities.
Parameters args : a collection of Points, Segments and/or Polygons
Returns convex_hull : Polygon if polygon is True else as a tuple (U, L) where
L and U are the lower and upper hulls, respectively.
See also:
sympy.geometry.point.Point (page 590), sympy.geometry.polygon.Polygon
(page 650)
Notes
This can only be performed on a set of points whose coordinates can be ordered on the
number line.
References
[1] http://en.wikipedia.org/wiki/Graham_scan
[2] Andrew’s Monotone Chain Algorithm (A.M. Andrew, “Another Efficient Algorithm for
Convex Hulls in Two Dimensions”, 1979) http://geomalgorithms.com/a10-_hull-1.html
Examples
sympy.geometry.util.are_similar(e1, e2)
Are two geometrical entities similar.
Can one geometrical entity be uniformly scaled to the other?
Parameters e1 : GeometryEntity
e2 : GeometryEntity
Returns are_similar : boolean
Raises GeometryError
When e1 and e2 cannot be compared.
See also:
sympy.geometry.entity.GeometryEntity.is_similar (page 585)
Notes
Examples
sympy.geometry.util.centroid(*args)
Find the centroid (center of mass) of the collection containing only Points, Segments
or Polygons. The centroid is the weighted average of the individual centroid where the
weights are the lengths (of segments) or areas (of polygons). Overlapping regions will
add to the weight of that region.
If there are no objects (or a mixture of objects) then None is returned.
See also:
sympy.geometry.point.Point (page 590), sympy.geometry.line.Segment (page 617),
sympy.geometry.polygon.Polygon (page 650)
Examples
Stacking 3 polygons on top of each other effectively triples the weight of that polygon:
Stacking the squares vertically above and below p has the same effect:
Points
class sympy.geometry.point.Point
A point in a n-dimensional Euclidean space.
Parameters coords : sequence of n-coordinate values. In the special
case where n=2 or 3, a Point2D or Point3D will be created as appro-
priate.
evaluate : if T rue (default), all floats are turn into
exact types.
dim : number of coordinates the point should have. If coordinates
are unspecified, they are padded with zeros.
on_morph : indicates what should happen when the number of
coordinates of a point need to be changed by adding or removing ze-
ros. Possible values are ′ warn′ , ′ error′ , or ignore (default). No warning
or error is given when ∗args is empty and dim is given. An error is
always raised when trying to remove nonzero coordinates.
Raises TypeError : When instantiating with anything but a Point or sequence
ValueError : when instantiating with a sequence with length < 2 or
when trying to reduce dimensions if keyword onm orph =′ error′ is set.
See also:
Examples
Floats are automatically converted to Rational unless the evaluate flag is False:
Attributes
length
origin: A P oint representing the origin of the appropriately-dimensioned space.
static affine_rank(*args)
The affine rank of a set of points is the dimension of the smallest affine space con-
taining all the points. For example, if the points lie on a line (and are not all the
same) their affine rank is 1. If the points lie on a plane but not a line, their affine
rank is 2. By convention, the empty set has affine rank -1.
ambient_dimension
Number of components this point has.
classmethod are_coplanar(*points)
Return True if there exists a plane in which all the points lie. A trivial True value is
returned if len(points) < 3 or all Points are 2-dimensional.
Parameters A set of points
Returns boolean
Raises ValueError : if less than 3 unique points are given
Examples
canberra_distance(p)
The Canberra Distance from self to point p.
Returns the weighted sum of horizontal and vertical distances to point p.
Parameters p : Point
Returns canberra_distance : The weighted sum of horizontal and vertical
distances to point p. The weight used is the sum of absolute values
of the coordinates.
Raises ValueError when both vectors are zero.
See also:
sympy.geometry.point.Point.distance (page 592)
Examples
distance(p)
The Euclidean distance from self to point p.
Parameters p : Point
Returns distance : number or symbolic expression.
See also:
sympy.geometry.line.Segment.length (page 619), sympy.geometry.point.
Point.taxicab_distance (page 596)
Examples
dot(p)
Return dot product of self with another Point.
equals(other)
Returns whether the coordinates of self and other agree.
evalf(prec=None, **options)
Evaluate the coordinates of the point.
This method will, where possible, create and return a new Point where the co-
ordinates are evaluated as floating point numbers to the precision indicated (de-
fault=15).
Parameters prec : int
Returns point : Point
Examples
intersection(other)
The intersection between this point and another GeometryEntity.
Parameters other : Point
Returns intersection : list of Points
Notes
The return value will either be an empty list if there is no intersection, otherwise it
will contain this point.
Examples
is_collinear(*args)
Returns T rue if there exists a line that contains self and points. Returns F alse other-
wise. A trivially True value is returned if no points are given.
Parameters args : sequence of Points
Returns is_collinear : boolean
See also:
sympy.geometry.line.Line (page 612)
Examples
is_concyclic(*args)
Do self and the given sequence of points lie in a circle?
Returns True if the set of points are concyclic and False otherwise. A trivial value
of True is returned if there are fewer than 2 other points.
Examples
>>> p1, p2, p3, p4 = Point(1, 0), (0, 1), (-1, 0), (0, -1)
>>> p = Point(1, 1)
is_nonzero
True if any coordinate is nonzero, False if every coordinate is zero, and None if it
cannot be determined.
is_scalar_multiple(p)
Returns whether each coordinate of self is a scalar multiple of the corresponding
coordinate in point p.
is_zero
True if every coordinate is zero, False if any coordinate is not zero, and None if it
cannot be determined.
length
Treating a Point as a Line, this returns 0 for the length of a Point.
Examples
midpoint(p)
The midpoint between self and point p.
Parameters p : Point
Returns midpoint : Point
See also:
sympy.geometry.line.Segment.midpoint (page 619)
Examples
n(prec=None, **options)
Evaluate the coordinates of the point.
This method will, where possible, create and return a new Point where the co-
ordinates are evaluated as floating point numbers to the precision indicated (de-
fault=15).
Parameters prec : int
Returns point : Point
Examples
origin
A point of all zeros of the same ambient dimension as the current point
orthogonal_direction
Returns a non-zero point that is orthogonal to the line containing self and the origin.
Examples
static project(a, b)
Project the point a onto the line between the origin and point b along the normal
direction.
Parameters a : Point
b : Point
Returns p : Point
See also:
sympy.geometry.line.LinearEntity.projection (page 611)
Examples
taxicab_distance(p)
The Taxicab Distance from self to point p.
Returns the sum of the horizontal and vertical distances to point p.
Parameters p : Point
Returns taxicab_distance : The sum of the horizontal
and vertical distances to point p.
See also:
sympy.geometry.point.Point.distance (page 592)
Examples
unit
Return the Point that is in the same direction as self and a distance of 1 from the
origin
class sympy.geometry.point.Point2D
A point in a 2-dimensional Euclidean space.
Parameters coords : sequence of 2 coordinate values.
Raises TypeError
When trying to add or subtract points with different dimensions.
When trying to create a point with more than two dimensions. When
intersection is called with object other than a Point.
See also:
Examples
Floats are automatically converted to Rational unless the evaluate flag is False:
Attributes
x
y
length
bounds
Return a tuple (xmin, ymin, xmax, ymax) representing the bounding rectangle for
the geometric figure.
rotate(angle, pt=None)
Rotate angle radians counterclockwise about Point pt.
See also:
rotate (page 597), scale (page 597)
Examples
Examples
transform(matrix)
Return the point after applying the transformation described by the 3x3 Matrix,
matrix.
See also:
geometry.entity.rotate, geometry.entity.scale, geometry.entity.translate
translate(x=0, y=0)
Shift the Point by adding x and y to the coordinates of the Point.
See also:
rotate (page 597), scale (page 597)
Examples
x
Returns the X coordinate of the Point.
Examples
y
Returns the Y coordinate of the Point.
Examples
class sympy.geometry.point.Point3D
A point in a 3-dimensional Euclidean space.
Parameters coords : sequence of 3 coordinate values.
Raises TypeError
When trying to add or subtract points with different dimensions.
When intersection is called with object other than a Point.
Examples
Floats are automatically converted to Rational unless the evaluate flag is False:
Attributes
x
y
z
length
static are_collinear(*points)
Is a sequence of points collinear?
Test whether or not a set of points are collinear. Returns True if the set of points
are collinear, or False otherwise.
Parameters points : sequence of Point
Returns are_collinear : boolean
See also:
sympy.geometry.line.Line3D (page 627)
Examples
direction_cosine(point)
Gives the direction cosine between 2 points
Parameters p : Point3D
Returns list
Examples
direction_ratio(point)
Gives the direction ratio between 2 points
Parameters p : Point3D
Returns list
Examples
intersection(other)
The intersection between this point and another point.
Parameters other : Point
Returns intersection : list of Points
Notes
The return value will either be an empty list if there is no intersection, otherwise it
will contain this point.
Examples
Examples
transform(matrix)
Return the point after applying the transformation described by the 4x4 Matrix,
matrix.
See also:
geometry.entity.rotate, geometry.entity.scale, geometry.entity.translate
translate(x=0, y=0, z=0)
Shift the Point by adding x and y to the coordinates of the Point.
See also:
rotate, scale (page 601)
Examples
x
Returns the X coordinate of the Point.
Examples
y
Returns the Y coordinate of the Point.
Examples
z
Returns the Z coordinate of the Point.
Examples
Lines
class sympy.geometry.line.LinearEntity
A base class for all linear entities (Line, Ray and Segment) in n-dimensional Euclidean
space.
See also:
sympy.geometry.entity.GeometryEntity (page 584)
Notes
Attributes
ambient_dimension
direction
length
p1
p2
points
ambient_dimension
A property method that returns the dimension of LinearEntity object.
Parameters p1 : LinearEntity
Returns dimension : integer
Examples
angle_between(l2)
Return the non-reflex angle formed by rays emanating from the origin with direc-
tions the same as the direction vectors of the linear entities.
Parameters l1 : LinearEntity
l2 : LinearEntity
Returns angle : angle in radians
See also:
is_perpendicular (page 607), Ray2D.closing_angle (page 624)
Notes
Examples
To obtain the non-obtuse angle at the intersection of lines, use the small-
est_angle_between method:
>>> sw.smallest_angle_between(e)
pi/4
arbitrary_point(parameter=’t’)
A parameterized point on the Line.
Parameters parameter : str, optional
The name of the parameter which will be used for the parametric
point. The default value is ‘t’. When this parameter is 0, the first
point used to define the line will be returned, and when it is 1 the
second point will be returned.
Returns point : Point
Raises ValueError
When parameter already appears in the Line’s definition.
See also:
sympy.geometry.point.Point (page 590)
Examples
static are_concurrent(*lines)
Is a sequence of linear entities concurrent?
Two or more linear entities are concurrent if they all intersect at a single point.
Parameters lines : a sequence of linear entities.
Returns True : if the set of linear entities intersect in one point
False : otherwise.
See also:
sympy.geometry.util.intersection (page 587)
Examples
contains(other)
Subclasses should implement this method and should return True if other is on the
boundaries of self; False if not on the boundaries of self; None if a determination
cannot be made.
direction
The direction vector of the LinearEntity.
Returns p : a Point; the ray from the origin to this point is the
direction of self
See also:
sympy.geometry.point.Point.unit (page 596)
Examples
intersection(other)
The intersection with another geometrical entity.
Parameters o : Point or LinearEntity
Returns intersection : list of geometrical entities
See also:
sympy.geometry.point.Point (page 590)
Examples
is_parallel(l2)
Are two linear entities parallel?
Parameters l1 : LinearEntity
l2 : LinearEntity
Returns True : if l1 and l2 are parallel,
False : otherwise.
See also:
coefficients
Examples
is_perpendicular(l2)
Are two linear entities perpendicular?
Parameters l1 : LinearEntity
l2 : LinearEntity
Returns True : if l1 and l2 are perpendicular,
False : otherwise.
See also:
coefficients
Examples
is_similar(other)
Return True if self and other are contained in the same line.
Examples
length
The length of the line.
Examples
p1
The first defining point of a linear entity.
See also:
sympy.geometry.point.Point (page 590)
Examples
p2
The second defining point of a linear entity.
See also:
sympy.geometry.point.Point (page 590)
Examples
parallel_line(p)
Create a new Line parallel to this linear entity which passes through the point p.
Parameters p : Point
Examples
perpendicular_line(p)
Create a new Line perpendicular to this linear entity which passes through the point
p.
Parameters p : Point
Returns line : Line
See also:
is_perpendicular (page 607), perpendicular_segment (page 609)
Examples
perpendicular_segment(p)
Create a perpendicular line segment from p to this line.
The enpoints of the segment are p and the closest point in the line containing self.
(If self is not a line, the point might not be in self.)
Parameters p : Point
Returns segment : Segment
See also:
perpendicular_line (page 609)
Notes
Examples
points
The two points used to define this linear entity.
Returns points : tuple of Points
See also:
sympy.geometry.point.Point (page 590)
Examples
projection(other)
Project a point, line, ray, or segment onto this linear entity.
Parameters other : Point or LinearEntity (Line, Ray, Segment)
Returns projection : Point or LinearEntity (Line, Ray, Segment)
The return type matches the type of the parameter other.
Raises GeometryError
When method is unable to perform projection.
See also:
sympy.geometry.point.Point (page 590), perpendicular_line (page 609)
Notes
A projection involves taking the two points that define the linear entity and pro-
jecting those points onto a Line and then reforming the linear entity using these
projections. A point P is projected onto a line L by finding the point on L that is
closest to P. This point is the intersection of L and the line perpendicular to L that
passes through P.
Examples
random_point(seed=None)
A random point on a LinearEntity.
Returns point : Point
See also:
sympy.geometry.point.Point (page 590)
Examples
smallest_angle_between(l2)
Return the smallest angle formed at the intersection of the lines containing the linear
entities.
Parameters l1 : LinearEntity
l2 : LinearEntity
Returns angle : angle in radians
See also:
angle_between (page 603), Ray2D.closing_angle (page 624)
Examples
class sympy.geometry.line.Line
An infinite line in space.
A line is declared with two distinct points. A 2D line may be declared with a point and
slope and a 3D line may be defined with a point and a direction ratio.
Parameters p1 : Point
p2 : Point
slope : sympy expression
direction_ratio : list
See also:
sympy.geometry.point.Point (page 590), sympy.geometry.line.Line2D (page 622),
sympy.geometry.line.Line3D (page 627)
Notes
Line will automatically subclass to Line2D or Line3D based on the dimension of p1. The
slope argument is only relevant for Line2D and the directionr atio argument is only relevant
for Line3D.
Examples
Attributes
is_Complement
is_EmptySet
is_Intersection
is_UniversalSet
contains(other)
Return True if other is on this Line, or False otherwise.
Examples
distance(other)
Finds the shortest distance between a line and a point.
Raises NotImplementedError is raised if ‘other‘ is not a Point
Examples
equals(other)
Returns True if self and other are the same mathematical entities
plot_interval(parameter=’t’)
The plot interval for the default geometric plot of line. Gives values that will produce
a line that is +/- 5 units long (where a unit is the distance between the two points
that define the line).
Parameters parameter : str, optional
Default value is ‘t’.
Returns plot_interval : list (plot interval)
[parameter, lower_bound, upper_bound]
Examples
class sympy.geometry.line.Ray
A Ray is a semi-line in the space with a source point and a direction.
Parameters p1 : Point
The source of the Ray
p2 : Point or radian value
This point determines the direction in which the Ray propagates. If
given as an angle it is interpreted in radians with the positive direc-
tion being ccw.
See also:
sympy.geometry.line.Ray2D (page 623), sympy.geometry.line.Ray3D (page 628),
sympy.geometry.point.Point (page 590), sympy.geometry.line.Line (page 612)
Notes
Ray will automatically subclass to Ray2D or Ray3D based on the dimension of p1.
Examples
Attributes
source
contains(other)
Is other GeometryEntity contained in this Ray?
Examples
distance(other)
Finds the shortest distance between the ray and a point.
Raises NotImplementedError is raised if ‘other‘ is not a Point
Examples
equals(other)
Returns True if self and other are the same mathematical entities
plot_interval(parameter=’t’)
The plot interval for the default geometric plot of the Ray. Gives values that will
produce a ray that is 10 units long (where a unit is the distance between the two
points that define the ray).
Parameters parameter : str, optional
Examples
source
The point from which the ray emanates.
See also:
sympy.geometry.point.Point (page 590)
Examples
class sympy.geometry.line.Segment
A line segment in space.
Parameters p1 : Point
p2 : Point
See also:
sympy.geometry.line.Segment2D (page 625), sympy.geometry.line.Segment3D
(page 630), sympy.geometry.point.Point (page 590), sympy.geometry.line.Line
(page 612)
Notes
Examples
Attributes
contains(other)
Is the other GeometryEntity contained within this Segment?
Examples
distance(other)
Finds the shortest distance between a line segment and a point.
Raises NotImplementedError is raised if ‘other‘ is not a Point
Examples
equals(other)
Returns True if self and other are the same mathematical entities
length
The length of the line segment.
See also:
sympy.geometry.point.Point.distance (page 592)
Examples
midpoint
The midpoint of the line segment.
See also:
sympy.geometry.point.Point.midpoint (page 594)
Examples
perpendicular_bisector(p=None)
The perpendicular bisector of this segment.
If no point is specified or the point specified is not on the bisector then the bisector
is returned as a Line. Otherwise a Segment is returned that joins the point specified
and the intersection of the bisector and the segment.
Parameters p : Point
Returns bisector : Line or Segment
See also:
LinearEntity.perpendicular_segment (page 609)
Examples
>>> s1.perpendicular_bisector(p3)
Segment2D(Point2D(5, 1), Point2D(3, 3))
plot_interval(parameter=’t’)
The plot interval for the default geometric plot of the Segment gives values that will
produce the full segment in a plot.
Parameters parameter : str, optional
Default value is ‘t’.
Returns plot_interval : list
[parameter, lower_bound, upper_bound]
Examples
class sympy.geometry.line.LinearEntity2D
A base class for all linear entities (line, ray and segment) in a 2-dimensional Euclidean
space.
See also:
Notes
Attributes
p1
p2
coefficients
slope
points
bounds
Return a tuple (xmin, ymin, xmax, ymax) representing the bounding rectangle for
the geometric figure.
perpendicular_line(p)
Create a new Line perpendicular to this linear entity which passes through the point
p.
Parameters p : Point
Returns line : Line
See also:
is_perpendicular, perpendicular_segment
Examples
slope
The slope of this linear entity, or infinity if vertical.
Returns slope : number or sympy expression
See also:
coefficients
Examples
>>> p3 = Point(0, 4)
>>> l2 = Line(p1, p3)
>>> l2.slope
oo
class sympy.geometry.line.Line2D
An infinite line in space 2D.
A line is declared with two distinct points or a point and slope as defined using keyword
slope.
Parameters p1 : Point
pt : Point
slope : sympy expression
See also:
sympy.geometry.point.Point (page 590)
Examples
Attributes
is_Complement
is_EmptySet
is_Intersection
is_UniversalSet
coefficients
The coefficients (a, b, c) for ax + by + c = 0.
See also:
sympy.geometry.line.Line.equation
Examples
>>> p3 = Point(x, y)
>>> l2 = Line(p1, p3)
>>> l2.coefficients
(-y, x, 0)
equation(x=’x’, y=’y’)
The equation of the line: ax + by + c.
Parameters x : str, optional
The name to use for the x-axis, default value is ‘x’.
y : str, optional
The name to use for the y-axis, default value is ‘y’.
Returns equation : sympy expression
See also:
LinearEntity.coefficients
Examples
class sympy.geometry.line.Ray2D
A Ray is a semi-line in the space with a source point and a direction.
Parameters p1 : Point
The source of the Ray
p2 : Point or radian value
This point determines the direction in which the Ray propagates. If
given as an angle it is interpreted in radians with the positive direc-
tion being ccw.
See also:
sympy.geometry.point.Point (page 590), Line (page 612)
Examples
Attributes
source
xdirection
ydirection
closing_angle(r2)
Return the angle by which r2 must be rotated so it faces the same direction as r1.
Parameters r1 : Ray2D
r2 : Ray2D
Returns angle : angle in radians (ccw angle is positive)
See also:
LinearEntity.angle_between (page 603)
Examples
xdirection
The x direction of the ray.
Positive infinity if the ray points in the positive x direction, negative infinity if the
ray points in the negative x direction, or 0 if the ray is vertical.
See also:
ydirection (page 625)
Examples
ydirection
The y direction of the ray.
Positive infinity if the ray points in the positive y direction, negative infinity if the
ray points in the negative y direction, or 0 if the ray is horizontal.
See also:
xdirection (page 625)
Examples
class sympy.geometry.line.Segment2D
A line segment in 2D space.
Parameters p1 : Point
p2 : Point
See also:
sympy.geometry.point.Point (page 590), Line (page 612)
Examples
Attributes
class sympy.geometry.line.LinearEntity3D
An base class for all linear entities (line, ray and segment) in a 3-dimensional Euclidean
space.
Notes
Attributes
p1
p2
direction_ratio
direction_cosine
points
direction_cosine
The normalized direction ratio of a given line in 3D.
See also:
sympy.geometry.line.Line.equation
Examples
direction_ratio
The direction ratio of a given line in 3D.
See also:
sympy.geometry.line.Line.equation
Examples
class sympy.geometry.line.Line3D
An infinite 3D line in space.
A line is declared with two distinct points or a point and direction_ratio as defined using
keyword directionr atio.
Parameters p1 : Point3D
pt : Point3D
direction_ratio : list
See also:
sympy.geometry.point.Point3D (page 598), sympy.geometry.line.Line (page 612),
sympy.geometry.line.Line2D (page 622)
Examples
Attributes
is_Complement
is_EmptySet
is_Intersection
is_UniversalSet
Examples
class sympy.geometry.line.Ray3D
A Ray is a semi-line in the space with a source point and a direction.
Parameters p1 : Point3D
The source of the Ray
p2 : Point or a direction vector
direction_ratio: Determines the direction in which the Ray propa-
gates.
See also:
sympy.geometry.point.Point3D (page 598), Line3D (page 627)
Examples
Attributes
source
xdirection
ydirection
zdirection
xdirection
The x direction of the ray.
Positive infinity if the ray points in the positive x direction, negative infinity if the
ray points in the negative x direction, or 0 if the ray is vertical.
See also:
ydirection (page 629)
Examples
ydirection
The y direction of the ray.
Positive infinity if the ray points in the positive y direction, negative infinity if the
ray points in the negative y direction, or 0 if the ray is horizontal.
See also:
xdirection (page 629)
Examples
zdirection
The z direction of the ray.
Positive infinity if the ray points in the positive z direction, negative infinity if the
ray points in the negative z direction, or 0 if the ray is horizontal.
See also:
xdirection (page 629)
Examples
class sympy.geometry.line.Segment3D
A line segment in a 3D space.
Parameters p1 : Point3D
p2 : Point3D
See also:
sympy.geometry.point.Point3D (page 598), Line3D (page 627)
Examples
Attributes
Curves
class sympy.geometry.curve.Curve
A curve in space.
A curve is defined by parametric functions for the coordinates, a parameter and the lower
and upper bounds for the parameter value.
Parameters function : list of functions
limits : 3-tuple
Function parameter and lower and upper bounds.
Raises ValueError
When f unctions are specified incorrectly. When limits are specified
incorrectly.
See also:
sympy.core.function.Function (page 189), sympy.polys.polyfuncs.interpolate
(page 890)
Examples
Attributes
functions
parameter
limits
arbitrary_point(parameter=’t’)
A parameterized point on the curve.
Parameters parameter : str or Symbol, optional
Default value is ‘t’; the Curve’s parameter is selected with None or
self.parameter otherwise the provided symbol is used.
Returns arbitrary_point : Point
Raises ValueError
When parameter already appears in the functions.
See also:
sympy.geometry.point.Point (page 590)
Examples
free_symbols
Return a set of symbols other than the bound symbols used to parametrically define
the Curve.
Examples
functions
The functions specifying the curve.
Returns functions : list of parameterized coordinate functions.
See also:
parameter (page 633)
Examples
length
The curve length.
Examples
limits
The limits for the curve.
Returns limits : tuple
Contains parameter and lower and upper limits.
See also:
plot_interval (page 634)
Examples
parameter
The curve function variable.
Returns parameter : SymPy symbol
See also:
functions (page 632)
Examples
plot_interval(parameter=’t’)
The plot interval for the default geometric plot of the curve.
Parameters parameter : str or Symbol, optional
Default value is ‘t’; otherwise the provided symbol is used.
Returns plot_interval : list (plot interval)
[parameter, lower_bound, upper_bound]
See also:
Examples
rotate(angle=0, pt=None)
Rotate angle radians counterclockwise about Point pt.
The default pt is the origin, Point(0, 0).
Examples
Examples
translate(x=0, y=0)
Translate the Curve by (x, y).
Examples
Ellipses
class sympy.geometry.ellipse.Ellipse
An elliptical GeometryEntity.
Parameters center : Point, optional
Default value is Point(0, 0)
hradius : number or SymPy expression, optional
vradius : number or SymPy expression, optional
eccentricity : number or SymPy expression, optional
Two of hradius, vradius and eccentricity must be supplied to create an
Ellipse. The third is derived from the two supplied.
Raises GeometryError
When hradius, vradius and eccentricity are incorrectly supplied as pa-
rameters.
TypeError
When center is not a Point.
See also:
Circle (page 647)
Notes
Constructed from a center and two radii, the first being the horizontal radius (along the
x-axis) and the second being the vertical radius (along the y-axis).
When symbolic value for hradius and vradius are used, any calculation that refers to the
foci or the major or minor axis will assume that the ellipse has its major radius on the
x-axis. If this is not true then a manual rotation is necessary.
Examples
Attributes
center
hradius
vradius
area
circumference
eccentricity
periapsis
apoapsis
focus_distance
foci
apoapsis
The apoapsis of the ellipse.
The greatest distance between the focus and the contour.
Returns apoapsis : number
See also:
periapsis (page 643) Returns shortest distance between foci and contour
Examples
arbitrary_point(parameter=’t’)
A parameterized point on the ellipse.
Parameters parameter : str, optional
Default value is ‘t’.
Returns arbitrary_point : Point
Raises ValueError
When parameter already appears in the functions.
See also:
sympy.geometry.point.Point (page 590)
Examples
area
The area of the ellipse.
Returns area : number
Examples
bounds
Return a tuple (xmin, ymin, xmax, ymax) representing the bounding rectangle for
the geometric figure.
center
The center of the ellipse.
Returns center : number
See also:
sympy.geometry.point.Point (page 590)
Examples
circumference
The circumference of the ellipse.
Examples
eccentricity
The eccentricity of the ellipse.
Returns eccentricity : number
Examples
encloses_point(p)
Return True if p is enclosed by (is inside of) self.
Parameters p : Point
Returns encloses_point : True, False or None
See also:
sympy.geometry.point.Point (page 590)
Notes
Examples
equation(x=’x’, y=’y’)
The equation of the ellipse.
Parameters x : str, optional
Label for the x-axis. Default value is ‘x’.
y : str, optional
Label for the y-axis. Default value is ‘y’.
Examples
evolute(x=’x’, y=’y’)
The equation of evolute of the ellipse.
Parameters x : str, optional
Label for the x-axis. Default value is ‘x’.
y : str, optional
Label for the y-axis. Default value is ‘y’.
Returns equation : sympy expression
Examples
foci
The foci of the ellipse.
Raises ValueError
When the major and minor axis cannot be determined.
See also:
sympy.geometry.point.Point (page 590)
focus_distance (page 640) Returns the distance between focus and center
Notes
The foci can only be calculated if the major/minor axes are known.
Examples
focus_distance
The focal distance of the ellipse.
The distance between the center and one focus.
Returns focus_distance : number
See also:
foci (page 639)
Examples
hradius
The horizontal radius of the ellipse.
Returns hradius : number
See also:
vradius (page 647), major (page 641), minor (page 642)
Examples
intersection(o)
The intersection of this ellipse and another geometrical entity o.
Parameters o : GeometryEntity
Returns intersection : list of GeometryEntity objects
See also:
sympy.geometry.entity.GeometryEntity (page 584)
Notes
Currently supports intersections with Point, Line, Segment, Ray, Circle and Ellipse
types.
Examples
is_tangent(o)
Is o tangent to the ellipse?
Parameters o : GeometryEntity
An Ellipse, LinearEntity or Polygon
Returns is_tangent: boolean
True if o is tangent to the ellipse, False otherwise.
Raises NotImplementedError
When the wrong type of argument is supplied.
See also:
tangent_lines (page 646)
Examples
major
Longer axis of the ellipse (if it can be determined) else hradius.
Examples
>>> a = Symbol('a')
>>> b = Symbol('b')
>>> Ellipse(p1, a, b).major
a
>>> Ellipse(p1, b, a).major
b
>>> m = Symbol('m')
>>> M = m + 1
>>> Ellipse(p1, m, M).major
m + 1
minor
Shorter axis of the ellipse (if it can be determined) else vradius.
Returns minor : number or expression
See also:
hradius (page 640), vradius (page 647), major (page 641)
Examples
>>> a = Symbol('a')
>>> b = Symbol('b')
>>> Ellipse(p1, a, b).minor
b
>>> Ellipse(p1, b, a).minor
a
>>> m = Symbol('m')
>>> M = m + 1
>>> Ellipse(p1, m, M).minor
m
normal_lines(p, prec=None)
Normal lines between p and the ellipse.
Parameters p : Point
Returns normal_lines : list with 1, 2 or 4 Lines
Examples
Off-axis points require the solution of a quartic equation. This often leads to very
large expressions that may be of little practical use. An approximate solution of prec
digits can be obtained by passing in the desired value:
Whereas the above solution has an operation count of 12, the exact solution has an
operation count of 2020.
periapsis
The periapsis of the ellipse.
The shortest distance between the focus and the contour.
Returns periapsis : number
See also:
apoapsis (page 636) Returns greatest distance between focus and contour
Examples
plot_interval(parameter=’t’)
The plot interval for the default geometric plot of the Ellipse.
Parameters parameter : str, optional
Default value is ‘t’.
Returns plot_interval : list
[parameter, lower_bound, upper_bound]
Examples
random_point(seed=None)
A random point on the ellipse.
Returns point : Point
See also:
sympy.geometry.point.Point (page 590)
arbitrary_point (page 636) Returns parameterized point on ellipse
Notes
When creating a random point, one may simply replace the parameter with a random
number. When doing so, however, the random number should be made a Rational
or else the point may not test as being in the ellipse:
Examples
reflect(line)
Override GeometryEntity.reflect since the radius is not a GeometryEntity.
Notes
Until the general ellipse (with no axis parallel to the x-axis) is supported a NotIm-
plemented error is raised and the equation whose zeros define the rotated ellipse is
given.
Examples
rotate(angle=0, pt=None)
Rotate angle radians counterclockwise about Point pt.
Note: since the general ellipse is not supported, only rotations that are integer
multiples of pi/2 are allowed.
Examples
Examples
second_moment_of_area(point=None)
Returns the second moment and product moment area of an ellipse.
Parameters point : Point, two-tuple of sympifiable objects, or
None(default=None)
point is the point about which second moment of area is to be found.
If “point=None” it will be calculated about the axis passing through
the centroid of the ellipse.
Returns I_xx, I_yy, I_xy : number or sympy expression
I_xx, I_yy are second moment of area of an ellise. I_xy is product
moment of area of an ellipse.
References
https://en.wikipedia.org/wiki/List_of_second_moments_of_area
Examples
semilatus_rectum
Calculates the semi-latus rectum of the Ellipse.
Semi-latus rectum is defined as one half of the the chord through a focus parallel to
the conic section directrix of a conic section.
Returns semilatus_rectum : number
See also:
apoapsis (page 636) Returns greatest distance between focus and contour
periapsis (page 643) The shortest distance between the focus and the contour
References
Examples
tangent_lines(p)
Tangent lines between p and the ellipse.
If p is on the ellipse, returns the tangent line through point p. Otherwise, returns
the tangent line(s) from p to the ellipse, or None if no tangent line is possible (e.g.,
p inside ellipse).
Parameters p : Point
Returns tangent_lines : list with 1 or 2 Lines
Raises NotImplementedError
Can only find tangent lines for a point, p, on the ellipse.
See also:
sympy.geometry.point.Point (page 590), sympy.geometry.line.Line (page 612)
Examples
vradius
The vertical radius of the ellipse.
Returns vradius : number
See also:
hradius (page 640), major (page 641), minor (page 642)
Examples
class sympy.geometry.ellipse.Circle
A circle in space.
Constructed simply from a center and a radius, or from three non-collinear points.
Parameters center : Point
radius : number or sympy expression
points : sequence of three Points
Raises GeometryError
When trying to construct circle from three collinear points. When
trying to construct circle from incorrect parameters.
See also:
Ellipse (page 635), sympy.geometry.point.Point (page 590)
Examples
Attributes
circumference
The circumference of the circle.
Returns circumference : number or SymPy expression
Examples
equation(x=’x’, y=’y’)
The equation of the circle.
Parameters x : str or Symbol, optional
Default value is ‘x’.
y : str or Symbol, optional
Default value is ‘y’.
Returns equation : SymPy expression
Examples
intersection(o)
The intersection of this circle with another geometrical entity.
Parameters o : GeometryEntity
Returns intersection : list of GeometryEntities
Examples
radius
The radius of the circle.
Returns radius : number or sympy expression
See also:
Ellipse.major (page 641), Ellipse.minor (page 642), Ellipse.hradius
(page 640), Ellipse.vradius (page 647)
Examples
reflect(line)
Override GeometryEntity.reflect since the radius is not a GeometryEntity.
Examples
Examples
vradius
This Ellipse property is an alias for the Circle’s radius.
Whereas hradius, major and minor can use Ellipse’s conventions, the vradius does
not exist for a circle. It is always a positive value in order that the Circle, like
Polygons, will have an area that can be positive or negative as determined by the
sign of the hradius.
Examples
Polygons
class sympy.geometry.polygon.Polygon
A two-dimensional polygon.
A simple polygon in space. Can be constructed from a sequence of points or from a
center, radius, number of sides and rotation angle.
Parameters vertices : sequence of Points
Raises GeometryError
If all parameters are not Points.
See also:
sympy.geometry.point.Point (page 590), sympy.geometry.line.Segment (page 617),
Triangle (page 665)
Notes
Polygons are treated as closed paths rather than 2D areas so some calculations can be
be negative or positive (e.g., area) based on the orientation of the points.
Any consecutive identical points are reduced to a single point and any points collinear
and between two points will be removed unless they are needed to define an explicit
intersection (see examples).
A Triangle, Segment or Point will be returned when there are 3 or fewer points provided.
Examples
The area of a polygon is calculated as positive when vertices are traversed in a ccw
direction. When the sides of a polygon cross the area will have positive and negative
contributions. The following defines a Z shape where the bottom right connects back to
the top left.
When the the keyword n is used to define the number of sides of the Polygon then a
RegularPolygon is created and the other arguments are interpreted as center, radius and
rotation. The unrotated RegularPolygon will always have a vertex at Point(r, 0) where r
is the radius of the circle that circumscribes the RegularPolygon. Its method spin can be
used to increment that angle.
Attributes
area
angles
perimeter
vertices
centroid
sides
angles
The internal angle at each vertex.
Returns angles : dict
A dictionary where each key is a vertex and each value is the internal
angle at that vertex. The vertices are represented as Points.
See also:
sympy.geometry.point.Point (page 590), sympy.geometry.line.LinearEntity.
angle_between (page 603)
Examples
arbitrary_point(parameter=’t’)
A parameterized point on the polygon.
The parameter, varying from 0 to 1, assigns points to the position on the perimeter
that is that fraction of the total perimeter. So the point evaluated at t=1/2 would
return the point from the first vertex that is 1/2 way around the polygon.
Parameters parameter : str, optional
Default value is ‘t’.
Returns arbitrary_point : Point
Raises ValueError
When parameter already appears in the Polygon’s definition.
See also:
sympy.geometry.point.Point (page 590)
Examples
area
The area of the polygon.
See also:
sympy.geometry.ellipse.Ellipse.area (page 637)
Notes
The area calculation can be positive or negative based on the orientation of the
points. If any side of the polygon crosses any other side, there will be areas having
opposite signs.
Examples
In the Z shaped polygon (with the lower right connecting back to the upper left) the
areas cancel out:
In the M shaped polygon, areas do not cancel because no side crosses any other
(though there is a point of contact).
>>> M = Polygon((0, 0), (0, 1), (2, 0), (3, 1), (3, 0))
>>> M.area
-3/2
bounds
Return a tuple (xmin, ymin, xmax, ymax) representing the bounding rectangle for
the geometric figure.
centroid
The centroid of the polygon.
Returns centroid : Point
See also:
sympy.geometry.point.Point (page 590), sympy.geometry.util.centroid
(page 589)
Examples
distance(o)
Returns the shortest distance between self and o.
If o is a point, then self does not need to be convex. If o is another polygon self and
o must be complex.
Examples
encloses_point(p)
Return True if p is enclosed by (is inside of) self.
Parameters p : Point
Returns encloses_point : True, False or None
See also:
Notes
References
[1] http://paulbourke.net/geometry/polygonmesh/#insidepoly
Examples
intersection(o)
The intersection of polygon and geometry entity.
The intersection may be empty and can contain individual Points and complete Line
Segments.
Parameters other: GeometryEntity
Returns intersection : list
The list of Segments and Points
See also:
sympy.geometry.point.Point (page 590), sympy.geometry.line.Segment
(page 617)
Examples
is_convex()
Is the polygon convex?
A polygon is convex if all its interior angles are less than 180 degrees and there are
no intersections between sides.
Returns is_convex : boolean
True if this polygon is convex, False otherwise.
See also:
sympy.geometry.util.convex_hull (page 587)
Examples
perimeter
The perimeter of the polygon.
Returns perimeter : number or Basic instance
See also:
sympy.geometry.line.Segment.length (page 619)
Examples
plot_interval(parameter=’t’)
The plot interval for the default geometric plot of the polygon.
Parameters parameter : str, optional
Default value is ‘t’.
Returns plot_interval : list (plot interval)
[parameter, lower_bound, upper_bound]
Examples
second_moment_of_area(point=None)
Returns the second moment and product moment of area of a two dimensional poly-
gon.
Parameters point : Point, two-tuple of sympifiable objects, or
None(default=None)
point is the point about which second moment of area is to be found.
If “point=None” it will be calculated about the axis passing through
the centroid of the polygon.
Returns I_xx, I_yy, I_xy : number or sympy expression
I_xx, I_yy are second moment of area of a two dimensional polygon.
I_xy is product moment of area of a two dimensional polygon.
References
https://en.wikipedia.org/wiki/Second_moment_of_area
Examples
sides
The directed line segments that form the sides of the polygon.
Returns sides : list of sides
Each side is a directed Segment.
See also:
sympy.geometry.point.Point (page 590), sympy.geometry.line.Segment
(page 617)
Examples
vertices
The vertices of the polygon.
Notes
When iterating over the vertices, it is more efficient to index self rather than to
request the vertices and index them. Only use the vertices when you want to process
all of them at once. This is even more important with RegularPolygons that calculate
each vertex.
Examples
class sympy.geometry.polygon.RegularPolygon
A regular polygon.
Such a polygon has all internal angles equal and all sides the same length.
Parameters center : Point
radius : number or Basic instance
The distance from the center to a vertex
n : int
The number of sides
Raises GeometryError
If the center is not a Point, or the radius is not a number or Basic
instance, or the number of sides, n, is less than three.
See also:
sympy.geometry.point.Point (page 590), Polygon (page 650)
Notes
Examples
Attributes
vertices
center
radius
rotation
apothem
interior_angle
exterior_angle
circumcircle
incircle
angles
angles
Returns a dictionary with keys, the vertices of the Polygon, and values, the interior
angle at each vertex.
Examples
apothem
The inradius of the RegularPolygon.
The apothem/inradius is the radius of the inscribed circle.
Returns apothem : number or instance of Basic
See also:
sympy.geometry.line.Segment.length (page 619), sympy.geometry.ellipse.
Circle.radius (page 649)
Examples
area
Returns the area.
Examples
args
Returns the center point, the radius, the number of sides, and the orientation angle.
Examples
center
The center of the RegularPolygon
This is also the center of the circumscribing circle.
Returns center : Point
See also:
sympy.geometry.point.Point (page 590), sympy.geometry.ellipse.Ellipse.
center (page 637)
Examples
centroid
The center of the RegularPolygon
This is also the center of the circumscribing circle.
Returns center : Point
See also:
sympy.geometry.point.Point (page 590), sympy.geometry.ellipse.Ellipse.
center (page 637)
Examples
circumcenter
Alias for center.
Examples
circumcircle
The circumcircle of the RegularPolygon.
Returns circumcircle : Circle
See also:
circumcenter (page 660), sympy.geometry.ellipse.Circle (page 647)
Examples
circumradius
Alias for radius.
Examples
encloses_point(p)
Return True if p is enclosed by (is inside of) self.
Parameters p : Point
Returns encloses_point : True, False or None
See also:
sympy.geometry.ellipse.Ellipse.encloses_point (page 638)
Notes
Examples
exterior_angle
Measure of the exterior angles.
Returns exterior_angle : number
See also:
sympy.geometry.line.LinearEntity.angle_between (page 603)
Examples
incircle
The incircle of the RegularPolygon.
Returns incircle : Circle
See also:
inradius (page 662), sympy.geometry.ellipse.Circle (page 647)
Examples
inradius
Alias for apothem.
Examples
interior_angle
Measure of the interior angles.
Returns interior_angle : number
See also:
sympy.geometry.line.LinearEntity.angle_between (page 603)
Examples
length
Returns the length of the sides.
The half-length of the side and the apothem form two legs of a right triangle whose
hypotenuse is the radius of the regular polygon.
Examples
radius
Radius of the RegularPolygon
Examples
reflect(line)
Override GeometryEntity.reflect since this is not made of only points.
Examples
rotate(angle, pt=None)
Override GeometryEntity.rotate to first rotate the RegularPolygon about its center.
See also:
rotation (page 663)
spin (page 664) Rotates a RegularPolygon in place
rotation
CCW angle by which the RegularPolygon is rotated
Returns rotation : number or instance of Basic
Examples
spin(angle)
Increment in place the virtual Polygon’s rotation by ccw angle.
See also: rotate method which moves the center.
See also:
rotation (page 663)
rotate (page 663) Creates a copy of the RegularPolygon rotated about a Point
vertices
The vertices of the RegularPolygon.
Returns vertices : list
Each vertex is a Point.
See also:
sympy.geometry.point.Point (page 590)
Examples
class sympy.geometry.polygon.Triangle
A polygon with three vertices and three sides.
Parameters points : sequence of Points
keyword: asa, sas, or sss to specify sides/angles of the triangle
Raises GeometryError
If the number of vertices is not equal to three, or one of the vertices
is not a Point, or a valid keyword is not given.
See also:
sympy.geometry.point.Point (page 590), Polygon (page 650)
Examples
Keywords sss, sas, or asa can be used to give the desired side lengths (in order) and
interior angles (in degrees) that define the triangle:
Attributes
vertices
altitudes
orthocenter
circumcenter
circumradius
circumcircle
inradius
incircle
exradii
medians
medial
nine_point_circle
altitudes
The altitudes of the triangle.
An altitude of a triangle is a segment through a vertex, perpendicular to the opposite
side, with length being the height of the vertex measured from the line containing
the side.
Returns altitudes : dict
The dictionary consists of keys which are vertices and values which
are Segments.
See also:
sympy.geometry.point.Point (page 590), sympy.geometry.line.Segment.length
(page 619)
Examples
bisectors()
The angle bisectors of the triangle.
An angle bisector of a triangle is a straight line through a vertex which cuts the
corresponding angle in half.
Returns bisectors : dict
Each key is a vertex (Point) and each value is the corresponding bi-
sector (Segment).
See also:
sympy.geometry.point.Point (page 590), sympy.geometry.line.Segment
(page 617)
Examples
circumcenter
The circumcenter of the triangle
The circumcenter is the center of the circumcircle.
Returns circumcenter : Point
See also:
sympy.geometry.point.Point (page 590)
Examples
circumcircle
The circle which passes through the three vertices of the triangle.
Returns circumcircle : Circle
See also:
sympy.geometry.ellipse.Circle (page 647)
Examples
circumradius
The radius of the circumcircle of the triangle.
Returns circumradius : number of Basic instance
See also:
sympy.geometry.ellipse.Circle.radius (page 649)
Examples
eulerline
The Euler line of the triangle.
The line which passes through circumcenter, centroid and orthocenter.
Returns eulerline : Line (or Point for equilateral triangles in which case
all
centers coincide)
Examples
exradii
The radius of excircles of a triangle.
An excircle of the triangle is a circle lying outside the triangle, tangent to one of its
sides and tangent to the extensions of the other two.
Returns exradii : dict
See also:
sympy.geometry.polygon.Triangle.inradius (page 669)
References
Examples
The exradius touches the side of the triangle to which it is keyed, e.g. the exradius
touching side 2 is:
incenter
The center of the incircle.
The incircle is the circle which lies inside the triangle and touches all three sides.
Returns incenter : Point
See also:
incircle (page 668), sympy.geometry.point.Point (page 590)
Examples
incircle
The incircle of the triangle.
The incircle is the circle which lies inside the triangle and touches all three sides.
Returns incircle : Circle
See also:
sympy.geometry.ellipse.Circle (page 647)
Examples
inradius
The radius of the incircle.
Returns inradius : number of Basic instance
See also:
incircle (page 668), sympy.geometry.ellipse.Circle.radius (page 649)
Examples
is_equilateral()
Are all the sides the same length?
Returns is_equilateral : boolean
See also:
sympy.geometry.entity.GeometryEntity.is_similar (page 585), RegularPoly-
gon (page 657), is_isosceles (page 670), is_right (page 670), is_scalene
(page 670)
Examples
is_isosceles()
Are two or more of the sides the same length?
Returns is_isosceles : boolean
See also:
is_equilateral (page 669), is_right (page 670), is_scalene (page 670)
Examples
is_right()
Is the triangle right-angled.
Returns is_right : boolean
See also:
sympy.geometry.line.LinearEntity.is_perpendicular (page 607),
is_equilateral (page 669), is_isosceles (page 670), is_scalene (page 670)
Examples
is_scalene()
Are all the sides of the triangle of different lengths?
Returns is_scalene : boolean
See also:
is_equilateral (page 669), is_isosceles (page 670), is_right (page 670)
Examples
is_similar(t2)
Is another triangle similar to this one.
Two triangles are similar if one can be uniformly scaled to the other.
Parameters other: Triangle
Returns is_similar : boolean
See also:
sympy.geometry.entity.GeometryEntity.is_similar (page 585)
Examples
medial
The medial triangle of the triangle.
The triangle which is formed from the midpoints of the three sides.
Returns medial : Triangle
See also:
sympy.geometry.line.Segment.midpoint (page 619)
Examples
medians
The medians of the triangle.
A median of a triangle is a straight line through a vertex and the midpoint of the
opposite side, and divides the triangle into two equal areas.
Returns medians : dict
Each key is a vertex (Point) and each value is the median (Segment)
at that point.
See also:
sympy.geometry.point.Point.midpoint (page 594), sympy.geometry.line.
Segment.midpoint (page 619)
Examples
nine_point_circle
The nine-point circle of the triangle.
Nine-point circle is the circumcircle of the medial triangle, which passes through
the feet of altitudes and the middle points of segments connecting the vertices and
the orthocenter.
Returns nine_point_circle : Circle
See also:
sympy.geometry.line.Segment.midpoint (page 619), sympy.geometry.polygon.
Triangle.medial (page 671), sympy.geometry.polygon.Triangle.orthocenter
(page 672)
Examples
orthocenter
The orthocenter of the triangle.
The orthocenter is the intersection of the altitudes of a triangle. It may lie inside,
outside or on the triangle.
Returns orthocenter : Point
See also:
sympy.geometry.point.Point (page 590)
Examples
vertices
The triangle’s vertices
Returns vertices : tuple
Each element in the tuple is a Point
See also:
sympy.geometry.point.Point (page 590)
Examples
Plane
class sympy.geometry.plane.Plane
A plane is a flat, two-dimensional surface. A plane is the two-dimensional analogue of a
point (zero-dimensions), a line (one-dimension) and a solid (three-dimensions). A plane
can generally be constructed by two types of inputs. They are three non-collinear points
and a point and the plane’s normal vector.
Examples
Attributes
p1
normal_vector
angle_between(o)
Angle between the plane and other geometric entity.
Parameters LinearEntity3D, Plane.
Returns angle : angle in radians
Notes
This method accepts only 3D entities as it’s parameter, but if you want to calculate
the angle between a 2D entity and a plane you should first convert to a 3D entity by
projecting onto a desired plane and then proceed to calculate the angle.
Examples
arbitrary_point(u=None, v=None)
Returns an arbitrary point on the Plane. If given two parameters, the point ranges
over the entire plane. If given 1 or no parameters, returns a point with one parame-
ter which, when varying from 0 to 2*pi, moves the point in a circle of radius 1 about
p1 of the Plane.
Returns Point3D
Examples
While arbitrary values of u and v can move the point anywhere in the plane, the
single-parameter point can be used to construct a ray whose arbitrary point can be
located at angle t and radius r from p.p1:
static are_concurrent(*planes)
Is a sequence of Planes concurrent?
Two or more Planes are concurrent if their intersections are a common line.
Parameters planes: list
Returns Boolean
Examples
distance(o)
Distance between the plane and another geometric entity.
Notes
This method accepts only 3D entities as it’s parameter, but if you want to calculate
the distance between a 2D entity and a plane you should first convert to a 3D entity
by projecting onto a desired plane and then proceed to calculate the distance.
Examples
equals(o)
Returns True if self and o are the same mathematical entities.
Examples
Examples
intersection(o)
The intersection with other geometrical entity.
Examples
is_coplanar(o)
Returns True if o is coplanar with self, else False.
Examples
is_parallel(l)
Is the given geometric entity parallel to the plane?
Parameters LinearEntity3D or Plane
Returns Boolean
Examples
is_perpendicular(l)
is the given geometric entity perpendicualar to the given plane?
Parameters LinearEntity3D or Plane
Returns Boolean
Examples
normal_vector
Normal vector of the given plane.
Examples
p1
The only defining point of the plane. Others can be obtained from the arbitrary_point
method.
See also:
sympy.geometry.point.Point3D (page 598)
Examples
parallel_plane(pt)
Plane parallel to the given plane and passing through the point pt.
Parameters pt: Point3D
Returns Plane
Examples
parameter_value(other, u, v=None)
Return the parameter(s) corresponding to the given point.
Examples
By default, the parameter value returned defines a point that is a distance of 1 from
the Plane’s p1 value and in line with the given point:
Moving the point twice as far from p1 does not change the parameter value:
If the 2-value parameter is desired, supply the two parameter symbols and a replace-
ment dictionary will be returned:
>>> p.parameter_value(on_circle, u, v)
{u: sqrt(10)/10, v: sqrt(10)/30}
>>> p.parameter_value(off_circle, u, v)
{u: sqrt(10)/5, v: sqrt(10)/15}
perpendicular_line(pt)
A line perpendicular to the given plane.
Parameters pt: Point3D
Returns Line3D
Examples
perpendicular_plane(*pts)
Return a perpendicular passing through the given points. If the direction ratio be-
tween the points is the same as the Plane’s normal vector then, to select from the
infinite number of possible planes, a third point will be chosen on the z-axis (or the
y-axis if the normal vector is already parallel to the z-axis). If less than two points
are given they will be supplied as follows: if no point is given then pt1 will be self.p1;
if a second point is not given it will be a point through pt1 on a line parallel to the
z-axis (if the normal is not already the z-axis, otherwise on the line parallel to the
y-axis).
Parameters pts: 0, 1 or 2 Point3D
Returns Plane
Examples
projection(pt)
Project the given point onto the plane along the plane normal.
Parameters Point or Point3D
Returns Point3D
Examples
The projection is along the normal vector direction, not the z axis, so (1, 1) does not
project to (1, 1, 2) on the plane A:
>>> b = Point3D(1, 1)
>>> A.projection(b)
Point3D(5/3, 5/3, 2/3)
>>> _ in A
True
projection_line(line)
Project the given line onto the plane through the normal plane containing the line.
Parameters LinearEntity or LinearEntity3D
Returns Point3D, Line3D, Ray3D or Segment3D
Notes
For the interaction between 2D and 3D lines(segments, rays), you should convert the
line to 3D by using this method. For example for finding the intersection between
a 2D and a 3D line, convert the 2D line to a 3D line by projecting it on a required
plane and then proceed to find the intersection between those lines.
Examples
random_point(seed=None)
Returns a random point on the Plane.
Returns Point3D
Examples
The random point can be moved to lie on the circle of radius 1 centered on p1:
The holonomic (page 680) module is intended to deal with holonomic functions along with
various operations on them like addition, multiplication, composition, integration and dif-
ferentiation. The module also implements various kinds of conversions such as converting
holonomic functions to a different form and the other way around.
5.12.1 Contents
This text aims to explain holonomic functions. We assume you have a basic idea of Differential
equations and Abstract algebra.
Definition
Holonomic function is a very general type of special function that includes lots of simple
known functions as its special cases. In fact the more known hypergeometric function and
Meijer G-function are also a special case of it.
L = p0 + p1 · D + p2 · D2 + ...pr · Dr
Here D is the Differential Operator and L is called the annihilator of the function.
A unique holonomic function can be defined from the annihilator and a set of initial conditons.
For instance:
f (x) = exp(x) : L = D − 1, f (0) = 1
f (x) = sin(x) : L = D2 + 1, f (0) = 0, f ′ (0) = 1
Other fundamental functions such as cos(x), log(x), bessel functions etc. are also holonomic.
The family of holonomic functions is closed under addition, multiplication, integration, com-
position. This means if two functions are given are holonomic, then the function resulting on
applying these operation on them will also be holonomic.
References
https://en.wikipedia.org/wiki/Holonomic_function
The polynomial coefficients will be members of the ring ZZ[x] in the example. The D operator
returned by the function DifferentialOperators() can be used to create annihilators just
like SymPy expressions. We currently use the older implementations of rings in SymPy for
priority mechanism.
Two holonomic functions can be added or multiplied with the result also a holonomic func-
tions.
>>> from sympy.holonomic.holonomic import HolonomicFunction,␣
,→DifferentialOperators
HolonomicFunction.integrate(limits, initcond=False)
Integrates the given holonomic function.
Examples
HolonomicFunction.diff(*args)
Differentiation of the given Holonomic function.
See also:
integrate (page 682)
Examples
Examples
HolonomicFunction.to_sequence(lb=True)
Finds recurrence relation for the coefficients in the series expansion of the function
about x0 , where x0 is the point at which the initial condition is stored.
If the point x0 is ordinary, solution of the form [(R, n0 )] is returned. Where R is the recur-
rence relation and n0 is the smallest n for which the recurrence holds true.
If the point x0 is regular singular, a list of solutions in the format (R, p, n0 ) is returned, i.e.
[(R, p, n0 ), ...]. Each tuple in this vector represents a recurrence relation R associated with
a root of the indicial equation p. Conditions of a different format can also be provided in
this case, see the docstring of HolonomicFunction class.
If it’s not possible to numerically compute a initial condition, it is returned as a symbol
Cj , denoting the coefficient of (x − x0 )j in the power series about x0 .
See also:
HolonomicFunction.series (page 684)
References
[R408], [R409]
Examples
Series expansion
Examples
Numerical evaluation
Examples
>>> r = [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1]
One can also observe that the value obtained using Runge-Kutta 4th order is much more
accurate than Euler’s method.
HolonomicFunction.to_hyper(as_list=False, _recur=None)
Returns a hypergeometric function (or linear combination of them) representing the
given holonomic function.
Returns an answer of the form: a1 · xb1 · hyper() + a2 · xb2 · hyper()...
This is very useful as one can now use hyperexpand to find the symbolic expres-
sions/functions.
See also:
from_hyper (page 687), from_meijerg (page 687)
Examples
HolonomicFunction.to_meijerg()
Returns a linear combination of Meijer G-functions.
See also:
to_hyper (page 686)
Examples
Convert to expressions
HolonomicFunction.to_expr()
Converts a Holonomic Function back to elementary functions.
Examples
Examples
Examples
Examples
Integration
One can perform integrations using holonomic functions by following these steps:
1. Convert the integrand to a holonomic function.
Examples
As you can see in the first example we converted the function to holonomic, integrated the
result and then converted back to symbolic expression.
Limitations
1. Converting to expressions is not always possible. The holonomic function should have a
hypergeometric series at x0.
2. Implementation of converting to holonomic sequence currently doesn’t support Frobenius
method when the solutions need to have log terms. This happens when at least one pair of
the roots of the indicial equation differ by an integer and frobenius method yields linearly
dependent series solutions. Since we use this while converting to expressions, sometimes
to_expr() (page 686) fails.
3. There doesn’t seem to be∫ a way for computing indefinite integrals, so integrate()
x
(page 682) basically computes x0 f (x)dx if no limits are given, where x0 is the point at which
initial conditions for the integrand are stored. Sometimes this gives an additional constant in
the result. For instance:
>>> expr_to_holonomic(sin(x)).integrate(x).to_expr()
-cos(x) + 1
>>> sin(x).integrate(x)
-cos(x)
∫x
The indefinite integral of sin(x) is − cos(x). But the output is − cos(x) + 1 which is 0
sin(x)dx.
Although both are considered correct but − cos(x) is simpler.
The integrals module in SymPy implements methods to calculate definite and indefinite
integrals of expressions.
Principal method in this module is integrate() (page 711)
∫
• integrate(f, x) returns the indefinite integral f dx
∫b
• integrate(f, (x, a, b)) returns the definite integral a
f dx
5.13.1 Examples
SymPy can integrate a vast array of functions. It can integrate polynomial functions:
Rational functions:
>>> integrate(x/(x**2+2*x+1), x)
1
log(x + 1) + -----
x + 1
even a few nonelementary integrals (in particular, some integrals involving the error function)
can be evaluated:
>>> integrate(exp(-x**2)*erf(x), x)
____ 2
\/ pi *erf (x)
--------------
4
SymPy has special support for definite integrals, and integral transforms.
sympy.integrals.transforms.mellin_transform(f, x, s, **hints)
Compute the Mellin transform F (s) of f (x),
∫ ∞
F (s) = xs−1 f (x)dx.
0
For all “sensible” functions, this converges absolutely in a strip a < Re(s) < b.
The Mellin transform is related via change of variables to the Fourier transform, and
also to the (bilateral) Laplace transform.
This function returns (F, (a, b), cond) where F is the Mellin transform of f, (a, b)
is the fundamental strip (as above), and cond are auxiliary convergence conditions.
If the integral cannot be computed in closed form, this function returns an unevaluated
MellinTransform object.
For a description of possible hints, refer to the docstring of sympy.integrals.
transforms.IntegralTransform.doit(). If noconds=False, then only F will be re-
turned (i.e. not cond, and also not the strip (a, b)).
>>> from sympy.integrals.transforms import mellin_transform
>>> from sympy import exp
>>> from sympy.abc import x, s
>>> mellin_transform(exp(-x), x, s)
(gamma(s), (0, oo), True)
See also:
inverse_mellin_transform (page 691), laplace_transform (page 692),
fourier_transform (page 693), hankel_transform (page 695), in-
verse_hankel_transform (page 696)
sympy.integrals.transforms.inverse_mellin_transform(F, s, x, strip, **hints)
Compute the inverse Mellin transform of F (s) over the fundamental strip given by
strip=(a, b).
This can be defined as
∫ c+i∞
1
f (x) = x−s F (s)ds,
2πi c−i∞
for any c in the fundamental strip. Under certain regularity conditions on F and/or f ,
this recovers f from its Mellin transform F (and vice versa), for positive real x.
One of a or b may be passed as None; a suitable c will be inferred.
If the integral cannot be computed in closed form, this function returns an unevaluated
InverseMellinTransform object.
Note that this function will assume x to be positive and real, regardless of the sympy
assumptions!
For a description of possible hints, refer to the docstring of sympy.integrals.
transforms.IntegralTransform.doit().
>>> from sympy.integrals.transforms import inverse_mellin_transform
>>> from sympy import oo, gamma
>>> from sympy.abc import x, s
>>> inverse_mellin_transform(gamma(s), s, x, (0, oo))
exp(-x)
See also:
mellin_transform (page 690), hankel_transform (page 695), in-
verse_hankel_transform (page 696)
sympy.integrals.transforms.laplace_transform(f, t, s, **hints)
Compute the Laplace Transform F (s) of f (t),
∫ ∞
F (s) = e−st f (t)dt.
0
For all “sensible” functions, this converges absolutely in a half plane a < Re(s).
This function returns (F, a, cond) where F is the Laplace transform of f, Re(s) > a is
the half-plane of convergence, and cond are auxiliary convergence conditions.
If the integral cannot be computed in closed form, this function returns an unevaluated
LaplaceTransform object.
For a description of possible hints, refer to the docstring of sympy.integrals.
transforms.IntegralTransform.doit(). If noconds=True, only F will be returned (i.e.
not cond, and also not the plane a).
>>> from sympy.integrals import laplace_transform
>>> from sympy.abc import t, s, a
>>> laplace_transform(t**a, t, s)
(s**(-a)*gamma(a + 1)/s, 0, -re(a) < 1)
See also:
inverse_laplace_transform (page 692), mellin_transform (page 690),
fourier_transform (page 693), hankel_transform (page 695), in-
verse_hankel_transform (page 696)
sympy.integrals.transforms.inverse_laplace_transform(F, s, t, plane=None,
**hints)
Compute the inverse Laplace transform of F (s), defined as
∫ c+i∞
1
f (t) = est F (s)ds,
2πi c−i∞
for c so large that F (s) has no singularites in the half-plane Re(s) > c − ϵ.
The plane can be specified by argument plane, but will be inferred if passed as None.
Under certain regularity conditions, this recovers f (t) from its Laplace Transform F (s),
for non-negative t, and vice versa.
If the integral cannot be computed in closed form, this function returns an unevaluated
InverseLaplaceTransform object.
Note that this function will always assume t to be real, regardless of the sympy assump-
tion on t.
For a description of possible hints, refer to the docstring of sympy.integrals.
transforms.IntegralTransform.doit().
>>> from sympy.integrals.transforms import inverse_laplace_transform
>>> from sympy import exp, Symbol
>>> from sympy.abc import s, t
>>> a = Symbol('a', positive=True)
>>> inverse_laplace_transform(exp(-a*s)/s, s, t)
Heaviside(-a + t)
See also:
laplace_transform (page 692), hankel_transform (page 695), in-
verse_hankel_transform (page 696)
sympy.integrals.transforms.fourier_transform(f, x, k, **hints)
Compute the unitary, ordinary-frequency Fourier transform of f , defined as
∫ ∞
F (k) = f (x)e−2πixk dx.
−∞
If the transform cannot be computed in closed form, this function returns an unevaluated
FourierTransform object.
For other Fourier transform conventions, see the function sympy.integrals.
transforms._fourier_transform().
For a description of possible hints, refer to the docstring of sympy.integrals.
transforms.IntegralTransform.doit(). Note that for this transform, by default no-
conds=True.
>>> from sympy import fourier_transform, exp
>>> from sympy.abc import x, k
>>> fourier_transform(exp(-x**2), x, k)
sqrt(pi)*exp(-pi**2*k**2)
>>> fourier_transform(exp(-x**2), x, k, noconds=False)
(sqrt(pi)*exp(-pi**2*k**2), True)
See also:
inverse_fourier_transform (page 693), sine_transform (page 693), in-
verse_sine_transform (page 694), cosine_transform (page 694), in-
verse_cosine_transform (page 695), hankel_transform (page 695), in-
verse_hankel_transform (page 696), mellin_transform (page 690),
laplace_transform (page 692)
sympy.integrals.transforms.inverse_fourier_transform(F, k, x, **hints)
Compute the unitary, ordinary-frequency inverse Fourier transform of F , defined as
∫ ∞
f (x) = F (k)e2πixk dk.
−∞
If the transform cannot be computed in closed form, this function returns an unevaluated
InverseFourierTransform object.
For other Fourier transform conventions, see the function sympy.integrals.
transforms._fourier_transform().
For a description of possible hints, refer to the docstring of sympy.integrals.
transforms.IntegralTransform.doit(). Note that for this transform, by default no-
conds=True.
>>> from sympy import inverse_fourier_transform, exp, sqrt, pi
>>> from sympy.abc import x, k
>>> inverse_fourier_transform(sqrt(pi)*exp(-(pi*k)**2), k, x)
exp(-x**2)
>>> inverse_fourier_transform(sqrt(pi)*exp(-(pi*k)**2), k, x, noconds=False)
(exp(-x**2), True)
See also:
fourier_transform (page 693), sine_transform (page 693), inverse_sine_transform
(page 694), cosine_transform (page 694), inverse_cosine_transform
(page 695), hankel_transform (page 695), inverse_hankel_transform (page 696),
mellin_transform (page 690), laplace_transform (page 692)
sympy.integrals.transforms.sine_transform(f, x, k, **hints)
Compute the unitary, ordinary-frequency sine transform of f , defined as
√ ∫ ∞
2
F (k) = f (x) sin(2πxk)dx.
π 0
If the transform cannot be computed in closed form, this function returns an unevaluated
SineTransform object.
For a description of possible hints, refer to the docstring of sympy.integrals.
transforms.IntegralTransform.doit(). Note that for this transform, by default no-
conds=True.
See also:
fourier_transform (page 693), inverse_fourier_transform (page 693),
inverse_sine_transform (page 694), cosine_transform (page 694), in-
verse_cosine_transform (page 695), hankel_transform (page 695), in-
verse_hankel_transform (page 696), mellin_transform (page 690),
laplace_transform (page 692)
sympy.integrals.transforms.inverse_sine_transform(F, k, x, **hints)
Compute the unitary, ordinary-frequency inverse sine transform of F , defined as
√ ∫ ∞
2
f (x) = F (k) sin(2πxk)dk.
π 0
If the transform cannot be computed in closed form, this function returns an unevaluated
InverseSineTransform object.
For a description of possible hints, refer to the docstring of sympy.integrals.
transforms.IntegralTransform.doit(). Note that for this transform, by default no-
conds=True.
See also:
fourier_transform (page 693), inverse_fourier_transform (page 693),
sine_transform (page 693), cosine_transform (page 694), in-
verse_cosine_transform (page 695), hankel_transform (page 695), in-
verse_hankel_transform (page 696), mellin_transform (page 690),
laplace_transform (page 692)
sympy.integrals.transforms.cosine_transform(f, x, k, **hints)
Compute the unitary, ordinary-frequency cosine transform of f , defined as
√ ∫ ∞
2
F (k) = f (x) cos(2πxk)dx.
π 0
If the transform cannot be computed in closed form, this function returns an unevaluated
CosineTransform object.
For a description of possible hints, refer to the docstring of sympy.integrals.
transforms.IntegralTransform.doit(). Note that for this transform, by default no-
conds=True.
>>> from sympy import cosine_transform, exp, sqrt, cos
>>> from sympy.abc import x, k, a
>>> cosine_transform(exp(-a*x), x, k)
sqrt(2)*a/(sqrt(pi)*(a**2 + k**2))
>>> cosine_transform(exp(-a*sqrt(x))*cos(a*sqrt(x)), x, k)
a*exp(-a**2/(2*k))/(2*k**(3/2))
See also:
fourier_transform (page 693), inverse_fourier_transform (page 693),
sine_transform (page 693), inverse_sine_transform (page 694), in-
verse_cosine_transform (page 695), hankel_transform (page 695), in-
verse_hankel_transform (page 696), mellin_transform (page 690),
laplace_transform (page 692)
sympy.integrals.transforms.inverse_cosine_transform(F, k, x, **hints)
Compute the unitary, ordinary-frequency inverse cosine transform of F , defined as
√ ∫ ∞
2
f (x) = F (k) cos(2πxk)dk.
π 0
If the transform cannot be computed in closed form, this function returns an unevaluated
InverseCosineTransform object.
For a description of possible hints, refer to the docstring of sympy.integrals.
transforms.IntegralTransform.doit(). Note that for this transform, by default no-
conds=True.
>>> from sympy import inverse_cosine_transform, exp, sqrt, pi
>>> from sympy.abc import x, k, a
>>> inverse_cosine_transform(sqrt(2)*a/(sqrt(pi)*(a**2 + k**2)), k, x)
exp(-a*x)
>>> inverse_cosine_transform(1/sqrt(k), k, x)
1/sqrt(x)
See also:
fourier_transform (page 693), inverse_fourier_transform (page 693),
sine_transform (page 693), inverse_sine_transform (page 694), cosine_transform
(page 694), hankel_transform (page 695), inverse_hankel_transform (page 696),
mellin_transform (page 690), laplace_transform (page 692)
sympy.integrals.transforms.hankel_transform(f, r, k, nu, **hints)
Compute the Hankel transform of f , defined as
∫ ∞
Fν (k) = f (r)Jν (kr)rdr.
0
If the transform cannot be computed in closed form, this function returns an unevaluated
HankelTransform object.
For a description of possible hints, refer to the docstring of sympy.integrals.
transforms.IntegralTransform.doit(). Note that for this transform, by default no-
conds=True.
>>> ht = hankel_transform(exp(-a*r), r, k, 0)
>>> ht
a/(k**3*(a**2/k**2 + 1)**(3/2))
>>> inverse_hankel_transform(ht, k, r, 0)
exp(-a*r)
See also:
fourier_transform (page 693), inverse_fourier_transform (page 693),
sine_transform (page 693), inverse_sine_transform (page 694), cosine_transform
(page 694), inverse_cosine_transform (page 695), inverse_hankel_transform
(page 696), mellin_transform (page 690), laplace_transform (page 692)
sympy.integrals.transforms.inverse_hankel_transform(F, k, r, nu, **hints)
Compute the inverse Hankel transform of F defined as
∫ ∞
f (r) = Fν (k)Jν (kr)kdk.
0
If the transform cannot be computed in closed form, this function returns an unevaluated
InverseHankelTransform object.
For a description of possible hints, refer to the docstring of sympy.integrals.
transforms.IntegralTransform.doit(). Note that for this transform, by default no-
conds=True.
>>> ht = hankel_transform(exp(-a*r), r, k, 0)
>>> ht
a/(k**3*(a**2/k**2 + 1)**(3/2))
>>> inverse_hankel_transform(ht, k, r, 0)
exp(-a*r)
See also:
fourier_transform (page 693), inverse_fourier_transform (page 693),
sine_transform (page 693), inverse_sine_transform (page 694), cosine_transform
(page 694), inverse_cosine_transform (page 695), hankel_transform (page 695),
mellin_transform (page 690), laplace_transform (page 692)
5.13.3 Internals
There is a general method for calculating antiderivatives of elementary functions, called the
Risch algorithm. The Risch algorithm is a decision procedure that can determine whether an
elementary solution exists, and in that case calculate it. It can be extended to handle many
nonelementary functions in addition to the elementary ones.
SymPy currently uses a simplified version of the Risch algorithm, called the Risch-Norman
algorithm. This algorithm is much faster, but may fail to find an antiderivative, although it is
still very powerful. SymPy also uses pattern matching and heuristics to speed up evaluation
of some types of integrals, e.g. polynomials.
For non-elementary definite integrals, SymPy uses so-called Meijer G-functions. Details are
described here:
This text aims do describe in some detail the steps (and subtleties) involved in using Meijer
G-functions for computing definite and indefinite integrals. We shall ignore proofs completely.
Overview
∫ ∫∞
The algorithm to compute f (x)dx or 0
f (x)dx generally consists of three steps:
1. Rewrite the integrand using Meijer G-functions (one or sometimes two).
2. Apply an integration theorem, to get the answer (usually expressed as another G-
function).
3. Expand the result in named special functions.
Step (3) is implemented in the function hyperexpand (q.v.). Steps (1) and (2) are described be-
low. Moreover, G-functions are usually branched. Thus our treatment of branched functions
is described first.
∫∞
Some other integrals (e.g. −∞ ) can also be computed by first recasting them into one of the
above forms. There is a lot of choice involved here, and the algorithm is heuristic at best.
Both Meijer G-Functions and Hypergeometric functions are typically branched (possible
branchpoints being 0, ±1, ∞). This is not very important when e.g. expanding a single hyper-
geometric function into named special functions, since sorting out the branches can be left
to the human user. However this algorithm manipulates and transforms G-functions, and to
do this correctly it needs at least some crude understanding of the branchings involved.
To begin, we consider the set S = {(r, θ) : r > 0, θ ∈ R}. We have a map p : S :→ C − {0}, (r, θ) 7→
reiθ . Decreeing this to be a local biholomorphism gives S both a topology and a complex
structure. This Riemann Surface is usually referred to as the Riemann Surface of the loga-
rithm, for the following reason: We can define maps Exp : C → S, (x + iy) 7→ (exp(x), y) and
Log : S → C, (ex , y) 7→ x + iy. These can both be shown to be holomorphic, and are indeed
mutual inverses.
We also sometimes formally attach a point “zero” (0) to S and denote the resulting object
S0 . Notably there is no complex structure defined near 0. A fundamental system of neigh-
bourhoods is given by {Exp(z) : ℜ(z) < k}, which at least defines a topology. Elements of
S0 shall be called polar numbers. We further define functions Arg : S → R, (r, θ) 7→ θ and
|.| : S0 → R>0 , (r, θ) 7→ r. These have evident meaning and are both continuous everywhere.
Using these maps many operations can be extended from C to S. We define Exp(a) Exp(b) =
Exp(a + b) for a, b ∈ C, also for a ∈ S and b ∈ C we define ab = Exp(b Log(a)). It can be checked
easily that using these definitions, many algebraic properties holding for positive reals (e.g.
(ab)c = ac bc ) which hold in C only for some numbers (because of branch cuts) hold indeed for
all polar numbers.
As one peculiarity it should be mentioned that addition of polar numbers is not usually defined.
However, formal sums of polar numbers √ can be used to express
√ branching behaviour. For
example, consider the functions F (z) = 1 + z and G(a, b) = a + b, where a, b, z are polar
numbers. The general rule is that functions of a single polar variable are defined in such a
way that they are continuous on circles, and agree with the usual definition for positive reals.
Thus if S(z) denotes the standard branch of the square root function on C, we are forced to
define
S(p(z)) : |z| < 1
F (z) = S(p(z)) : −π < Arg(z) + 4πn ≤ π for some n ∈ Z .
−S(p(z)) : else
√ are omitting |z| = 1 here, this does not matter for integration.) Finally we define G(a, b) =
(We
aF (b/a).
Suppose we are given an integrand f (x) and are trying to rewrite it as a single G-function.
To do this, we first split f (x) into the form xs g(x) (where g(x) is supposed to be simpler than
f (x)). This is because multiplicative powers can be absorbed into the G-function later. This
splitting is done by _split_mul(f, x). Then we assemble a tuple of functions that occur in
f (e.g. if f (x) = ex cos x, we would assemble the tuple (cos, exp)). This is done by the function
_mytype(f, x). Next we index a lookup table (created using _create_lookup_table()) with
this tuple. This (hopefully) yields a list of Meijer G-function formulae involving these func-
tions, we then pattern-match all of them. If one fits, we were successful, otherwise not and
we have to try something else.
Suppose now we want to rewrite as a product of two G-functions. To do this, we (try to) find
all inequivalent ways of splitting f (x) into a product f1 (x)f2 (x). We could try these splittings
in any order, but it is often a good idea to minimize (a) the number of powers occurring
in fi (x) and (b) the number of different functions occurring in fi (x). Thus given e.g. f (x) =
sin x ex sin 2x we should try f1 (x) = sin x sin 2x, f2 (x) = ex first. All of this is done by the function
_mul_as_two_parts(f).
Finally, we can try a recursive Mellin transform technique. Since the Meijer G-function is
defined essentially as a certain inverse mellin transform, if we want to write a function f (x)
as a G-function, we can compute its mellin transform F (s). If F (s) is in the right form, the
G-function expression can be read off. This technique generalises many standard rewritings,
e.g. eax ebx = e(a+b)x .
One twist is that some functions don’t have mellin transforms, even though they can be written
as G-functions. This is true for example for f (x) = ex sin x (the function grows too rapidly to
have a mellin transform). However if the function is recognised to be analytic, then we can
try to compute the mellin-transform of f (ax) for a parameter a, and deduce the G-function
expression by analytic continuation. (Checking for analyticity is easy. Since we can only
deal with a certain subset of functions anyway, we only have to filter out those which are not
analyitc.)
The function _rewrite_single does the table lookup and recursive mellin transform.
The functions _rewrite1 and _rewrite2 respectively use above-mentioned helpers and
_rewrite_single to rewrite their argument as respectively one or two G-functions.
If the integrand has been recast into G-functions, evaluating the integral is relatively easy.
We first do some substitutions to reduce e.g. the exponent of the argument of the G-function
to unity (see _rewrite_saxena_1 and _rewrite_saxena, respectively, for one or two G-
functions). Next we go through a list of conditions under which the integral theorem applies.
It can fail for basically two reasons: either the integral does not exist, or the manipulations
in deriving the theorem may not be allowed (for more details, see this [BlogPost]).
Sometimes this can be remedied by reducing the argument of the G-functions involved. For
example it is clear that the G-function representing ez is satisfies G(Exp(2πi)z) = G(z) for all
z ∈ S. The function meijerg.get_period() can be used to discover this, and the function
principal_branch(z, period) in functions/elementary/complexes.py can be used to ex-
ploit the information. This is done transparently by the integration code.
This section intends to display in detail the definite integration theorems used in the code.
The following two formulae go back to Meijer (In fact he proved more general formulae;
indeed in the literature formulae are usually staded in more general form. However it is very
easy to deduce the general formulae from the ones we give here. It seemed best to keep the
theorems as simple as possible, since they are very complicated anyway.):
1.
∫ ∞ ( ) ∏m ∏n
a1 , · · · , a p j=1 Γ(bj + 1) Γ(−aj )
Gm,n ηx dx = ∏q ∏p j=1
0
p,q b1 , · · · , bq η j=m+1 Γ(−bj ) j=n+1 Γ(aj + 1)
2.
∫ ∞ ( ) ( ) ( )
c1 , · · · , c u a1 , · · · , ap m+t,n+s a1 , · · · , an , −d1 , · · · , −dv , an+1 , · · · , ap ω
Gs,t m,n
σx Gp,q ωx dx = Gv+p,u+q
0
u,v d1 , · · · , dv b1 , · · · , b q b1 , · · · , bm , −c1 , · · · , −cu , bm+1 , · · · , bq σ
The more interesting question is under what conditions these formulae are valid. Below we
detail the conditions implemented in SymPy. They are an amalgamation of conditions found
in [Prudnikov1990] and [Luke1969]; please let us know if you find any errors.
We can without loss of generality assume p ≤ q, since the G-functions of indices m, n, p, q and
of indices n, m, q, p can be related easily (see e.g. [Luke1969], section 5.3). We introduce the
following notation:
ξ =m+n−p
p+q
δ =m+n−
2
C3 : −ℜ(bj ) < 1 for j = 1, . . . , m
0 < −ℜ(aj ) for j = 1, . . . , n
C3∗ : −ℜ(bj ) < 1 for j = 1, . . . , q
0 < −ℜ(aj ) for j = 1, . . . , p
q+1−p
C4 : −ℜ(δ) +
>q−p
2
The convergence conditions will be detailed in several “cases”, numbered one to five. For
later use it will be helpful to separate conditions “at infinity” from conditions “at zero”. By
conditions “at infinity” we mean conditions that only depend on the behaviour of the integrand
for large, positive values of x, whereas by conditions “at zero” we mean conditions that only
depend on the behaviour of the integrand on (0, ϵ) for any ϵ > 0. Since all our conditions are
specified in terms of parameters of the G-functions, this distinction is not immediately visible.
They are, however, of very distinct character mathematically; the conditions at infinity being
in particular much harder to control.
In order for the integral theorem to be valid, conditions n “at zero” and “at infinity” both have
to be fulfilled, for some n.
These are the conditions “at infinity”:
1.
where
A=1≤n∧p<q∧1≤m
B = 1 ≤ p ∧ 1 ≤ m ∧ q = p + 1 ∧ ¬(n = 0 ∧ m = p + 1)
⌈ ⌉
δ
C = 1 ≤ n ∧ q = p ∧ | arg(η)| ̸= (δ − 2k)π for k = 0, 1, . . . .
2
2.
n = 0 ∧ p + 1 ≤ m ∧ | arg(η)| < δπ
3.
4.
∑
p
p = q ∧ δ = 0 ∧ arg(η) = 0 ∧ η ̸= 0 ∧ ℜ bj − aj < 0
j=1
5.
η ̸= 0 ∧ C3
2.
C3
3.
C3 ∧ C4
4.
C3
5.
C3
∑
q ∑
p
p−q
µ= bj − aj + +1
j=1 j=1
2
u−v
ϕ=q−p− +1
2
η = 1 − (v − u) − µ − ρ
π(q − m − n) + | arg(ω)|
ψ=
q−p
π(v − s − t) + | arg(σ)|)
θ=
v−u
λc = (q − p)|ω|1/(q−p) cos ψ + (v − u)|σ|1/(v−u) cos θ
C1 : (ai − bj ∈
/ Z>0 for i = 1, . . . , n, j = 1, . . . , m)
∧(ci − dj ∈
/ Z>0 for i = 1, . . . , t, j = 1, . . . , s)
C13 : | arg(ω)| = c∗ π
1
C14 : (z0 ̸= 1 ∧ | arg(1 − z0 )| < π) ∨ (z0 = 1 ∧ ℜ(µ + ρ − u + v) < 1)
2
C14 : (z1 ̸= 1 ∧ | arg(1 − z1 )| < π) ∨ (z1 = 1 ∧ ℜ(µ + ρ − p + q) < 1)
C14 : ϕ = 0 ∧ b∗ + c∗ ≤ 1 ∧ (C14
1
∨ C14
2
)
Note that C16 and C17 are the reason we split the convergence conditions for integral (1).
With this notation established, the implemented convergence conditions can be enumerated
as follows:
1.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.
13.
14.
15.
16.
17.
18.
19.
20.
21.
22.
23.
24.
p < m + n ∧ t = 0 ∧ ϕ = 0 ∧ 0 < s ∧ 0 < b∗ ∧ c∗ < 0 ∧ |arg (ω)| < π (m + n − p + 1) ∧ C1 ∧ C2 ∧ C10 ∧ C14 ∧ C15
25.
q < m + n ∧ s = 0 ∧ ϕ = 0 ∧ 0 < t ∧ 0 < b∗ ∧ c∗ < 0 ∧ |arg (ω)| < π (m + n − q + 1) ∧ C1 ∧ C3 ∧ C10 ∧ C14 ∧ C15
26.
27.
28.
p < q − 1 ∧ t = 0 ∧ ϕ = 0 ∧ 0 < s ∧ 0 < b∗ ∧ 0 ≤ c∗ ∧ πc∗ < |arg (ω)| ∧ |arg (ω)| < π (m + n − p + 1) ∧ C1 ∧ C2 ∧ C10 ∧
29.
q + 1 < p ∧ s = 0 ∧ ϕ = 0 ∧ 0 < t ∧ 0 < b∗ ∧ 0 ≤ c∗ ∧ πc∗ < |arg (ω)| ∧ |arg (ω)| < π (m + n − q + 1) ∧ C1 ∧ C3 ∧ C10 ∧
30.
n = 0 ∧ ϕ = 0 ∧ 0 < s + t ∧ 0 < m ∧ 0 < c∗ ∧ b∗ < 0 ∧ |arg (σ)| < π (s + t − u + 1) ∧ C1 ∧ C2 ∧ C12 ∧ C14 ∧ C15
31.
m = 0 ∧ ϕ = 0 ∧ v < s + t ∧ 0 < n ∧ 0 < c∗ ∧ b∗ < 0 ∧ |arg (σ)| < π (s + t − v + 1) ∧ C1 ∧ C3 ∧ C12 ∧ C14 ∧ C15
32.
n = 0 ∧ ϕ = 0 ∧ u = v − 1 ∧ 0 < m ∧ 0 < c∗ ∧ 0 ≤ b∗ ∧ πb∗ < |arg (σ)| ∧ |arg (σ)| < π (b∗ + 1) ∧ C1 ∧ C2 ∧ C12 ∧ C14 ∧
33.
m = 0 ∧ ϕ = 0 ∧ u = v + 1 ∧ 0 < n ∧ 0 < c∗ ∧ 0 ≤ b∗ ∧ πb∗ < |arg (σ)| ∧ |arg (σ)| < π (b∗ + 1) ∧ C1 ∧ C3 ∧ C12 ∧ C14 ∧
34.
n = 0 ∧ ϕ = 0 ∧ u < v − 1 ∧ 0 < m ∧ 0 < c∗ ∧ 0 ≤ b∗ ∧ πb∗ < |arg (σ)| ∧ |arg (σ)| < π (s + t − u + 1) ∧ C1 ∧ C2 ∧ C12 ∧
35.
m = 0 ∧ ϕ = 0 ∧ v + 1 < u ∧ 0 < n ∧ 0 < c∗ ∧ 0 ≤ b∗ ∧ πb∗ < |arg (σ)| ∧ |arg (σ)| < π (s + t − v + 1) ∧ C1 ∧ C3 ∧ C12 ∧
36.
37.
38.
39.
The inverse laplace transform of a Meijer G-function can be expressed as another G-function.
This is a fairly versatile method for computing this transform. However, I could not find the
details in the literature, so I work them out here. In [Luke1969], section 5.6.3, there is a
formula for the inverse Laplace transform of a G-function of argument bz, and convergence
conditions are also given. However, we need a formula for argument bz a for rational a.
We are asked to compute
∫ c+i∞
1
f (t) = ezt G(bz a )dz,
2πi c−i∞
We shall work formally for now. Denote by ∆(s) the product of gamma functions appearing in
the definition of G, so that
∫
1
G(z) = ∆(s)z s ds.
2πi L
Thus
∫ c+i∞ ∫
1
f (t) = ezt ∆(s)bs z as dsdz.
(2πi)2 c−i∞ L
There are a number of obstacles in this computation. Interchange of integrals is only valid
if all integrals involved are absolutely convergent. In particular the inner integral has to
converge. Also, for our identification of the final integral as a Fox H / Meijer G-function to be
correct, the poles of the newly obtained gamma function must be separated properly.
It is easy to check that the inner integral converges absolutely for ℜ(as) < −1. Thus the contour
L has to run left of the line ℜ(as) = −1. Under this condition, the poles of the newly-introduced
gamma function are separated properly.
It remains to observe that the Meijer G-function is an analytic, unbranched function of its
parameters, and of the coefficient b. Hence so is f (t). Thus the final computation remains
valid as long as the initial integral converges, and if there exists a changed set of parameters
where the computation is valid. If we assume w.l.o.g. that a > 0, then the latter condition is
fulfilled if G converges along contours (2) or (3) of [Luke1969], section 5.2, i.e. either δ >= a2
or p ≥ 1, p ≥ q.
Using [Luke1969], section 5.10, for any given meijer G-function we can find a dominant term
c
of the form z a ebz (although this expression might not be the best possible, because of cancel-
lation).
We must thus investigate
∫ c+iT
c
lim ezt z a ebz dz.
T →∞ c−iT
(This principal value integral is the exact statement used in the Laplace inversion theorem.)
We write z = c + iτ . Then arg(z) → ± π2 , and so ezt ∼ eitτ (where ∼ shall always mean “asymptot-
ically equivalent up to a positive real multiplicative constant”). Also z x+iy ∼ |τ |x eiy log |τ | e±xi 2 .
π
( q√ √ )b a 2 bG
2,2 0, 12 a
−z 2 p + z q p + a = √
2 π
(b )
( )b + 12 − 2b + 1 q
√ a 2 − 2 G2,1
b 1
q √ 2 2 z p
z 2 p + zq p + a 2,2 0, 12 a
√ q = √
z p+a π
( )
( )b − 2b + 12 b
+ 1 q
√ a 2 − 2 G2,1
b 1
√ q 2 2 z p
−z p + zq p + a
2 2,2 0, 12 a
√ q = √
z p+a π
Functions involving |z q p − b|:
( ) ( πa )
−a −a + 1 − a2 + 1 zq p −a
|z q p − b| = 2G1,1 2 sin |b| Γ (−a + 1) , if ℜ(a) < 1
2,2 0 − a2 + 1
2 b 2
Functions involving Chi (z q p):
( 1
)
2, 1 z p
3 2q 2
π 2 G2,0
2,4 1 1 4
0, 0 2, 2
Chi (z q p) = −
2
Functions involving Ci (z q p):
( )
√ 1 z 2q p2
πG2,0
1,3 1 4
0, 0
Ci (z q p) = − 2
2
Functions involving Ei (z q p):
( ) ( ) ( )
1 1 1
q
Ei (z p) = −iπG1,0
1,1 z − G2,0
1,2 z peq iπ
− iπG0,1
1,1 z
0 0, 0 0
q
peiπ
Functions involving ez :
( )
z q peiπ
e = G1,0
0,1
q
z p
0
for computing definite integrals from zero to infinity of moderately complicated combi-
nations of special functions, or indefinite integrals of very simple combinations of special
functions.
The strategy employed by the integration code is as follows:
• If computing a definite integral, and both limits are real, and at least one limit is +-
oo, try the G-function method of definite integration first.
• Try to find an antiderivative, using all available methods, ordered by performance
(that is try fastest method first, slowest last; in particular polynomial integration is
tried first, Meijer G-functions second to last, and heuristic Risch last).
• If still not successful, try G-functions irrespective of the limits.
The option meijerg=True, False, None can be used to, respectively: always use G-
function methods and no others, never use G-function methods, or use all available meth-
ods (in order as described above). It defaults to None.
See also:
Integral (page 714), Integral.doit (page 716)
Examples
>>> integrate(x*y, x)
x**2*y/2
>>> integrate(log(x), x)
x*log(x) - x
>>> integrate(x)
x**2/2
>>> integrate(x*y)
Traceback (most recent call last):
...
ValueError: specify integration variables to integrate x*y
Note that integrate(x) syntax is meant only for convenience in interactive sessions and
should be avoided in library code.
Examples
The class Integral represents an unevaluated integral and has some methods that help in the
integration of an expression.
class sympy.integrals.Integral
Represents unevaluated integral.
is_commutative
Returns whether all the free symbols in the integral are commutative.
as_sum(n=None, method=’midpoint’, evaluate=True)
Approximates a definite integral by a sum.
See also:
Examples
For demonstration purposes, this interval will only be split into 2 regions, bounded
by [3, 5] and [5, 7].
The left-hand rule uses function evaluations at the left of each interval:
The right-hand rule uses function evaluations at the right of each interval:
The trapezoid rule uses function evaluations on both sides of the intervals. This is
equivalent to taking the average of the left and right hand rule results:
The left- or trapezoid method will encounter the discontinuity and return infinity:
The number of intervals can be symbolic. If omitted, a dummy symbol will be used
for it. >>> e = Integral(x**2, (x, 0, 2)) >>> e.as_sum(n, ‘right’).expand() 8/3 + 4/n
+ 4/(3*n**2)
This shows that the midpoint rule is more accurate, as its error term decays as the
square of n: >>> e.as_sum(method=’midpoint’).expand() 8/3 - 2/(3*_n**2)
A symbolic sum is returned with evaluate=False: >>> e.as_sum(n, ‘midpoint’, eval-
uate=False) 2*Sum((2*_k/n - 1/n)**2, (_k, 1, n))/n
Arguments
Examples
free_symbols
This method returns the symbols that will exist when the integral is evaluated. This
is useful if one is trying to determine whether an integral depends on a certain
symbol or not.
See also:
function, limits, variables
Examples
principal_value(**kwargs)
Compute the Cauchy Principal Value of the definite integral of a real function in
the given interval on the real axis. In mathematics, the Cauchy principal value, is
a method for assigning values to certain improper integrals which would otherwise
be undefined.
References
[R410], [R411]
Examples
transform(x, u)
Performs a change of variables from x to u using the relationship given by x and u
which will define the transformations f and F (which are inverses of each other) as
follows:
1) If x is a Symbol (which is a variable of integration) then u will be interpreted as
some function, f(u), with inverse F(u). This, in effect, just makes the substitution
of x with f(x).
2) If u is a Symbol then x will be interpreted as some function, F(x), with inverse
f(u). This is commonly referred to as u-substitution.
Once f and F have been identified, the transformation is made as follows:
∫ b ∫ F (b)
d
xdx → f (x)
a F (a) dx
where F (x) is the inverse of f (x) and the limits and integrand have been corrected
so as to retain the same value after integration.
See also:
Notes
The mappings, F(x) or f(u), must lead to a unique integral. Linear or rational linear
expression, 2 ∗ x, 1/x and sqrt(x), will always work; quadratic expressions like x ∗ ∗2 − 1
are acceptable as long as the resulting integrand does not depend on the sign of the
solutions (see examples).
The integral will be returned unchanged if x is not a variable of integration.
x must be (or contain) only one of of the integration variables. If u has more than
one free symbol then it should be sent as a tuple (u, uvar) where uvar identifies which
variable is replacing the integration variable. XXX can it contain another integration
variable?
Examples
>>> i.transform(x, u)
Integral(u*cos(u**2 - 1), (u, 0, 1))
>>> i.transform(x**2 - 1, u)
Integral(cos(u)/2, (u, -1, 0))
This attempt fails because x = +/-sqrt(u + 1) and the sign does not cancel out of the
integrand:
transform can do a substitution. Here, the previous result is transformed back into
the original expression using “u-substitution”:
>>> ui = _
>>> _.transform(sqrt(u + 1), x) == i
True
If the x does not contain a symbol of integration then the integral will be returned
unchanged. Integral i does not have an integration variable a so no change is made:
>>> i.transform(a, x) == i
True
When u has more than one free symbol the symbol that is replacing x must be iden-
tified by passing u as a tuple:
There are still lots of functions that SymPy does not know how to integrate. For bugs related
to this module, see https://github.com/sympy/sympy/issues?q=label%3AIntegration
SymPy has functions to calculate points and weights for Gaussian quadrature of any order
and any precision:
sympy.integrals.quadrature.gauss_legendre(n, n_digits)
Computes the Gauss-Legendre quadrature [R412] points and weights.
The Gauss-Legendre quadrature approximates the integral:
∫ 1 ∑
n
f (x) dx ≈ wi f (xi )
−1 i=1
The nodes xi of an order n quadrature rule are the roots of Pn and the weights wi are
given by:
2
wi = 2
(1 − x2i ) (Pn′ (xi ))
Parameters n : the order of quadrature
n_digits : number of significant digits of the points and weights to return
Returns (x, w) : the x and w are lists of points and weights as Floats.
The points xi and weights wi are returned as (x, w) tuple of lists.
See also:
gauss_laguerre (page 719), gauss_gen_laguerre (page 721), gauss_hermite
(page 720), gauss_chebyshev_t (page 722), gauss_chebyshev_u (page 723),
gauss_jacobi (page 724), gauss_lobatto
References
[R412], [R413]
Examples
sympy.integrals.quadrature.gauss_laguerre(n, n_digits)
Computes the Gauss-Laguerre quadrature [R414] points and weights.
The Gauss-Laguerre quadrature approximates the integral:
∫ ∞ ∑
n
e−x f (x) dx ≈ wi f (xi )
0 i=1
The nodes xi of an order n quadrature rule are the roots of Ln and the weights wi are
given by:
xi
wi = 2
(n + 1)2 (Ln+1 (xi ))
Parameters n : the order of quadrature
n_digits : number of significant digits of the points and weights to return
Returns (x, w) : the x and w are lists of points and weights as Floats.
The points xi and weights wi are returned as (x, w) tuple of lists.
See also:
gauss_legendre (page 719), gauss_gen_laguerre (page 721), gauss_hermite
(page 720), gauss_chebyshev_t (page 722), gauss_chebyshev_u (page 723),
gauss_jacobi (page 724), gauss_lobatto
References
[R414], [R415]
Examples
sympy.integrals.quadrature.gauss_hermite(n, n_digits)
Computes the Gauss-Hermite quadrature [R416] points and weights.
The Gauss-Hermite quadrature approximates the integral:
∫ ∞ ∑
n
e−x f (x) dx ≈
2
wi f (xi )
−∞ i=1
The nodes xi of an order n quadrature rule are the roots of Hn and the weights wi are
given by:
√
2n−1 n! π
wi = 2
n2 (Hn−1 (xi ))
References
Examples
>>> x, w = gauss_hermite(6, 5)
>>> x
[-2.3506, -1.3358, -0.43608, 0.43608, 1.3358, 2.3506]
>>> w
[0.00453, 0.15707, 0.72463, 0.72463, 0.15707, 0.00453]
Γ(α + n)
wi = α+1
nΓ(n)Lα
n−1 (xi )Ln−1 (xi )
See also:
gauss_legendre (page 719), gauss_laguerre (page 719), gauss_hermite (page 720),
gauss_chebyshev_t (page 722), gauss_chebyshev_u (page 723), gauss_jacobi
(page 724), gauss_lobatto
References
[R419], [R420]
Examples
sympy.integrals.quadrature.gauss_chebyshev_t(n, n_digits)
Computes the Gauss-Chebyshev quadrature [R421] points and weights of the first kind.
The Gauss-Chebyshev quadrature of the first kind approximates the integral:
∫ 1 ∑n
1
√ f (x) dx ≈ wi f (xi )
−1 1 − x2 i=1
The nodes xi of an order n quadrature rule are the roots of Tn and the weights wi are
given by:
π
wi =
n
Parameters n : the order of quadrature
n_digits : number of significant digits of the points and weights to return
Returns (x, w) : the x and w are lists of points and weights as Floats.
The points xi and weights wi are returned as (x, w) tuple of lists.
See also:
gauss_legendre (page 719), gauss_laguerre (page 719), gauss_hermite (page 720),
gauss_gen_laguerre (page 721), gauss_chebyshev_u (page 723), gauss_jacobi
(page 724), gauss_lobatto
References
[R421], [R422]
Examples
>>> x, w = gauss_chebyshev_t(6, 5)
>>> x
[0.96593, 0.70711, 0.25882, -0.25882, -0.70711, -0.96593]
>>> w
[0.5236, 0.5236, 0.5236, 0.5236, 0.5236, 0.5236]
sympy.integrals.quadrature.gauss_chebyshev_u(n, n_digits)
Computes the Gauss-Chebyshev quadrature [R423] points and weights of the second
kind.
The Gauss-Chebyshev quadrature of the second kind approximates the integral:
∫ 1 √ ∑
n
1 − x2 f (x) dx ≈ wi f (xi )
−1 i=1
The nodes xi of an order n quadrature rule are the roots of Un and the weights wi are
given by:
( )
π 2 i
wi = sin π
n+1 n+1
References
[R423], [R424]
Examples
>>> x, w = gauss_chebyshev_u(6, 5)
>>> x
[0.90097, 0.62349, 0.22252, -0.22252, -0.62349, -0.90097]
>>> w
[0.084489, 0.27433, 0.42658, 0.42658, 0.27433, 0.084489]
(α,β)
The nodes xi of an order n quadrature rule are the roots of Pn and the weights wi are
given by:
References
Examples
>>> x, w = gauss_jacobi(6, 1, 1, 5)
>>> x
[-0.87174, -0.5917, -0.2093, 0.2093, 0.5917, 0.87174]
>>> w
[0.050584, 0.22169, 0.39439, 0.39439, 0.22169, 0.050584]
The intpoly module in SymPy implements methods to calculate the integral of a polynomial
over 2/3-Polytopes. Uses evaluation techniques as described in Chin et al. (2015) [1].
The input for 2-Polytope or Polygon uses the already existing Polygon data structure in SymPy.
See sympy.geometry.polygon (page 650) for how to create a polygon.
For the 3-Polytope or Polyhedron, the most economical representation is to specify a list of
vertices and then to provide each constituting face(Polygon) as a list of vertex indices.
For example, consider the unit cube. Here is how it would be represented.
unit_cube = [[(0, 0, 0), (0, 0, 1), (0, 1, 0), (0, 1, 1), (1, 0, 0),(1, 0, 1), (1, 1, 0), (
[3, 7, 6, 2], [1, 5, 7, 3], [5, 4, 6, 7], [0, 4, 5, 1], [2, 0, 1, 3], [2,
6, 4, 0]]
Here, the first sublist is the list of vertices. The other smaller lists such as [3, 7, 6, 2]
represent a 2D face of the polyhedra with vertices having index 3, 7, 6 and 2 in the first
sublist(in that order).
Principal method in this module is polytope_integrate() (page 727)
• polytope_integrate(Polygon((0, 0), (0, 1), (1, 0)), x) returns the integral of
x over the triangle with vertices (0, 0), (0, 1) and (1, 0)
• polytope_integrate(unit_cube, x + y + z) returns the integral of x + y + z over the
unit cube.
5.15.1 References
[1] : Chin, Eric B., Jean B. Lasserre, and N. Sukumar. “Numerical integration of homogeneous
functions on convex and nonconvex polygons and polyhedra.” Computational Mechanics 56.6
(2015): 967-981
PDF link : http://dilbert.engr.ucdavis.edu/~suku/quadrature/cls-integration.pdf
5.15.2 Examples
Single Polynomial:
>>> polytope_integrate(Polygon((0, 0), (0, 1), (1, 0)), [3, x*y + y**2, x**4], max_
,→degree=4)
4 2
{3: 3/2, x : 1/30, x*y + y : 1/8}
>>> polytope_integrate(Polygon((0, 0), (0, 1), (1, 0)), [1.125, x, x**2, 6.89*x**3,␣
,→x*y + y**2, x**4], max_degree=4)
2 3 689 4 2
{1.125: 9/16, x: 1/6, x : 1/12, 6.89*x : ----, x : 1/30, x*y + y : 1/8}
2000
Single Polynomial:
,→[3, 4, 5], [3, 5, 1], [3, 1, 0], [3, 0, 4], [4, 0, 2], [4, 2, 5], [2, 0, 1], [5, 2,␣
,→1]]
>>> polytope_integrate(Polygon((0, 0), (0, 1), (1, 0)), [3, x*y + y**2, x**4], max_
,→degree=4)
4 2
{3: 3/2, x : 1/30, x*y + y : 1/8}
>>> polytope_integrate(Polygon((0, 0), (0, 1), (1, 0)), [1.125, x, x**2, 6.89*x**3,␣
,→x*y + y**2, x**4], max_degree=4)
2 3 689 4 2
{1.125: 9/16, x: 1/6, x : 1/12, 6.89*x : ----, x : 1/30, x*y + y : 1/8}
2000
Examples
class sympy.liealgebras.root_system.RootSystem
Represent the root system of a simple Lie algebra
Every simple Lie algebra has a unique root system. To find the root system, we first con-
sider the Cartan subalgebra of g, which is the maximal abelian subalgebra, and consider
the adjoint action of g on this subalgebra. There is a root system associated with this
action. Now, a root system over a vector space V is a set of finite vectors Φ (called roots),
which satisfy:
1. The roots span V
2. The only scalar multiples of x in Φ are x and -x
3. For every x in Φ, the set Φ is closed under reflection through the hyperplane perpen-
dicular to x.
4. If x and y are roots in Φ, then the projection of y onto the line through x is a half-
integral multiple of x.
Now, there is a subset of Φ, which we will call ∆, such that: 1. ∆ is a basis of V 2. Each
root x in Φ can be written x = Σ k_y y for y in ∆
The elements of ∆ are called the simple roots. Therefore, we see that the simple roots
span the root space of a given simple Lie algebra.
References: https://en.wikipedia.org/wiki/Root_system Lie Algebras and Repre-
sentation Theory - Humphreys
add_as_roots(root1, root2)
Add two roots together if and only if their sum is also a root
It takes as input two vectors which should be roots. It then computes their sum and
checks if it is in the list of all possible roots. If it is, it returns the sum. Otherwise it
returns a string saying that the sum is not a root.
Examples
add_simple_roots(root1, root2)
Add two simple roots together
The function takes as input two integers, root1 and root2. It then uses these integers
as keys in the dictionary of simple roots, and gets the corresponding simple roots,
and then adds them together.
Examples
all_roots()
Generate all the roots of a given root system
The result is a dictionary where the keys are integer numbers. It generates the roots
by getting the dictionary of all positive roots from the bases classes, and then taking
each root, and multiplying it by -1 and adding it to the dictionary. In this way all the
negative roots are generated.
cartan_matrix()
Cartan matrix of Lie algebra associated with this root system
Examples
dynkin_diagram()
Dynkin diagram of the Lie algebra associated with this root system
Examples
root_space()
Return the span of the simple roots
The root space is the vector space spanned by the simple roots, i.e. it is a vector
space with a distinguished basis, the simple roots. This method returns a string that
represents the root space as the span of the simple roots, alpha[1],…., alpha[n].
Examples
simple_roots()
Generate the simple roots of the Lie algebra
The rank of the Lie algebra determines the number of simple roots that it has. This
method obtains the rank of the Lie algebra, and then uses the simple_root method
from the Lie algebra classes to generate all the simple roots.
Examples
class sympy.liealgebras.type_a.TypeA
This class contains the information about the A series of simple Lie algebras. ====
basic_root(i, j)
This is a method just to generate roots with a 1 iin the ith position and a -1 in the jth
position.
basis()
Returns the number of independent generators of A_n
cartan_matrix()
Returns the Cartan matrix for A_n. The Cartan matrix matrix for a Lie algebra is
generated by assigning an ordering to the simple roots, (alpha[1], …., alpha[l]). Then
the ijth entry of the Cartan matrix is (<alpha[i],alpha[j]>).
Examples
dimension()
Dimension of the vector space V underlying the Lie algebra
Examples
highest_root()
Returns the highest weight root for A_n
lie_algebra()
Returns the Lie algebra associated with A_n
positive_roots()
This method generates all the positive roots of A_n. This is half of all of the roots of
A_n; by multiplying all the positive roots by -1 we get the negative roots.
Examples
roots()
Returns the total number of roots for A_n
simple_root(i)
Every lie algebra has a unique root system. Given a root system Q, there is a subset
of the roots such that an element of Q is called a simple root if it cannot be written
as the sum of two elements in Q. If we let D denote the set of simple roots, then it is
clear that every element of Q can be written as a linear combination of elements of
D with all coefficients non-negative.
In A_n the ith simple root is the root which has a 1 in the ith position, a -1 in the
(i+1)th position, and zeroes elsewhere.
This method returns the ith simple root for the A series.
Examples
class sympy.liealgebras.type_b.TypeB
basic_root(i, j)
This is a method just to generate roots with a 1 iin the ith position and a -1 in the jth
position.
basis()
Returns the number of independent generators of B_n
cartan_matrix()
Returns the Cartan matrix for B_n. The Cartan matrix matrix for a Lie algebra is
generated by assigning an ordering to the simple roots, (alpha[1], …., alpha[l]). Then
the ijth entry of the Cartan matrix is (<alpha[i],alpha[j]>).
Examples
dimension()
Dimension of the vector space V underlying the Lie algebra
Examples
lie_algebra()
Returns the Lie algebra associated with B_n
positive_roots()
This method generates all the positive roots of A_n. This is half of all of the roots of
B_n; by multiplying all the positive roots by -1 we get the negative roots.
Examples
roots()
Returns the total number of roots for B_n”
simple_root(i)
Every lie algebra has a unique root system. Given a root system Q, there is a subset
of the roots such that an element of Q is called a simple root if it cannot be written
as the sum of two elements in Q. If we let D denote the set of simple roots, then it is
clear that every element of Q can be written as a linear combination of elements of
D with all coefficients non-negative.
In B_n the first n-1 simple roots are the same as the roots in A_(n-1) (a 1 in the ith
position, a -1 in the (i+1)th position, and zeroes elsewhere). The n-th simple root is
the root with a 1 in the nth position and zeroes elsewhere.
This method returns the ith simple root for the B series.
Examples
class sympy.liealgebras.type_c.TypeC
basic_root(i, j)
Generate roots with 1 in ith position and a -1 in jth position
basis()
Returns the number of independent generators of C_n
cartan_matrix()
The Cartan matrix for C_n
The Cartan matrix matrix for a Lie algebra is generated by assigning an ordering to
the simple roots, (alpha[1], …., alpha[l]). Then the ijth entry of the Cartan matrix is
(<alpha[i],alpha[j]>).
Examples
dimension()
Dimension of the vector space V underlying the Lie algebra
Examples
lie_algebra()
Returns the Lie algebra associated with C_n”
positive_roots()
Generates all the positive roots of A_n
This is half of all of the roots of C_n; by multiplying all the positive roots by -1 we
get the negative roots.
Examples
roots()
Returns the total number of roots for C_n”
simple_root(i)
The ith simple root for the C series
Every lie algebra has a unique root system. Given a root system Q, there is a subset
of the roots such that an element of Q is called a simple root if it cannot be written
as the sum of two elements in Q. If we let D denote the set of simple roots, then it is
clear that every element of Q can be written as a linear combination of elements of
D with all coefficients non-negative.
In C_n, the first n-1 simple roots are the same as the roots in A_(n-1) (a 1 in the ith
position, a -1 in the (i+1)th position, and zeroes elsewhere). The nth simple root is
the root in which there is a 2 in the nth position and zeroes elsewhere.
Examples
class sympy.liealgebras.type_d.TypeD
basic_root(i, j)
This is a method just to generate roots with a 1 iin the ith position and a -1 in the jth
position.
basis()
Returns the number of independent generators of D_n
cartan_matrix()
Returns the Cartan matrix for D_n. The Cartan matrix matrix for a Lie algebra is
generated by assigning an ordering to the simple roots, (alpha[1], …., alpha[l]). Then
the ijth entry of the Cartan matrix is (<alpha[i],alpha[j]>).
Examples
dimension()
Dmension of the vector space V underlying the Lie algebra
Examples
lie_algebra()
Returns the Lie algebra associated with D_n”
positive_roots()
This method generates all the positive roots of A_n. This is half of all of the roots of
D_n by multiplying all the positive roots by -1 we get the negative roots.
Examples
roots()
Returns the total number of roots for D_n”
simple_root(i)
Every lie algebra has a unique root system. Given a root system Q, there is a subset
of the roots such that an element of Q is called a simple root if it cannot be written
as the sum of two elements in Q. If we let D denote the set of simple roots, then it is
clear that every element of Q can be written as a linear combination of elements of
D with all coefficients non-negative.
In D_n, the first n-1 simple roots are the same as the roots in A_(n-1) (a 1 in the ith
position, a -1 in the (i+1)th position, and zeroes elsewhere). The nth simple root is
the root in which there 1s in the nth and (n-1)th positions, and zeroes elsewhere.
This method returns the ith simple root for the D series.
Examples
class sympy.liealgebras.type_e.TypeE
basic_root(i, j)
This is a method just to generate roots with a -1 in the ith position and a 1 in the jth
position.
basis()
Returns the number of independent generators of E_n
cartan_matrix()
Returns the Cartan matrix for G_2 The Cartan matrix matrix for a Lie algebra is
generated by assigning an ordering to the simple roots, (alpha[1], …., alpha[l]). Then
the ijth entry of the Cartan matrix is (<alpha[i],alpha[j]>).
Examples
dimension()
Dimension of the vector space V underlying the Lie algebra
Examples
positive_roots()
This method generates all the positive roots of A_n. This is half of all of the roots of
E_n; by multiplying all the positive roots by -1 we get the negative roots.
Examples
roots()
Returns the total number of roots of E_n
simple_root(i)
Every lie algebra has a unique root system. Given a root system Q, there is a subset
of the roots such that an element of Q is called a simple root if it cannot be written
as the sum of two elements in Q. If we let D denote the set of simple roots, then it is
clear that every element of Q can be written as a linear combination of elements of
D with all coefficients non-negative.
This method returns the ith simple root for E_n.
Examples
class sympy.liealgebras.type_f.TypeF
basic_root(i, j)
Generate roots with 1 in ith position and -1 in jth position
basis()
Returns the number of independent generators of F_4
cartan_matrix()
The Cartan matrix for F_4
The Cartan matrix matrix for a Lie algebra is generated by assigning an ordering to
the simple roots, (alpha[1], …., alpha[l]). Then the ijth entry of the Cartan matrix is
(<alpha[i],alpha[j]>).
Examples
dimension()
Dimension of the vector space V underlying the Lie algebra
Examples
positive_roots()
Generate all the positive roots of A_n
This is half of all of the roots of F_4; by multiplying all the positive roots by -1 we
get the negative roots.
Examples
roots()
Returns the total number of roots for F_4
simple_root(i)
The ith simple root of F_4
Every lie algebra has a unique root system. Given a root system Q, there is a subset
of the roots such that an element of Q is called a simple root if it cannot be written
as the sum of two elements in Q. If we let D denote the set of simple roots, then it is
clear that every element of Q can be written as a linear combination of elements of
D with all coefficients non-negative.
Examples
class sympy.liealgebras.type_g.TypeG
basis()
Returns the number of independent generators of G_2
cartan_matrix()
The Cartan matrix for G_2
The Cartan matrix matrix for a Lie algebra is generated by assigning an ordering to
the simple roots, (alpha[1], …., alpha[l]). Then the ijth entry of the Cartan matrix is
(<alpha[i],alpha[j]>).
Examples
dimension()
Dimension of the vector space V underlying the Lie algebra
Examples
positive_roots()
Generate all the positive roots of A_n
This is half of all of the roots of A_n; by multiplying all the positive roots by -1 we
get the negative roots.
Examples
roots()
Returns the total number of roots of G_2”
simple_root(i)
The ith simple root of G_2
Every lie algebra has a unique root system. Given a root system Q, there is a subset
of the roots such that an element of Q is called a simple root if it cannot be written
as the sum of two elements in Q. If we let D denote the set of simple roots, then it is
clear that every element of Q can be written as a linear combination of elements of
D with all coefficients non-negative.
Examples
class sympy.liealgebras.weyl_group.WeylGroup
For each semisimple Lie group, we have a Weyl group. It is a subgroup of the isometry
group of the root system. Specifically, it’s the subgroup that is generated by reflections
through the hyperplanes orthogonal to the roots. Therefore, Weyl groups are reflection
groups, and so a Weyl group is a finite Coxeter group.
coxeter_diagram()
This method returns the Coxeter diagram corresponding to a Weyl group. The Cox-
eter diagram can be obtained from a Lie algebra’s Dynkin diagram by deleting all
arrows; the Coxeter diagram is the undirected graph. The vertices of the Coxeter
diagram represent the generating reflections of the Weyl group, , s_i. An edge is
drawn between s_i and s_j if the order m(i, j) of s_i*s_j is greater than two. If there
is one edge, the order m(i, j) is 3. If there are two edges, the order m(i, j) is 4, and
if there are three edges, the order m(i, j) is 6.
Examples
delete_doubles(reflections)
This is a helper method for determining the order of an element in the Weyl group of
G2. It takes a Weyl element and if repeated simple reflections in it, it deletes them.
element_order(weylelt)
This method returns the order of a given Weyl group element, which should be spec-
ified by the user in the form of products of the generating reflections, i.e. of the form
r1*r2 etc.
For types A-F, this method current works by taking the matrix form of the specified
element, and then finding what power of the matrix is the identity. It then returns
this power.
Examples
generators()
This method creates the generating reflections of the Weyl group for a given Lie
algebra. For a Lie algebra of rank n, there are n different generating reflections.
This function returns them as a list.
Examples
group_name()
This method returns some general information about the Weyl group for a given Lie
algebra. It returns the name of the group and the elements it acts on, if relevant.
group_order()
This method returns the order of the Weyl group. For types A, B, C, D, and E the
order depends on the rank of the Lie algebra. For types F and G, the order is fixed.
Examples
matrix_form(weylelt)
This method takes input from the user in the form of products of the generating
reflections, and returns the matrix corresponding to the element of the Weyl group.
Since each element of the Weyl group is a reflection of some type, there is a corre-
sponding matrix representation. This method uses the standard representation for
all the generating reflections.
Examples
class sympy.liealgebras.cartan_type.CartanType_generator
Constructor for actually creating things
class sympy.liealgebras.cartan_type.Standard_Cartan
Concrete base class for Cartan types such as A4, etc
rank()
Returns the rank of the Lie algebra
series()
Returns the type of the Lie algebra
sympy.liealgebras.dynkin_diagram.DynkinDiagram(t)
Display the Dynkin diagram of a given Lie algebra
Works by generating the CartanType for the input, t, and then returning the Dynkin
diagram method from the individual classes.
Examples
>>> print(DynkinDiagram(”B4”))
0---0---0=>=0
1 2 3 4
sympy.liealgebras.cartan_matrix.CartanMatrix(ct)
Access the Cartan matrix of a specific Lie algebra
Examples
This method works by returning the Cartan matrix which corresponds to Cartan type t.
5.17.1 Introduction
The logic module for SymPy allows to form and manipulate logic expressions using symbolic
and Boolean values.
You can build Boolean expressions with the standard python operators & (And), | (Or), ~ (Not):
>>> x >> y
Implies(x, y)
>>> x << y
Implies(y, x)
The logic module also includes the following functions to derive boolean expressions from
their truth tables-
sympy.logic.boolalg.SOPform(variables, minterms, dontcares=None)
The SOPform function uses simplified_pairs and a redundant group- eliminating algo-
rithm to convert the list of all input combos that generate ‘1’ (the minterms) into the
smallest Sum of Products form.
The variables must be given as the first argument.
Return a logical Or function (i.e., the “sum of products” or “SOP” form) that gives the
desired outcome. If there are inputs that can be ignored, pass them as a list, too.
The result will be one of the (perhaps many) functions that satisfy the conditions.
References
[R428]
Examples
References
[R429]
Examples
class sympy.logic.boolalg.BooleanTrue
SymPy version of True, a singleton that can be accessed via S.true.
This is the SymPy version of True, for use in the logic module. The primary advantage
of using true instead of True is that shorthand boolean operations like ~ and >> will
work as expected on this class, whereas with True they act bitwise on 1. Functions in
the logic module will return this class when they evaluate to true.
See also:
sympy.logic.boolalg.BooleanFalse (page 745)
Notes
There is liable to be some confusion as to when True should be used and when S.true
should be used in various contexts throughout SymPy. An important thing to remember
is that sympify(True) returns S.true. This means that for the most part, you can just
use True and it will automatically be converted to S.true when necessary, similar to how
you can generally use 1 instead of S.One.
The rule of thumb is:
“If the boolean in question can be replaced by an arbitrary symbolic Boolean, like Or(x,
y) or x > 1, use S.true. Otherwise, use True”
In other words, use S.true only on those contexts where the boolean is being used as a
symbolic representation of truth. For example, if the object ends up in the .args of any
expression, then it must necessarily be S.true instead of True, as elements of .args
must be Basic. On the other hand, == is not a symbolic operation in SymPy, since it
always returns True or False, and does so in terms of structural equality rather than
mathematical, so it should return True. The assumptions system should use True and
False. Aside from not satisfying the above rule of thumb, the assumptions system uses
a three-valued logic (True, False, None), whereas S.true and S.false represent a two-
valued logic. When in doubt, use True.
“S.true == True is True.”
While “S.true is True” is False, “S.true == True” is True, so if there is any doubt
over whether a function or expression will return S.true or True, just use == instead of
is to do the comparison, and it will work in either case. Finally, for boolean flags, it’s
better to just use if x instead of if x is True. To quote PEP 8:
Don’t compare boolean values to True or False using ==.
• Yes: if greeting:
• No: if greeting == True:
• Worse: if greeting is True:
Examples
Python operators give a boolean result for true but a bitwise result for True
Python operators give a boolean result for true but a bitwise result for True
class sympy.logic.boolalg.BooleanFalse
SymPy version of False, a singleton that can be accessed via S.false.
This is the SymPy version of False, for use in the logic module. The primary advantage
of using false instead of False is that shorthand boolean operations like ~ and >> will
work as expected on this class, whereas with False they act bitwise on 0. Functions in
the logic module will return this class when they evaluate to false.
See also:
sympy.logic.boolalg.BooleanTrue (page 743)
Notes
Examples
Python operators give a boolean result for false but a bitwise result for False
class sympy.logic.boolalg.And
Logical AND function.
It evaluates its arguments in order, giving False immediately if any of them are False,
and True if they are all True.
Notes
The & operator is provided as a convenience, but note that its use here is different from
its normal use in Python, which is bitwise and. Hence, And(a, b) and a & b will return
different things if a and b are integers.
Examples
class sympy.logic.boolalg.Or
Logical OR function
It evaluates its arguments in order, giving True immediately if any of them are True, and
False if they are all False.
Notes
The | operator is provided as a convenience, but note that its use here is different from
its normal use in Python, which is bitwise or. Hence, Or(a, b) and a | b will return
different things if a and b are integers.
Examples
class sympy.logic.boolalg.Not
Logical Not function (negation)
Returns True if the statement is False Returns False if the statement is True
Notes
• The ~ operator is provided as a convenience, but note that its use here is different
from its normal use in Python, which is bitwise not. In particular, ~a and Not(a) will
be different if a is an integer. Furthermore, since bools in Python subclass from int,
~True is the same as ~1 which is -2, which has a boolean value of True. To avoid
this issue, use the SymPy boolean types true and false.
Examples
class sympy.logic.boolalg.Xor
Logical XOR (exclusive OR) function.
Returns True if an odd number of the arguments are True and the rest are False.
Returns False if an even number of the arguments are True and the rest are False.
Notes
The ^ operator is provided as a convenience, but note that its use here is different from
its normal use in Python, which is bitwise xor. In particular, a ^ b and Xor(a, b) will
be different if a and b are integers.
>>> Xor(x, y).subs(y, 0)
x
Examples
class sympy.logic.boolalg.Nand
Logical NAND function.
It evaluates its arguments in order, giving True immediately if any of them are False, and
False if they are all True.
Returns True if any of the arguments are False Returns False if all arguments are True
Examples
class sympy.logic.boolalg.Nor
Logical NOR function.
It evaluates its arguments in order, giving False immediately if any of them are True, and
True if they are all False.
Returns False if any argument is True Returns True if all arguments are False
Examples
class sympy.logic.boolalg.Implies
Logical implication.
A implies B is equivalent to !A v B
Accepts two Boolean arguments; A and B. Returns False if A is True and B is False Returns
True otherwise.
Notes
The >> and << operators are provided as a convenience, but note that their use here is
different from their normal use in Python, which is bit shifts. Hence, Implies(a, b) and
a >> b will return different things if a and b are integers. In particular, since Python
considers True and False to be integers, True >> True will be the same as 1 >> 1, i.e.,
0, which has a truth value of False. To avoid this issue, use the SymPy objects true and
false.
Examples
class sympy.logic.boolalg.Equivalent
Equivalence relation.
Equivalent(A, B) is True iff A and B are both True or both False
Returns True if all of the arguments are logically equivalent. Returns False otherwise.
Examples
class sympy.logic.boolalg.ITE
If then else clause.
ITE(A, B, C) evaluates and returns the result of B if A is true else it returns the result of
C. All args must be Booleans.
Examples
The following functions can be used to handle Conjunctive and Disjunctive Normal forms-
sympy.logic.boolalg.to_cnf(expr, simplify=False)
Convert a propositional logical sentence s to conjunctive normal form. That is, of the
form ((A | ~B | …) & (B | C | …) & …) If simplify is True, the expr is evaluated to its
simplest CNF form.
Examples
sympy.logic.boolalg.to_dnf(expr, simplify=False)
Convert a propositional logical sentence s to disjunctive normal form. That is, of the
form ((A & ~B & …) | (B & C & …) | …) If simplify is True, the expr is evaluated to its
simplest DNF form.
Examples
sympy.logic.boolalg.is_cnf(expr)
Test whether or not an expression is in conjunctive normal form.
Examples
sympy.logic.boolalg.is_dnf(expr)
Test whether or not an expression is in disjunctive normal form.
Examples
Examples
>>> S(b)
(z & ~x & ~y) | (~x & ~y & ~z)
>>> simplify_logic(_)
~x & ~y
SymPy’s simplify() function can also be used to simplify logic expressions to their simplest
forms.
sympy.logic.boolalg.bool_map(bool1, bool2)
Return the simplified version of bool1, and the mapping of variables that makes the two
expressions bool1 and bool2 represent the same logical behaviour for some correspon-
dence between the variables of each. If more than one mappings of this sort exist, one
of them is returned. For example, And(x, y) is logically equivalent to And(a, b) for the
mapping {x: a, y:b} or {x: b, y:a}. If no such mapping exists, return False.
Examples
>>> from sympy import SOPform, bool_map, Or, And, Not, Xor
>>> from sympy.abc import w, x, y, z, a, b, c, d
>>> function1 = SOPform([x, z, y],[[1, 0, 1], [0, 0, 1]])
>>> function2 = SOPform([a, b, c],[[1, 0, 1], [1, 0, 0]])
>>> bool_map(function1, function2)
(y & ~z, {y: a, z: b})
The results are not necessarily unique, but they are canonical. Here, (w, z) could be
(a, d) or (d, a):
5.17.5 Inference
The function satisfiable will test that a given Boolean expression is satisfiable, that is, you can
assign values to the variables to make the sentence T rue.
For example, the expression x & ~x is not satisfiable, since there are no values for x that
make this sentence True. On the other hand, (x | y) & (x | ~y) & (~x | y) is satisfiable
with both x and y being True.
As you see, when a sentence is satisfiable, it returns a model that makes that sentence True.
If it is not satisfiable it will return False.
sympy.logic.inference.satisfiable(expr, algorithm=’dpll2’, all_models=False)
Check satisfiability of a propositional sentence. Returns a model when it succeeds. Re-
turns {true: true} for trivially true expressions.
On setting all_models to True, if given expr is satisfiable then returns a generator of
models. However, if expr is unsatisfiable then returns a generator containing the single
element False.
Examples
5.18 Matrices
Creating Matrices
The linear algebra module is designed to be as simple as possible. First, we import and
declare our first Matrix object:
>>> from sympy.interactive.printing import init_printing
>>> init_printing(use_unicode=False, wrap_line=False)
>>> from sympy.matrices import Matrix, eye, zeros, ones, diag, GramSchmidt
>>> M = Matrix([[1,0,0], [0,0,0]]); M
[1 0 0]
[ ]
[0 0 0]
>>> Matrix([M, (0, 0, -1)])
[1 0 0 ]
[ ]
[0 0 0 ]
[ ]
[0 0 -1]
>>> Matrix([[1, 2, 3]])
[1 2 3]
>>> Matrix([1, 2, 3])
[1]
[ ]
[2]
[ ]
[3]
In addition to creating a matrix from a list of appropriately-sized lists and/or matrices, SymPy
also supports more advanced methods of matrix creation including a single list of values and
dimension inputs:
>>> Matrix(2, 3, [1, 2, 3, 4, 5, 6])
[1 2 3]
[ ]
[4 5 6]
More interesting (and useful), is the ability to use a 2-variable function (or lambda) to create
a matrix. Here we create an indicator function which is 1 on the diagonal and then use it to
make the identity matrix:
>>> def f(i,j):
... if i == j:
... return 1
... else:
... return 0
(continues on next page)
Finally let’s use lambda to create a 1-line matrix with 1’s in the even permutation entries:
There are also a couple of special constructors for quick matrix construction: eye is the
identity matrix, zeros and ones for matrices of all zeros and ones, respectively, and diag to
put matrices or elements along the diagonal:
>>> eye(4)
[1 0 0 0]
[ ]
[0 1 0 0]
[ ]
[0 0 1 0]
[ ]
[0 0 0 1]
>>> zeros(2)
[0 0]
[ ]
[0 0]
>>> zeros(2, 5)
[0 0 0 0 0]
[ ]
[0 0 0 0 0]
>>> ones(3)
[1 1 1]
[ ]
[1 1 1]
[ ]
[1 1 1]
>>> ones(1, 3)
[1 1 1]
>>> diag(1, Matrix([[1, 2], [3, 4]]))
[1 0 0]
[ ]
[0 1 2]
[ ]
[0 3 4]
Basic Manipulation
While learning to work with matrices, let’s choose one where the entries are readily identi-
fiable. One useful thing to know is that while matrices are 2-dimensional, the storage is not
and so it is allowable - though one should be careful - to access the entries as if they were a
1-d list.
Now, the more standard entry access is a pair of indices which will always return the value
at the corresponding row and column of the matrix:
>>> M[1, 2]
6
>>> M[0, 0]
1
>>> M[1, 1]
5
Since this is Python we’re also able to slice submatrices; slices always give a matrix in return,
even if the dimension is 1 x 1:
In the second example above notice that the slice 2:2 gives an empty range. Note also (in
keeping with 0-based indexing of Python) the first row/column is 0.
You cannot access rows or columns that are not present unless they are in a slice:
Slicing an empty matrix works as long as you use a slice for the coordinate that has no size:
Slicing gives a copy of what is sliced, so modifications of one object do not affect the other:
>>> M2 = M[:, :]
>>> M2[0, 0] = 100
>>> M[0, 0] == 100
False
Notice that changing M2 didn’t change M. Since we can slice, we can also assign entries:
>>> M = Matrix(([1,2,3,4],[5,6,7,8],[9,10,11,12],[13,14,15,16]))
>>> M
[1 2 3 4 ]
[ ]
[5 6 7 8 ]
[ ]
[9 10 11 12]
[ ]
[13 14 15 16]
>>> M[2,2] = M[0,3] = 0
>>> M
[1 2 3 0 ]
[ ]
[5 6 7 8 ]
[ ]
[9 10 0 12]
[ ]
[13 14 15 16]
>>> M.row_del(0)
>>> M
[4 5 6]
[ ]
[7 8 9]
>>> M.col_del(1)
>>> M
[4 6]
[ ]
[7 9]
>>> v1 = Matrix([1,2,3])
>>> v2 = Matrix([4,5,6])
>>> v3 = v1.cross(v2)
>>> v1.dot(v2)
32
>>> v2.dot(v3)
0
>>> v1.dot(v3)
0
Recall that the row_del() and col_del() operations don’t return a value - they simply change
the matrix object. We can also ‘’glue’’ together matrices of the appropriate size:
>>> M1 = eye(3)
>>> M2 = zeros(3, 4)
>>> M1.row_join(M2)
[1 0 0 0 0 0 0]
[ ]
[0 1 0 0 0 0 0]
[ ]
[0 0 1 0 0 0 0]
>>> M3 = zeros(4, 3)
>>> M1.col_join(M3)
[1 0 0]
[ ]
[0 1 0]
(continues on next page)
Operations on entries
>>> M = eye(3)
>>> 2*M
[2 0 0]
[ ]
[0 2 0]
[ ]
[0 0 2]
>>> 3*M
[3 0 0]
[ ]
[0 3 0]
[ ]
[0 0 3]
but we can also apply functions to our matrix entries using applyfunc(). Here we’ll declare
a function that double any input number. Then we apply it to the 3x3 identity matrix:
One more useful matrix-wide entry application function is the substitution function. Let’s
declare a matrix with symbolic entries then substitute a value. Remember we can substitute
anything - even another symbol!:
Linear algebra
Now that we have the basics out of the way, let’s see what we can do with the actual matrices.
Of course, one of the first things that comes to mind is the determinant:
Another common operation is the inverse: In SymPy, this is computed by Gaussian elimination
by default (for dense matrices) but we can specify it be done by LU decomposition as well:
>>> M2.inv()
[1 0 0]
[ ]
[0 1 0]
[ ]
[0 0 1]
>>> M2.inv(method=”LU”)
[1 0 0]
[ ]
[0 1 0]
[ ]
[0 0 1]
>>> M.inv(method=”LU”)
[-3/14 1/14 1/2 ]
[ ]
[-1/28 5/28 -1/4]
[ ]
[ 3/7 -1/7 0 ]
>>> M * M.inv(method=”LU”)
[1 0 0]
[ ]
[0 1 0]
[ ]
[0 0 1]
>>> A = Matrix([[1,1,1],[1,1,3],[2,3,4]])
>>> Q, R = A.QRdecomposition()
>>> Q
[ ___ ___ ___ ]
[\/ 6 -\/ 3 -\/ 2 ]
[----- ------- -------]
[ 6 3 2 ]
[ ]
[ ___ ___ ___ ]
[\/ 6 -\/ 3 \/ 2 ]
[----- ------- ----- ]
[ 6 3 2 ]
[ ]
[ ___ ___ ]
[\/ 6 \/ 3 ]
[----- ----- 0 ]
[ 3 3 ]
>>> R
[ ___ ]
[ ___ 4*\/ 6 ___]
[\/ 6 ------- 2*\/ 6 ]
[ 3 ]
[ ]
[ ___ ]
[ \/ 3 ]
[ 0 ----- 0 ]
[ 3 ]
[ ]
[ ___ ]
[ 0 0 \/ 2 ]
>>> Q*R
[1 1 1]
[ ]
[1 1 3]
[ ]
[2 3 4]
In addition to the solvers in the solver.py file, we can solve the system Ax=b by passing
the b vector to the matrix A’s LUsolve function. Here we’ll cheat a little choose A and x then
multiply to get b. Then we can solve for x and check that it’s correct:
There’s also a nice Gram-Schmidt orthogonalizer which will take a set of vectors and orthog-
onalize them with respect to another. There is an optional argument which specifies whether
or not the output should also be normalized, it defaults to False. Let’s take some vectors and
orthogonalize them - one normalized and one not:
We can spot-check their orthogonality with dot() and their normality with norm():
>>> out1[0].dot(out1[1])
0
>>> out1[0].dot(out1[2])
0
>>> out1[1].dot(out1[2])
0
>>> out2[0].norm()
1
>>> out2[1].norm()
1
>>> out2[2].norm()
1
So there is quite a bit that can be done with the module including eigenvalues, eigenvectors,
nullspace calculation, cofactor expansion tools, and so on. From here one might want to look
over the matrices.py file for all functionality.
class sympy.matrices.matrices.MatrixBase
Base class for matrix objects.
Attributes
cols
rows
D
Return Dirac conjugate (if self.rows == 4).
See also:
H Hermite conjugation
Examples
If the matrix does not have 4 rows an AttributeError will be raised because this
property is only defined for matrices with 4 rows.
>>> Matrix(eye(2)).D
Traceback (most recent call last):
...
AttributeError: Matrix has no attribute D.
LDLdecomposition()
Returns the LDL Decomposition (L, D) of matrix A, such that L * D * L.H == A This
method eliminates the use of square root. Further this ensures that all the diagonal
entries of L are 1. A must be a Hermitian positive-definite matrix.
See also:
cholesky (page 767), LUdecomposition (page 764), QRdecomposition (page 766)
Examples
LDLsolve(rhs)
Solves Ax = B using LDL decomposition, for a general square and non-singular ma-
trix.
For a non-square matrix with rows > cols, the least squares solution is returned.
See also:
LDLdecomposition (page 763), lower_triangular_solve (page 773), up-
per_triangular_solve (page 778), gauss_jordan_solve (page 770),
cholesky_solve (page 768), diagonal_solve (page 769), LUsolve (page 766),
QRsolve (page 767), pinv_solve (page 774)
Examples
Examples
LUdecompositionFF()
Compute a fraction-free LU decomposition.
Returns 4 matrices P, L, D, U such that PA = L D**-1 U. If the elements of the matrix
belong to some integral domain I, then all elements of L, D and U are guaranteed to
belong to I.
Reference
• W. Zhou & D.J. Jeffrey, “Fraction-free matrix factors: new forms for LU and
QR factors”. Frontiers in Computer Science in China, Vol 2, no. 1, pp. 67-80,
2008.
See also:
LUdecomposition (page 764), LUdecomposition_Simple (page 765), LUsolve
(page 766)
LUdecomposition_Simple(iszerofunc=<function _iszero>, simpfunc=None,
rankcheck=False)
Compute an lu decomposition of m x n matrix A, where P*A = L*U
• L is m x m lower triangular with unit diagonal
• U is m x n upper triangular
• P is an m x m permutation matrix
Returns an m x n matrix lu, and an m element list perm where each element of perm
is a pair of row exchange indices.
The factors L and U are stored in lu as follows: The subdiagonal elements of L are
stored in the subdiagonal elements of lu, that is lu[i, j] = L[i, j] whenever i > j. The
elements on the diagonal of L are all 1, and are not explicitly stored. U is stored
in the upper triangular portion of lu, that is lu[i ,j] = U[i, j] whenever i <= j. The
output matrix can be visualized as:
Matrix([ [u, u, u, u], [l, u, u, u], [l, l, u, u], [l, l, l, u]])
where l represents a subdiagonal entry of the L factor, and u represents an entry
from the upper triangular entry of the U factor.
perm is a list row swap index pairs such that if A is the original matrix, then A =
(L*U).permuteBkwd(perm), and the row permutation matrix P such that P*A = L*U
can be computed by P=eye(A.row).permuteFwd(perm).
The keyword argument rankcheck determines if this function raises a ValueError
when passed a matrix whose rank is strictly less than min(num rows, num cols).
The default behavior is to decompose a rank deficient matrix. Pass rankcheck=True
to raise a ValueError instead. (This mimics the previous behavior of this function).
The keyword arguments iszerofunc and simpfunc are used by the pivot search algo-
rithm. iszerofunc is a callable that returns a boolean indicating if its input is zero, or
None if it cannot make the determination. simpfunc is a callable that simplifies its
input. The default is simpfunc=None, which indicate that the pivot search algorithm
should not attempt to simplify any candidate pivots. If simpfunc fails to simplify its
input, then it must return its input instead of a copy.
When a matrix contains symbolic entries, the pivot search algorithm differs from
the case where every entry can be categorized as zero or nonzero. The algorithm
searches column by column through the submatrix whose top left entry coincides
with the pivot position. If it exists, the pivot is the first entry in the current search
column that iszerofunc guarantees is nonzero. If no such candidate exists, then each
candidate pivot is simplified if simpfunc is not None. The search is repeated, with the
difference that a candidate may be the pivot if iszerofunc() cannot guarantee that
it is nonzero. In the second search the pivot is the first candidate that iszerofunc
can guarantee is nonzero. If no such candidate exists, then the pivot is the first
candidate for which iszerofunc returns None. If no such candidate exists, then the
search is repeated in the next column to the right. The pivot search algorithm differs
from the one in rref (), which relies on _find_reasonable_pivot(). Future versions
of LUdecomposition_simple() may use _find_reasonable_pivot().
See also:
LUdecomposition (page 764), LUdecompositionFF (page 765), LUsolve (page 766)
LUsolve(rhs, iszerofunc=<function _iszero>)
Solve the linear system Ax = rhs for x where A = self.
This is for symbolic matrices, for real or complex ones use mpmath.lu_solve or mp-
math.qr_solve.
See also:
lower_triangular_solve (page 773), upper_triangular_solve (page 778),
gauss_jordan_solve (page 770), cholesky_solve (page 768), diagonal_solve
(page 769), LDLsolve (page 764), QRsolve (page 767), pinv_solve (page 774),
LUdecomposition (page 764)
QRdecomposition()
Return Q, R where A = Q*R, Q is orthogonal and R is upper triangular.
See also:
cholesky (page 767), LDLdecomposition (page 763), LUdecomposition (page 764),
QRsolve (page 767)
Examples
QRsolve(b)
Solve the linear system ‘Ax = b’.
‘self’ is the matrix ‘A’, the method argument is the vector ‘b’. The method returns
the solution vector ‘x’. If ‘b’ is a matrix, the system is solved for each column of ‘b’
and the return value is a matrix of the same shape as ‘b’.
This method is slower (approximately by a factor of 2) but more stable for floating-
point arithmetic than the LUsolve method. However, LUsolve usually uses an exact
arithmetic, so you don’t need to use QRsolve.
This is mainly for educational purposes and symbolic matrices, for real (or complex)
matrices use mpmath.qr_solve.
See also:
lower_triangular_solve (page 773), upper_triangular_solve (page 778),
gauss_jordan_solve (page 770), cholesky_solve (page 768), diagonal_solve
(page 769), LDLsolve (page 764), LUsolve (page 766), pinv_solve (page 774),
QRdecomposition (page 766)
add(b)
Return self + b
cholesky()
Returns the Cholesky decomposition L of a matrix A such that L * L.H = A
A must be a Hermitian positive-definite matrix.
See also:
LDLdecomposition (page 763), LUdecomposition (page 764), QRdecomposition
(page 766)
Examples
cholesky_solve(rhs)
Solves Ax = B using Cholesky decomposition, for a general square non-singular ma-
trix. For a non-square matrix with rows > cols, the least squares solution is returned.
See also:
lower_triangular_solve (page 773), upper_triangular_solve (page 778),
gauss_jordan_solve (page 770), diagonal_solve (page 769), LDLsolve (page 764),
LUsolve (page 766), QRsolve (page 767), pinv_solve (page 774)
condition_number()
Returns the condition number of a matrix.
This is the maximum singular value divided by the minimum singular value
See also:
singular_values
Examples
copy()
Returns the copy of a matrix.
Examples
cross(b)
Return the cross product of self and b relaxing the condition of compatible dimen-
sions: if each has 3 elements, a matrix of the same type and shape as self will be
returned. If b has the same shape as self then common identities for the cross
product (like a × b = −b × a) will hold.
Parameters b : 3x1 or 1x3 Matrix
See also:
dot (page 769), multiply (page 773), multiply_elementwise
diagonal_solve(rhs)
Solves Ax = B efficiently, where A is a diagonal Matrix, with non-zero diagonal en-
tries.
See also:
lower_triangular_solve (page 773), upper_triangular_solve (page 778),
gauss_jordan_solve (page 770), cholesky_solve (page 768), LDLsolve (page 764),
LUsolve (page 766), QRsolve (page 767), pinv_solve (page 774)
Examples
dot(b)
Return the dot product of two vectors of equal length. self must be a Matrix of
size 1 x n or n x 1, and b must be either a matrix of size 1 x n, n x 1, or a list/tuple of
length n. A scalar is returned.
See also:
cross (page 769), multiply (page 773), multiply_elementwise
Examples
dual()
Returns the dual of a matrix, which is:
(1/2) ∗ levicivita(i, j, k, l) ∗ M (k, l) summed over indices k and l
Since the levicivita method is anti_symmetric for any pairwise exchange of indices,
the dual of a symmetric matrix is the zero matrix. Strictly speaking the dual defined
here assumes that the ‘matrix’ M is a contravariant anti_symmetric second rank
tensor, so that the dual is a covariant second rank tensor.
exp()
Return the exponentiation of a square matrix.
gauss_jordan_solve(b, freevar=False)
Solves Ax = b using Gauss Jordan elimination.
There may be zero, one, or infinite solutions. If one solution exists, it will be re-
turned. If infinite solutions exist, it will be returned parametrically. If no solutions
exist, It will throw ValueError.
Parameters b : Matrix
The right hand side of the equation to be solved for. Must have the
same number of rows as matrix A.
freevar : List
If the system is underdetermined (e.g. A has more columns than
rows), infinite solutions are possible, in terms of arbitrary values of
free variables. Then the index of the free variables in the solutions
(column Matrix) will be returned by freevar, if the flag f reevar is set
to T rue.
Returns x : Matrix
The matrix that will satisfy Ax = B. Will have as many rows as matrix
A has columns, and as many columns as matrix B.
params : Matrix
If the system is underdetermined (e.g. A has more columns than
rows), infinite solutions are possible, in terms of arbitrary parame-
ters. These arbitrary parameters are returned as params Matrix.
See also:
lower_triangular_solve (page 773), upper_triangular_solve (page 778),
cholesky_solve (page 768), diagonal_solve (page 769), LDLsolve (page 764), LU-
solve (page 766), QRsolve (page 767), pinv (page 774)
References
[R430]
Examples
inv(method=None, **kwargs)
Return the inverse of a matrix.
CASE 1: If the matrix is a dense matrix.
Return the matrix inverse using the method indicated (default is Gauss elimination).
Parameters method : (‘GE’, ‘LU’, or ‘ADJ’)
Raises ValueError
If the determinant of the matrix is zero.
See also:
inverse_LU (page 772), inverse_GE (page 772), inverse_ADJ (page 772)
Notes
Kwargs
Examples
inverse_ADJ(iszerofunc=<function _iszero>)
Calculates the inverse using the adjugate matrix and a determinant.
See also:
inv (page 771), inverse_LU (page 772), inverse_GE (page 772)
inverse_GE(iszerofunc=<function _iszero>)
Calculates the inverse using Gaussian elimination.
See also:
inv (page 771), inverse_LU (page 772), inverse_ADJ (page 772)
inverse_LU(iszerofunc=<function _iszero>)
Calculates the inverse using LU decomposition.
See also:
inv (page 771), inverse_GE (page 772), inverse_ADJ (page 772)
is_nilpotent()
Checks if a matrix is nilpotent.
A matrix B is nilpotent if for some integer k, B**k is a zero matrix.
Examples
key2bounds(keys)
Converts a key with potentially mixed types of keys (integer and slice) into a tuple
of ranges and raises an error if any index is out of self’s range.
See also:
key2ij (page 773)
key2ij(key)
Converts key into canonical form, converting integers or indexable items into valid
integers for self’s range or returning slices unchanged.
See also:
key2bounds (page 772)
lower_triangular_solve(rhs)
Solves Ax = B, where A is a lower triangular matrix.
See also:
upper_triangular_solve (page 778), gauss_jordan_solve (page 770),
cholesky_solve (page 768), diagonal_solve (page 769), LDLsolve (page 764),
LUsolve (page 766), QRsolve (page 767), pinv_solve (page 774)
multiply(b)
Returns self*b
See also:
dot (page 769), cross (page 769), multiply_elementwise
norm(ord=None)
Return the Norm of a Matrix or Vector. In the simplest case this is the geometric
size of the vector Other norms can be specified by the ord parameter
See also:
normalized (page 774)
Examples
normalized()
Return the normalized version of self.
See also:
norm (page 773)
pinv()
Calculate the Moore-Penrose pseudoinverse of the matrix.
The Moore-Penrose pseudoinverse exists and is unique for any matrix. If the matrix
is invertible, the pseudoinverse is the same as the inverse.
See also:
inv (page 771), pinv_solve (page 774)
References
[R431]
Examples
pinv_solve(B, arbitrary_matrix=None)
Solve Ax = B using the Moore-Penrose pseudoinverse.
There may be zero, one, or infinite solutions. If one solution exists, it will be re-
turned. If infinite solutions exist, one will be returned based on the value of arbi-
trary_matrix. If no solutions exist, the least-squares solution is returned.
Parameters B : Matrix
The right hand side of the equation to be solved for. Must have the
same number of rows as matrix A.
arbitrary_matrix : Matrix
If the system is underdetermined (e.g. A has more columns than
rows), infinite solutions are possible, in terms of an arbitrary matrix.
This parameter may be set to a specific matrix to use for that purpose;
if so, it must be the same shape as x, with as many rows as matrix A
has columns, and as many columns as matrix B. If left as None, an ap-
propriate matrix containing dummy symbols in the form of wn_m will
be used, with n and m being row and column position of each symbol.
Returns x : Matrix
The matrix that will satisfy Ax = B. Will have as many rows as matrix
A has columns, and as many columns as matrix B.
See also:
lower_triangular_solve (page 773), upper_triangular_solve (page 778),
gauss_jordan_solve (page 770), cholesky_solve (page 768), diagonal_solve
(page 769), LDLsolve (page 764), LUsolve (page 766), QRsolve (page 767), pinv
(page 774)
Notes
This may return either exact solutions or least squares solutions. To determine
which, check A * A.pinv() * B == B. It will be True if exact solutions exist, and
False if only a least-squares solution exists. Be aware that the left hand side of that
equation may need to be simplified to correctly compare to the right hand side.
References
[R432]
Examples
print_nonzero(symb=’X’)
Shows location of non-zero entries for fast shape lookup.
Examples
project(v)
Return the projection of self onto the line containing v.
Examples
solve(rhs, method=’GE’)
Return solution to self*soln = rhs using given inversion method.
For a list of possible inversion methods, see the .inv() docstring.
solve_least_squares(rhs, method=’CH’)
Return the least-square fit to the data.
By default the cholesky_solve routine is used (method=’CH’); other methods of ma-
trix inversion can be used. To find out which are available, see the docstring of the
.inv() method.
Examples
If each line of S represent coefficients of Ax + By and x and y are [2, 3] then S*xy is:
But let’s add 1 to the middle value and then solve for the least-squares value of xy:
>>> S*xy - r
Matrix([
[1/3],
[1/3],
[1/3]])
>>> _.norm().n(2)
0.58
Examples
upper_triangular_solve(rhs)
Solves Ax = B, where A is an upper triangular matrix.
See also:
lower_triangular_solve (page 773), gauss_jordan_solve (page 770),
cholesky_solve (page 768), diagonal_solve (page 769), LDLsolve (page 764),
LUsolve (page 766), QRsolve (page 767), pinv_solve (page 774)
vech(diagonal=True, check_symmetry=True)
Return the unique elements of a symmetric Matrix as a one column matrix by stack-
ing the elements in the lower triangle.
Arguments: diagonal – include the diagonal cells of self or not check_symmetry –
checks symmetry of self but not completely reliably
See also:
vec
Examples
class sympy.matrices.matrices.MatrixError
class sympy.matrices.matrices.ShapeError
Wrong matrix shape
class sympy.matrices.matrices.NonSquareMatrixError
sympy.matrices.dense.matrix_multiply_elementwise(A, B)
Return the Hadamard product (elementwise product) of A and B
>>> from sympy.matrices import matrix_multiply_elementwise
>>> from sympy.matrices import Matrix
>>> A = Matrix([[0, 1, 2], [3, 4, 5]])
>>> B = Matrix([[1, 10, 100], [100, 10, 1]])
>>> matrix_multiply_elementwise(A, B)
Matrix([
[ 0, 10, 200],
[300, 40, 5]])
See also:
__mul__
sympy.matrices.dense.zeros(*args, **kwargs)
Returns a matrix of zeros with rows rows and cols columns; if cols is omitted a square
matrix will be returned.
See also:
ones (page 779), eye (page 779), diag (page 780)
sympy.matrices.dense.ones(*args, **kwargs)
Returns a matrix of ones with rows rows and cols columns; if cols is omitted a square
matrix will be returned.
See also:
zeros (page 779), eye (page 779), diag (page 780)
sympy.matrices.dense.eye(*args, **kwargs)
Create square identity matrix n x n
See also:
diag (page 780), zeros (page 779), ones (page 779)
sympy.matrices.dense.diag(*values, **kwargs)
Create a sparse, diagonal matrix from a list of diagonal values.
See also:
eye (page 779)
Notes
Examples
The diagonal elements can be matrices; diagonal filling will continue on the diagonal
from the last element of the matrix:
When diagonal elements are lists, they will be treated as arguments to Matrix:
A given band off the diagonal can be made by padding with a vertical or horizontal “kern-
ing” vector:
The type is mutable by default but can be made immutable by setting the mutable flag
to False:
>>> type(diag(1))
<class 'sympy.matrices.dense.MutableDenseMatrix'>
>>> from sympy.matrices import ImmutableMatrix
>>> type(diag(1, cls=ImmutableMatrix))
<class 'sympy.matrices.immutable.ImmutableDenseMatrix'>
sympy.matrices.dense.jordan_cell(eigenval, n)
Create a Jordan block:
Examples
References
http://en.wikipedia.org/wiki/Hessian_matrix
Examples
sympy.matrices.dense.GramSchmidt(vlist, orthonormal=False)
Apply the Gram-Schmidt process to a set of vectors.
see: http://en.wikipedia.org/wiki/Gram%E2%80%93Schmidt_process
sympy.matrices.dense.wronskian(functions, var, method=’bareiss’)
Compute Wronskian for [] of functions
| f1 f2 ... fn |
| f1' f2' ... fn' |
| . . . . |
W(f1, ..., fn) = | . . . . |
| . . . . |
| (n) (n) (n) |
| D (f1) D (f2) ... D (fn) |
see: http://en.wikipedia.org/wiki/Wronskian
See also:
sympy.matrices.mutable.Matrix.jacobian, hessian (page 781)
sympy.matrices.dense.casoratian(seqs, n, zero=True)
Given linear difference operator L of order ‘k’ and homogeneous equation Ly = 0 we
want to compute kernel of L, which is a set of ‘k’ sequences: a(n), b(n), … z(n).
Solutions of L are linearly independent iff their Casoratian, denoted as C(a, b, …, z), do
not vanish for n = 0.
Casoratian is defined by k x k determinant:
Examples
Examples
If you want multiple symarrays to contain distinct symbols, you must provide unique
prefixes:
>>> a = symarray('', 3)
>>> b = symarray('', 3)
>>> a[0] == b[0]
True
>>> a = symarray('a', 3)
>>> b = symarray('b', 3)
>>> a[0] == b[0]
False
>>> symarray('a', 3)
[a_0 a_1 a_2]
For more than one dimension, the shape must be given as a tuple:
sympy.matrices.dense.rot_axis1(theta)
Returns a rotation matrix for a rotation of theta (in radians) about the 1-axis.
See also:
rot_axis2 (page 786) Returns a rotation matrix for a rotation of theta (in radians)
about the 2-axis
rot_axis3 (page 786) Returns a rotation matrix for a rotation of theta (in radians)
about the 3-axis
Examples
>>> rot_axis1(pi/2)
Matrix([
[1, 0, 0],
[0, 0, 1],
[0, -1, 0]])
sympy.matrices.dense.rot_axis2(theta)
Returns a rotation matrix for a rotation of theta (in radians) about the 2-axis.
See also:
rot_axis1 (page 785) Returns a rotation matrix for a rotation of theta (in radians)
about the 1-axis
rot_axis3 (page 786) Returns a rotation matrix for a rotation of theta (in radians)
about the 3-axis
Examples
>>> rot_axis2(pi/2)
Matrix([
[0, 0, -1],
[0, 1, 0],
[1, 0, 0]])
sympy.matrices.dense.rot_axis3(theta)
Returns a rotation matrix for a rotation of theta (in radians) about the 3-axis.
See also:
rot_axis1 (page 785) Returns a rotation matrix for a rotation of theta (in radians)
about the 1-axis
rot_axis2 (page 786) Returns a rotation matrix for a rotation of theta (in radians)
about the 2-axis
Examples
sympy.matrices.matrices.a2idx(j, n=None)
class sympy.matrices.dense.MutableDenseMatrix
Attributes
cols
rows
col_del(i)
Delete the given column.
See also:
col, row_del (page 789)
Examples
col_op(j, f)
In-place operation on col j using two-arg functor whose args are interpreted as
(self[i, j], i).
See also:
col, row_op (page 790)
Examples
col_swap(i, j)
Swap the two given columns of the matrix in-place.
See also:
col, row_swap (page 790)
Examples
copyin_list(key, value)
Copy in elements from a list.
Parameters key : slice
The section of this matrix to replace.
value : iterable
The iterable to copy values from.
See also:
copyin_matrix (page 789)
Examples
copyin_matrix(key, value)
Copy in values from a matrix into the given bounds.
Parameters key : slice
The section of this matrix to replace.
value : Matrix
The matrix to copy values from.
See also:
copyin_list (page 788)
Examples
fill(value)
Fill the matrix with the scalar value.
See also:
zeros (page 779), ones (page 779)
row_del(i)
Delete the given row.
See also:
row, col_del (page 787)
Examples
row_op(i, f)
In-place operation on row i using two-arg functor whose args are interpreted as
(self[i, j], j).
See also:
row, zip_row_op (page 791), col_op (page 787)
Examples
row_swap(i, j)
Swap the two given rows of the matrix in-place.
See also:
row, col_swap (page 788)
Examples
Examples
class sympy.matrices.immutable.ImmutableDenseMatrix
Create an immutable version of a matrix.
Examples
is_zero
Checks if a matrix is a zero matrix.
A matrix is zero if every element is zero. A matrix need not be square to be consid-
ered zero. The empty matrix is zero by the principle of vacuous truth. For a matrix
that may or may not be zero (e.g. contains a symbol), this will be None
Examples
class sympy.matrices.sparse.SparseMatrix
A sparse matrix (a matrix with a large number of zero elements).
See also:
sympy.matrices.dense.Matrix
Examples
Attributes
cols
rows
CL
Alternate faster representation
LDLdecomposition()
Returns the LDL Decomposition (matrices L and D) of matrix A, such that L * D *
L.T == A. A must be a square, symmetric, positive-definite and non-singular.
This method eliminates the use of square root and ensures that all the diagonal
entries of L are 1.
Examples
RL
Alternate faster representation
applyfunc(f)
Apply a function to each element of the matrix.
Examples
as_immutable()
Returns an Immutable version of this Matrix.
as_mutable()
Returns a mutable version of this matrix.
Examples
cholesky()
Returns the Cholesky decomposition L of a matrix A such that L * L.T = A
A must be a square, symmetric, positive-definite and non-singular matrix
Examples
col_list()
Returns a column-sorted list of non-zero elements of the matrix.
See also:
col_op, row_list (page 795)
Examples
liupc()
Liu’s algorithm, for pre-determination of the Elimination Tree of the given matrix,
used in row-based symbolic Cholesky factorization.
References
Symbolic Sparse Cholesky Factorization using Elimination Trees, Jeroen Van Gron-
delle (1999) http://citeseerx.ist.psu.edu/viewdoc/summary?doi=10.1.1.39.7582
Examples
nnz()
Returns the number of non-zero elements in Matrix.
row_list()
Returns a row-sorted list of non-zero elements of the matrix.
See also:
row_op, col_list (page 794)
Examples
row_structure_symbolic_cholesky()
Symbolic cholesky factorization, for pre-determination of the non-zero structure of
the Cholesky factororization.
References
Symbolic Sparse Cholesky Factorization using Elimination Trees, Jeroen Van Gron-
delle (1999) http://citeseerx.ist.psu.edu/viewdoc/summary?doi=10.1.1.39.7582
Examples
scalar_multiply(scalar)
Scalar element-wise multiplication
solve(rhs, method=’LDL’)
Return solution to self*soln = rhs using given inversion method.
For a list of possible inversion methods, see the .inv() docstring.
solve_least_squares(rhs, method=’LDL’)
Return the least-square fit to the data.
By default the cholesky_solve routine is used (method=’CH’); other methods of ma-
trix inversion can be used. To find out which are available, see the docstring of the
.inv() method.
Examples
If each line of S represent coefficients of Ax + By and x and y are [2, 3] then S*xy is:
But let’s add 1 to the middle value and then solve for the least-squares value of xy:
>>> S*xy - r
Matrix([
[1/3],
[1/3],
[1/3]])
>>> _.norm().n(2)
0.58
class sympy.matrices.sparse.MutableSparseMatrix
Attributes
cols
rows
col_del(k)
Delete the given column of the matrix.
See also:
row_del (page 799)
Examples
col_join(other)
Returns B augmented beneath A (row-wise joining):
[A]
[B]
Examples
Joining along columns is the same as appending rows at the end of the matrix:
col_op(j, f)
In-place operation on col j using two-arg functor whose args are interpreted as
(self[i, j], i) for i in range(self.rows).
Examples
col_swap(i, j)
Swap, in place, columns i and j.
Examples
fill(value)
Fill self with the given value.
Notes
Unless many values are going to be deleted (i.e. set to zero) this will create a matrix
that is slower than a dense matrix in operations.
Examples
row_del(k)
Delete the given row of the matrix.
See also:
col_del (page 797)
Examples
row_join(other)
Returns B appended after A (column-wise augmenting):
[A B]
Examples
Joining at row ends is the same as appending columns at the end of the matrix:
>>> C == A.col_insert(A.cols, B)
True
row_op(i, f)
In-place operation on row i using two-arg functor whose args are interpreted as
(self[i, j], j).
See also:
row, zip_row_op (page 800), col_op (page 798)
Examples
row_swap(i, j)
Swap, in place, columns i and j.
Examples
zip_row_op(i, k, f)
In-place operation on row i using two-arg functor whose args are interpreted as
(self[i, j], self[k, j]).
See also:
row, row_op (page 800), col_op (page 798)
Examples
class sympy.matrices.immutable.ImmutableSparseMatrix
Create an immutable version of a sparse matrix.
Examples
Attributes
cols
rows
The standard Matrix class in SymPy is mutable. This is important for performance reasons
but means that standard matrices cannot interact well with the rest of SymPy. This is because
the Basic object, from which most SymPy classes inherit, is immutable.
The mission of the ImmutableDenseMatrix class, which is aliased as ImmutableMatrix for
short, is to bridge the tension between performance/mutability and safety/immutability. Im-
mutable matrices can do almost everything that normal matrices can do but they inherit from
Basic and can thus interact more naturally with the rest of SymPy. ImmutableMatrix also
inherits from MatrixExpr, allowing it to interact freely with SymPy’s Matrix Expression mod-
ule.
You can turn any Matrix-like object into an ImmutableMatrix by calling the constructor
class sympy.matrices.immutable.ImmutableDenseMatrix
Create an immutable version of a matrix.
Examples
is_zero
Checks if a matrix is a zero matrix.
A matrix is zero if every element is zero. A matrix need not be square to be consid-
ered zero. The empty matrix is zero by the principle of vacuous truth. For a matrix
that may or may not be zero (e.g. contains a symbol), this will be None
Examples
The Matrix expression module allows users to write down statements like
>>> Matrix(X)
Matrix([
[X[0, 0], X[0, 1], X[0, 2]],
[X[1, 0], X[1, 1], X[1, 2]],
[X[2, 0], X[2, 1], X[2, 2]]])
>>> (X*Y)[1, 2]
X[1, 0]*Y[0, 2] + X[1, 1]*Y[1, 2] + X[1, 2]*Y[2, 2]
where X and Y are MatrixSymbol (page 805)’s rather than scalar symbols.
class sympy.matrices.expressions.MatrixExpr
Superclass for Matrix Expressions
MatrixExprs represent abstract matrices, linear transformations represented within a
particular basis.
Examples
Attributes
is_Identity
T
Matrix transposition.
as_coeff_Mul(rational=False)
Efficiently extract the coefficient of a product.
as_explicit()
Returns a dense Matrix with elements represented explicitly
Returns an object of type ImmutableDenseMatrix.
See also:
Examples
as_mutable()
Returns a dense, mutable matrix with elements represented explicitly
See also:
Examples
equals(other)
Test elementwise equality between matrices, potentially of different types
Optional parameter first_index: specify which free index to use as the index start-
ing the expression.
Examples
Transposition is detected:
>>> expr = Sum(A[i, j]*B[k, j]*A[l, k], (j, 0, N-1), (k, 0, N-1))
>>> MatrixExpr.from_index_summation(expr)
A*B.T*A.T
class sympy.matrices.expressions.MatrixSymbol
Symbolic representation of a Matrix object
Creates a SymPy Symbol to represent a Matrix. This matrix has a shape and can be
included in Matrix Expressions
Attributes
is_Identity
class sympy.matrices.expressions.MatAdd
A Sum of Matrix Expressions
MatAdd inherits from and operates like SymPy Add
>>> from sympy import MatAdd, MatrixSymbol
>>> A = MatrixSymbol('A', 5, 5)
>>> B = MatrixSymbol('B', 5, 5)
>>> C = MatrixSymbol('C', 5, 5)
>>> MatAdd(A, B, C)
A + B + C
Attributes
is_Identity
class sympy.matrices.expressions.MatMul
A product of matrix expressions
Examples
Attributes
is_Identity
class sympy.matrices.expressions.MatPow
Attributes
is_Identity
class sympy.matrices.expressions.Inverse
The multiplicative inverse of a matrix expression
This is a symbolic object that simply stores its argument without evaluating it. To actually
compute the inverse, use the .inverse() method of matrices.
Examples
Attributes
is_Identity
class sympy.matrices.expressions.Transpose
The transpose of a matrix expression.
This is a symbolic object that simply stores its argument without evaluating it. To actually
compute the transpose, use the transpose() function, or the .T attribute of matrices.
Examples
Attributes
is_Identity
class sympy.matrices.expressions.Trace
Matrix Trace
Represents the trace of a matrix expression.
class sympy.matrices.expressions.FunctionMatrix
Represents a Matrix using a function (Lambda)
This class is an alternative to SparseMatrix
Attributes
is_Identity
class sympy.matrices.expressions.Identity
The Matrix Identity I - multiplicative identity
class sympy.matrices.expressions.ZeroMatrix
The Matrix Zero 0 - additive identity
Attributes
is_Identity
Block Matrices
Block matrices allow you to construct larger matrices out of smaller sub-blocks. They can
work with MatrixExpr (page 803) or ImmutableMatrix objects.
class sympy.matrices.expressions.blockmatrix.BlockMatrix
A BlockMatrix is a Matrix composed of other smaller, submatrices
The submatrices are stored in a SymPy Matrix object but accessed as part of a Matrix
Expression
>>> print(block_collapse(C*B))
Matrix([[X, Z*Y + Z]])
transpose()
Return transpose of matrix.
Examples
class sympy.matrices.expressions.blockmatrix.BlockDiagMatrix
A BlockDiagMatrix is a BlockMatrix with matrices only along the diagonal
sympy.matrices.expressions.blockmatrix.block_collapse(expr)
Evaluates a block matrix expression
>>> print(block_collapse(C*B))
Matrix([[X, Z*Y + Z]])
Computations with polynomials are at the core of computer algebra and having a fast and ro-
bust polynomials manipulation module is a key for building a powerful symbolic manipulation
system. SymPy has a dedicated module sympy.polys for computing in polynomial algebras
over various coefficient domains.
There is a vast number of methods implemented, ranging from simple tools like polynomial
division, to advanced concepts including Gröbner bases and multivariate factorization over
algebraic number domains.
5.19.1 Contents
Introduction
This tutorial tries to give an overview of the functionality concerning polynomials within
SymPy. All code examples assume:
Basic concepts
Polynomials
Given a family (xi ) of symbols, or other suitable objects, including numbers, expressions de-
rived from them by repeated addition, subtraction and multiplication are called polynomial
expressions in the generators xi .
By the distributive law it is possible to perform multiplications before additions and subtrac-
tions. The products of generators thus obtained are called monomials. They are usually
written in the form xν11 xν22 · · · xνnn where the exponents νi are nonnegative integers. It is often
convenient to write this briefly as xν where x = (x1 , x2 , . . . , xn ) denotes the family of generators
and ν = (ν1 , ν2 , . . . , νn ) is the family of exponents.
When all monomials having the same exponents are combined, the polynomial expression
becomes a sum of products cν xν , called the terms of the polynomial, where the coefficients cν
are integers. If some of the xi are manifest numbers, they are incorporated in the coefficients
and not regarded as generators. Such coefficients are typically rational, real or complex
numbers. Some symbolic numbers, e.g., pi, can be either coefficients or generators.
A polynomial expression that is a sum of terms with different monomials is uniquely deter-
mined by its family of coefficients (cν ). Such an expression is customarily called a polynomial,
though, more properly, that name does stand for the coefficient family once the generators
are given. SymPy implements polynomials by default as dictionaries with monomials as keys
and coefficients as values. Another implementation consists of nested lists of coefficients.
The set of all polynomials with integer coefficients in the generators xi is a ring, i.e., the
sums, differences and products of its elements are again polynomials in the same generators.
This ring is denoted Z[x1 , x2 , . . . , xn ], or Z[(xi )], and called the ring of polynomials in the xi with
integer coefficients.
More generally, the coefficients of a polynomial can be elements of any commutative ring A,
and the corresponding polynomial ring is then denoted A[x1 , x2 , . . . , xn ]. The ring A can also be
a polynomial ring. In SymPy, the coefficient ring is called the domain of the polynomial ring,
and it can be given as a keyword parameter. By default, it is determined by the coefficients
of the polynomial arguments.
Polynomial expressions can be transformed into polynomials by the method as_poly():
If a polynomial expression contains numbers that are not integers, they are regarded as co-
efficients and the coefficient ring is extended accordingly. In particular, division by integers
leads to rational coefficients:
Symbolic numbers are considered generators unless they are explicitly excluded, in which
case they are adjoined to the coefficient ring:
>>> e = (x + 2*pi)*y
>>> e.as_poly()
Poly(x*y + 2*y*pi, x, y, pi, domain='ZZ')
>>> e.as_poly(x, y)
Poly(x*y + 2*pi*y, x, y, domain='ZZ[pi]')
>>> e = (x + 2*pi)*y
>>> e.as_poly(domain=ZZ[pi])
Poly(x*y + 2*pi*y, x, y, domain='ZZ[pi]')
Note that the ring Z[π][x, y] of polynomials in x and y with coefficients in Z[π] is mathematically
equivalent to Z[π, x, y], only their implementations differ.
If an expression contains functions of the generators, other than their positive integer powers,
these are interpreted as new generators:
>>> e = x*sin(y) - y
>>> e.as_poly()
Poly(x*(sin(y)) - y, x, y, sin(y), domain='ZZ')
Since y and sin(y) are algebraically independent they can both appear as generators in a poly-
nomial. However, polynomial expressions must not contain negative powers of generators:
>>> e = x - 1/x
>>> e.as_poly()
Poly(x - (1/x), x, 1/x, domain='ZZ')
It is important to realize that the generators x and 1/x = x−1 are treated as algebraically
independent variables. In particular, their product is not equal to 1. Hence generators in de-
nominators should be avoided even if they raise no error in the current implementation. This
behavior is undesirable and may change in the future.
√ Similar problems emerge with rational
powers of generators. So, for example, x and x = x1/2 are not recognized as algebraically
dependent.
If there are algebraic numbers in an expression, it is possible to adjoin them to the coefficient
ring by setting the keyword extension:
>>> e = x + sqrt(2)
>>> e.as_poly()
Poly(x + (sqrt(2)), x, sqrt(2), domain='ZZ')
>>> e.as_poly(extension=True)
Poly(x + sqrt(2), x, domain='QQ<sqrt(2)>')
√
With the default setting extension=False, both x and 2 are incorrectly √ considered alge-
braically independent variables. With coefficients in the extension field Q( 2) the square root
is treated properly as an algebraic number. Setting extension=True whenever algebraic
numbers are involved is definitely recommended even though it is not forced in the current
implementation.
Divisibility
The fourth rational operation, division, or inverted multiplication, is not generally possible in
rings. If a and b are two elements of a ring A, then there may exist a third element q in A such
that a = bq. In fact, there may exist several such elements.
If also a = bq ′ for some q ′ in A, then b(q − q ′ ) = 0. Hence either b or q − q ′ is zero, or they are
both zero divisors, nonzero elements whose product is zero.
Integral domains
Commutative rings with no zero divisors are called integral domains. Most of the commonly
encountered rings, the ring of integers, fields, and polynomial rings over integral domains
are integral domains.
Assume now that A is an integral domain, and consider the set P of its nonzero elements,
which is closed under multiplication. If a and b are in P , and there exists an element q in P
such that a = bq, then q is unique and called the quotient, a/b, of a by b. Moreover, it is said
that
• a is divisible by b,
• b is a divisor of a,
• a is a multiple of b,
• b is a factor of a.
An element a of P is a divisor of 1 if and only if it is invertible in A, with the inverse a−1 = 1/a.
Such elements are called units. The units of the ring of integers are 1 and −1. The invertible
elements in a polynomial ring over a field are the nonzero constant polynomials.
If two elements of P , a and b, are divisible by each other, then the quotient a/b is invertible with
inverse b/a, or equivalently, b = ua where u is a unit. Such elements are said to be associated
with, or associates of, each other. The associates of an integer n are n and −n. In a polynomial
ring over a field the associates of a polynomial are its constant multiples.
Each element of P is divisible by its associates and the units. An element is irreducible if it
has no other divisors and is not a unit. The irreducible elements in the ring of integers are
the prime numbers p and their opposites −p. In a field, every nonzero element is invertible
and there are no irreducible elements.
Factorial domains
In the ring of integers, each nonzero element can be represented as a product of irreducible
elements and optionally a unit ±1. Moreover, any two such products have the same number of
irreducible factors which are associated with each other in a suitable order. Integral domains
having this property are called factorial, or unique factorization domains. In addition to the
ring of integers, all polynomial rings over a field are factorial, and so are more generally
polynomial rings over any factorial domain. Fields are trivially factorial since there are only
units. The irreducible elements of a factorial domain are usually called primes.
A family of integers has only a finite number of common divisors and the greatest of them is
divisible by all of them. More generally, given a family of nonzero elements (ai ) in an integral
domain, a common divisor d of the elements is called a greatest common divisor, abbreviated
gcd, of the family if it is a multiple of all common divisors. A greatest common divisor, if it
exists, is not unique in general; all of its associates have the same property. It is denoted by
d = gcd(a1 , . . . , an ) if there is no danger of confusion. A least common multiple, or lcm, of a
family (ai ) is defined analogously as a common multiple m that divides all common multiples.
It is denoted by m = lcm(a1 , . . . , an ).
In a factorial domain, greatest common divisors always exists. They can be found, at least in
principle, by factoring each element of a family into a product of prime powers and an optional
unit, and, for each prime, taking the least power that appears in the factorizations. The
product of these prime powers is then a greatest common divisor. A least common multiple
can be obtained from the same factorizations as the product of the greatest powers for each
prime.
Euclidean domains
A practical algorithm for computing a greatest common divisor can be implemented in Eu-
clidean domains. They are integral domains that can be endowed with a function w assigning
a nonnegative integer to each nonzero element of the domain and having the following prop-
erty:
if a and b are nonzero, there are q and r that satisfy the division identity
a = qb + r
such that either r = 0 or w(r) < w(b).
The ring of integers and all univariate polynomial rings over fields are Euclidean domains
with w(a) = |a| resp. w(a) = deg(a).
The division identity for integers is implemented in Python as the built-in function divmod()
that can also be applied to SymPy Integers:
For polynomials the division identity is given in SymPy by the function div():
The division identity can be used to determine the divisibility of elements in a Euclidean
domain. If r = 0 in the division identity, then a is divisible by b. Conversely, if a = cb for some
element c, then (c − q)b = r. It follows that c = q and r = 0 if w has the additional property:
if a and b are nonzero, then w(ab) ≥ w(b).
This is satisfied by the functions given above. (And it is always possible to redefine w(a) by
taking the minimum of the values w(xa) for x ̸= 0.)
The principal application of the division identity is the efficient computation of a greatest com-
mon divisor by means of the Euclidean algorithm. It applies to two elements of a Euclidean
domain. A gcd of several elements can be obtained by iteration.
The function for computing the greatest common divisor of integers in SymPy is currently
igcd():
>>> igcd(2, 4)
2
>>> igcd(5, 10, 15)
5
For univariate polynomials over a field the function has its common name gcd(), and the
returned polynomial is monic:
>>> f = 4*x**2 - 1
>>> g = 8*x**3 + 1
>>> gcd(f, g, domain=QQ)
x + 1/2
Divisibility of polynomials
The ring A = Z[x] of univariate polynomials over the ring of integers is not Euclidean but it is
still factorial. To see this, consider the divisibility in A.
Let f and g be two nonzero polynomials in A. If f is divisible by g in A, then it is also divisible
in the ring B = Q[x] of polynomials with rational coefficients. Since B is Euclidean, this can
be determined by means of the division identity.
Assume, conversely, that f = gh for some polynomial h in B. Then f is divisible by g in A if
and only if the coefficients of h are integers. To find out when this is true it is necessary to
consider the divisibility of the coefficients.
For a polynomial f in A, let c be the greatest common divisor of its coefficients. Then f is
divisible by the constant polynomial c in A, and the quotient f /c = p is a polynomial whose
coefficients are integers that have no common divisor apart from the units. Such polynomials
are called primitive. A polynomial with rational coefficients can also be written as f = cp,
where c is a rational number and p is a primitive polynomial. The constant c is called the
content of f , and p is its primitive part. These components can be found by the method
as_content_primitive():
Let f , f ′ be polynomials with contents c, c′ and primitive parts p, p′ . Then f f ′ = (cc′ )(pp′ ) where
the product pp′ is primitive by Gauss’s lemma. It follows that
the content of a product of polynomials is the product of their contents and the
primitive part of the product is the product of the primitive parts.
Returning to the divisibility in the ring Z[x], assume that f and g are two polynomials with
integer coefficients such that the division identity in Q[x] yields the equality f = gh for some
polynomial h with rational coefficients. Then the content of f is equal to the content of g
multiplied by the content of h. As h has integer coefficients if and only if its content is an
integer, we get the following criterion:
f is divisible by g in the ring Z[x] if and only if
i. f is divisible by g in Q[x], and
ii. the content of f is divisible by the content of g in Z.
If f = cp is irreducible in Z[x], then either c or p must be a unit. If p is not a unit, it must be
irreducible also in Q[x]. For if it is a product of two polynomials, it is also the product of their
primitive parts, and one of them must be a unit. Hence there are two kinds of irreducible
elements in Z[x]:
i. prime numbers of Z, and
ii. primitive polynomials that are irreducible in Q[x].
It follows that each polynomial in Z[x] is a product of irreducible elements. It suffices to factor
its content and primitive part separately. These products are essentially unique; hence Z[x] is
also factorial.
Another important consequence is that a greatest common divisor of two polynomials in Z[x]
can be found efficiently by applying the Euclidean algorithm separately to their contents and
primitive parts in the Euclidean domains Z and Q[x]. This is also implemented in SymPy:
>>> f = 4*x**2 - 1
>>> g = 8*x**3 + 1
>>> gcd(f, g)
2*x + 1
>>> gcd(6*f, 3*g)
6*x + 3
Basic functionality
These functions provide different algorithms dealing with polynomials in the form of SymPy
expression, like symbols, sums etc.
Division
The function div() provides division of polynomials with remainder. That is, for polynomials
f and g, it computes q and r, such that f = g · q + r and deg(r) < q. For polynomials in one
variables with coefficients in a field, say, the rational numbers, q and r are uniquely defined
this way:
>>> f = 5*x**2 + 10*x + 3
>>> g = 2*x + 2
As you can see, q has a non-integer coefficient. If you want to do division only in the ring of
polynomials with integer coefficients, you can specify an additional parameter:
>>> q, r = div(f, g, domain='ZZ')
>>> q
0
>>> r
2
5*x + 10*x + 3
But be warned, that this ring is no longer Euclidean and that the degree of the remainder
doesn’t need to be smaller than that of f. Since 2 doesn’t divide 5, 2x doesn’t divide 5x2 , even
if the degree is smaller. But:
>>> g = 5*x + 1
In the last examples, all of the three variables x, y and z are assumed to be variables of
the polynomials. But if you have some unrelated constant as coefficient, you can specify the
variables explicitly:
>>> a, b, c = symbols('a,b,c')
>>> f = a*x**2 + b*x + c
>>> g = 3*x + 2
>>> q, r = div(f, g, domain='QQ')
>>> q
a*x 2*a b
--- - --- + -
3 9 3
(continues on next page)
>>> r
4*a 2*b
--- - --- + c
9 3
With division, there is also the computation of the greatest common divisor and the least
common multiple.
When the polynomials have integer coefficients, the contents’ gcd is also considered:
But if the polynomials have rational coefficients, then the returned polynomial is monic:
>>> f = 3*x**2/2
>>> g = 9*x/4
>>> gcd(f, g)
x
It also works with multiple variables. In this case, the variables are ordered alphabetically,
be default, which has influence on the leading coefficient:
>>> gcd(f, g)
x + 2*y
The lcm is connected with the gcd and one can be computed using the other:
Square-free factorization
The square-free factorization of a univariate polynomial is the product of all factors (not nec-
essarily irreducible) of degree 1, 2 etc.:
>>> f = 2*x**2 + 5*x**3 + 4*x**4 + x**5
>>> sqf_list(f)
(1, [(x + 2, 1), (x, 2), (x + 1, 2)])
>>> sqf(f)
2 2
x *(x + 1) *(x + 2)
Factorization
This function provides factorization of univariate and multivariate polynomials with rational
coefficients:
>>> factor(x**4/2 + 5*x**3/12 - x**2/3)
2
x *(2*x - 1)*(3*x + 4)
----------------------
12
Groebner bases
Solving Equations
We have (incomplete) methods to find the complex or even symbolic roots of polynomials and
to solve some systems of polynomial equations:
>>> from sympy import roots, solve_poly_system
>>> p = Symbol('p')
>>> q = Symbol('q')
Introduction
In this tutorial we present examples from Wester’s article concerning comparison and critique
of mathematical abilities of several computer algebra systems (see [Wester1999]). All the
examples are related to polynomial and algebraic computations and SymPy specific remarks
were added to all of them.
Examples
All examples in this tutorial are computable, so one can just copy and paste them into a Python
shell and do something useful with them. All computations were done using the following
setup:
>>> var('x,y,z,s,c,n')
(x, y, z, s, c, n)
To achieve the same effect in a more systematic way use primitive() function, which returns
the content and the primitive part of the input polynomial:
Note: The content and the primitive part can be computed only over a ring. To simplify
coefficients of a polynomial over a field use monic().
We can compute the greatest common divisor (GCD) of two polynomials using gcd() function:
>>> gcd(f, g)
1
We see that f and g have no common factors. However, f*h and g*h have an obvious factor
h:
Factorization of large univariate polynomials (of degree 120 in this case) over integers is also
possible:
>>> factor(expand(f*g))
⎛ 60 47 34 8 5 ⎞ ⎛ 60 52 39 25 ␣
,→ 23 10 ⎞
-⎝16⋅x + 21⋅x - 64⋅x + 126⋅x + 46⋅x + 81⎠⋅⎝72⋅x - 83⋅x - 22⋅x - 25⋅x -␣
,→19⋅x + 54⋅x + 81⎠
What can be done in univariate case, can be also done for multivariate polynomials. Consider
the following polynomials f, g and h in Z[x, y, z]:
>>> gcd(f, g)
1
>>> factor(expand(f*g))
7 ⎛ 9 9 3 7 6 5 12 7⎞ ⎛ 22 17 5 8 15 ␣
,→9 2 19 8 ⎞
-2⋅y ⋅z⋅⎝6⋅x ⋅y ⋅z + 10⋅x ⋅z + 17⋅x ⋅y⋅z + 40⋅y ⎠⋅⎝3⋅x + 47⋅x ⋅y ⋅z - 6⋅x ⋅y␣
,→⋅z - 24⋅x⋅y ⋅z - 5⎠
Polynomial manipulation functions provided by sympy.polys are mostly used with integer
exponents. However, it’s perfectly valid to compute with symbolic exponents, e.g.:
To test if two polynomials have a root in common we can use resultant() function. The
theory says that the resultant of two polynomials vanishes if there is a common zero of those
polynomials. For example:
In both cases we obtained the factor x + 1 which tells us that the common root is x = −1.
To remove common factors from the numerator and the denominator of a rational function
the elegant way, use cancel() function. For example:
One can work easily we expressions in both expanded and factored forms. Consider a poly-
nomial f in expanded form. We differentiate it and factor the result back:
>>> g = diff(f, x)
>>> factor(g)
19
20⋅(x + 1)
SymPy can very efficiently decompose polynomials of the form xn ± 1 in terms of cyclotomic
polynomials:
>>> factor(x**15 - 1)
⎛ 2 ⎞ ⎛ 4 3 2 ⎞ ⎛ 8 7 5 4 3 ⎞
(x - 1)⋅⎝x + x + 1⎠⋅⎝x + x + x + x + 1⎠⋅⎝x - x + x - x + x - x + 1⎠
The original Wester‘s example was x100 − 1, but was truncated for readability purpose. Note
that this is not a big struggle for factor() to decompose polynomials of degree 1000 or
greater.
As the result we got a splitting factorization of f with monic factors (this is a general rule
when computing in a field with SymPy). The gaussian keyword is useful for improving code
readability, however the same result can be computed using more general syntax:
We would like to reduce degrees of the numerator and the denominator of a rational function
f/g. To do this we employ cancel() function:
>>> cancel(f/g)
3 2 2
x - 2⋅x + √2⋅x - 3⋅x - 2⋅√2⋅x - 3⋅√2
───────────────────────────────────────
2
x - 2
√
Unfortunately nothing interesting happened. This is because by default SymPy treats 2 as a
generator, obtaining a bivariate
√ polynomial for the numerator. To make cancel() recognize
algebraic properties of 2, one needs to use extension keyword:
With sympy.polys we can obtain factorizations of f over different domains, which includes:
• rationals:
>>> factor(f)
⎛ 2 ⎞ ⎛ 2 ⎞
⎝x - x - 1⎠⋅⎝x + x - 1⎠
• finite fields:
>>> factor(f, modulus=5)
2 2
(x - 2) ⋅(x + 2)
• algebraic numbers:
>>> alg = AlgebraicNumber((sqrt(5) - 1)/2, alias='alpha')
Currently SymPy can factor polynomials into irreducibles over various domains, which can re-
sult in a splitting factorization (into linear factors). However, there is currently no systematic
way to infer a splitting field (algebraic number field) automatically. In future the following
syntax will be implemented:
>>> factor(x**3 + x**2 - 7, split=True)
Traceback (most recent call last):
...
NotImplementedError: 'split' option is not implemented yet
Note this is different from extension=True, because the later only tells how expression pars-
ing should be done, not what should be the domain of computation. One can simulate the
split keyword for several classes of polynomials using solve() function.
>>> expand(_)
11
x + x + 1
obtaining polynomial f. This was done using symmetric polynomial representation over finite
fields The same thing can be done using non-symmetric representation:
As with symmetric representation we can expand the factorization to get the input polynomial
back. This time, however, we need to truncate coefficients of the expanded polynomial modulo
65537:
To compute a reduced Gröbner basis for a set of polynomials use groebner() function. The
function accepts various monomial orderings, e.g.: lex, grlex and grevlex, or a user de-
fined one, via order keyword. The lex ordering is the most interesting because it has elim-
ination property, which means that if the system of polynomial equations to groebner() is
zero-dimensional (has finite number of solutions) the last element of the basis is a univariate
polynomial. Consider the following example:
The result is an ordinary Python list, so we can easily apply a function to all its elements, for
example we can factor those elements:
From the above we can easily find all solutions of the system of polynomial equations. Or we
can use solve() to achieve this in a more systematic way:
>>> apart(f)
3 2 2
───── - ───── + ────────
x + 2 x + 1 2
(x + 1)
To return from partial fractions to the rational function use a composition of together() and
cancel():
>>> cancel(together(_))
2
x + 2⋅x + 3
───────────────────
3 2
x + 4⋅x + 5⋅x + 2
Literature
Examples
Examples
Examples
Examples
Examples
Examples
Examples
Examples
Examples
Examples
Examples
Examples
Examples
Examples
>>> exquo(x**2 - 1, x - 1)
x + 1
Examples
Examples
Examples
>>> invert(x**2 - 1, x - 1)
Traceback (most recent call last):
...
NotInvertible: zero divisor
>>> mod_inverse(3, 5)
2
>>> (S(2)/5).invert(S(7)/3)
5/2
Examples
Examples
Examples
For two polynomials f (x) and g(x) with deg f > 0 and deg g > 0 the dispersion dis(f, g) is
defined as:
dis(f, g) := max{J(f, g) ∪ {0}}
= max{{a ∈ N| gcd(f (x), g(x + a)) ̸= 1} ∪ {0}}
and for a single polynomial dis(f ) := dis(f, f ). Note that we make the definition max{} :=
−∞.
See also:
dispersionset (page 835)
References
1. [ManWright94]
2. [Koepf98]
3. [Abramov71]
4. [Man93]
Examples
We can even perform the computations for polynomials having symbolic coefficients:
References
1. [ManWright94]
2. [Koepf98]
3. [Abramov71]
4. [Man93]
Examples
We can even perform the computations for polynomials having symbolic coefficients:
Examples
The default action of polys routines is to expand the expression given to them. terms_gcd
follows this behavior:
>>> terms_gcd((3+3*x)*(x+x*y))
3*x*(x*y + x + y + 1)
If this is not desired then the hint expand can be set to False. In this case the expression
will be treated as though it were comprised of one or more terms:
In order to traverse factors of a Mul or the arguments of other functions, the deep hint
can be used:
Only the y-term had a coefficient that was a fraction; if one does not want to factor out
the 1/2 in cases like this, the flag clear can be set to False:
Examples
Examples
Examples
Examples
Examples
Examples
Examples
Examples
Examples
>>> eq = (2 + 2*x)*x + 2
>>> primitive(eq)
(2, x**2 + x + 1)
Set expand to False to shut this off. Note that the extraction will not be recursive; use
the as_content_primitive method for recursive, non-destructive Rational extraction.
>>> eq.as_content_primitive()
(2, x*(x + 1) + 1)
Examples
>>> compose(x**2 + x, x - 1)
x**2 - x
Examples
Examples
Examples
>>> gff_list(f)
[(Poly(x, x, domain='ZZ'), 1), (Poly(x + 2, x, domain='ZZ'), 4)]
>>> gff_list(f)
[(Poly(x**3 + 7, x, domain='ZZ'), 2), (Poly(x**2 + 5*x, x, domain='ZZ'), 3)]
Examples
Examples
Examples
Examples
Examples
Examples
>>> factor(x**2 + 1)
x**2 + 1
>>> factor(x**2 + 1, modulus=2)
(x + 1)**2
>>> factor(x**2 + 1, gaussian=True)
(x - I)*(x + I)
Examples
>>> intervals(x**2 - 3)
[((-2, -1), 1), ((1, 2), 1)]
>>> intervals(x**2 - 3, eps=1e-2)
[((-26/15, -19/11), 1), ((19/11, 26/15), 1)]
Examples
Examples
sympy.polys.polytools.real_roots(f, multiple=True)
Return a list of real roots with multiplicities of f.
Examples
Examples
Examples
Examples
>>> g
(x**2 - x + 1)**2
Examples
Examples
Use the order argument to set the monomial ordering that will be used to compute the
basis. Allowed orders are lex, grlex and grevlex. If no order is specified, it defaults to
lex.
For more information on Groebner bases, see the references and the docstring of
solvep olys ystem().
References
1. [Buchberger01]
2. [Cox97]
[Buchberger01], [Cox97]
Examples
References
David A. Cox, John B. Little, Donal O’Shea. Ideals, Varieties and Algorithms, 3rd edition,
p. 230
class sympy.polys.polytools.Poly
Generic class for representing and operating on polynomial expressions. Subclasses
Expr class.
See also:
sympy.core.expr.Expr (page 115)
Examples
EC(order=None)
Returns the last non-zero coefficient of f.
Examples
EM(order=None)
Returns the last non-zero monomial of f.
Examples
ET(order=None)
Returns the last non-zero term of f.
Examples
LC(order=None)
Returns the leading coefficient of f.
Examples
LM(order=None)
Returns the leading monomial of f.
The Leading monomial signifies the monomial having the highest power of the prin-
cipal generator in the expression f.
Examples
LT(order=None)
Returns the leading term of f.
The Leading term signifies the term having the highest power of the principal gen-
erator in the expression f along with its coefficient.
Examples
TC()
Returns the trailing coefficient of f.
Examples
abs()
Make all coefficients in f positive.
Examples
add(g)
Add two polynomials f and g.
Examples
add_ground(coeff)
Add an element of the ground domain to f.
Examples
all_coeffs()
Returns all coefficients from a univariate polynomial f.
Examples
all_monoms()
Returns all monomials from a univariate polynomial f.
See also:
all_terms (page 851)
Examples
all_roots(multiple=True, radicals=True)
Return a list of real and complex roots with multiplicities.
Examples
all_terms()
Returns all terms from a univariate polynomial f.
Examples
args
Don’t mess up with the core.
Examples
as_dict(native=False, zero=False)
Switch to a dict representation.
Examples
as_expr(*gens)
Convert a Poly instance to an Expr instance.
Examples
>>> f.as_expr()
x**2 + 2*x*y**2 - y
>>> f.as_expr({x: 5})
10*y**2 - y + 25
>>> f.as_expr(5, 6)
379
as_list(native=False)
Switch to a list representation.
cancel(g, include=False)
Cancel common factors in a rational function f/g.
Examples
clear_denoms(convert=False)
Clear denominators, but keep the ground domain.
Examples
>>> f.clear_denoms()
(6, Poly(3*x + 2, x, domain='QQ'))
>>> f.clear_denoms(convert=True)
(6, Poly(3*x + 2, x, domain='ZZ'))
coeff_monomial(monom)
Returns the coefficient of monom in f if there, else None.
See also:
nth (page 874) more efficient query using exponents of the monomial’s generators
Examples
>>> p.coeff_monomial(x)
23
>>> p.coeff_monomial(y)
0
>>> p.coeff_monomial(x*y)
24*exp(8)
Note that Expr.coeff() behaves differently, collecting terms if possible; the Poly
must be converted to an Expr to use that method, however:
>>> p.as_expr().coeff(x)
24*y*exp(8) + 23
>>> p.as_expr().coeff(y)
24*x*exp(8)
>>> p.as_expr().coeff(x*y)
24*exp(8)
coeffs(order=None)
Returns all non-zero coefficients from f in lex order.
See also:
all_coeffs (page 850), coeff_monomial (page 852), nth (page 874)
Examples
cofactors(g)
Returns the GCD of f and g and their cofactors.
Returns polynomials (h, cff, cfg) such that h = gcd(f, g), and cff = quo(f,
h) and cfg = quo(g, h) are, so called, cofactors of f and g.
Examples
compose(g)
Computes the functional composition of f and g.
Examples
content()
Returns the GCD of polynomial coefficients.
Examples
count_roots(inf=None, sup=None)
Return the number of roots of f in [inf, sup] interval.
Examples
decompose()
Computes a functional decomposition of f.
Examples
deflate()
Reduce degree of f by mapping x_i**m to y_i.
Examples
degree(gen=0)
Returns degree of f in x_j.
The degree of 0 is negative infinity.
Examples
degree_list()
Returns a list of degrees of f.
Examples
diff(*specs, **kwargs)
Computes partial derivative of f.
Examples
discriminant()
Computes the discriminant of f.
Examples
dispersion(g=None)
Compute the dispersion of polynomials.
For two polynomials f (x) and g(x) with deg f > 0 and deg g > 0 the dispersion dis(f, g)
is defined as:
dis(f, g) := max{J(f, g) ∪ {0}}
= max{{a ∈ N| gcd(f (x), g(x + a)) ̸= 1} ∪ {0}}
References
1. [ManWright94]
2. [Koepf98]
3. [Abramov71]
4. [Man93]
Examples
We can even perform the computations for polynomials having symbolic coefficients:
>>> sorted(dispersionset(fp))
[0, 1]
dispersionset(g=None)
Compute the dispersion set of two polynomials.
For two polynomials f (x) and g(x) with deg f > 0 and deg g > 0 the dispersion set
J(f, g) is defined as:
References
1. [ManWright94]
2. [Koepf98]
3. [Abramov71]
4. [Man93]
Examples
We can even perform the computations for polynomials having symbolic coefficients:
>>> sorted(dispersionset(fp))
[0, 1]
div(g, auto=True)
Polynomial division with remainder of f by g.
Examples
domain
Get the ground domain of self.
eject(*gens)
Eject selected generators into the ground domain.
Examples
>>> f.eject(x)
Poly(x*y**3 + (x**2 + x)*y + 1, y, domain='ZZ[x]')
>>> f.eject(y)
Poly(y*x**2 + (y**3 + y)*x + 1, x, domain='ZZ[y]')
Examples
exclude()
Remove unnecessary generators from f.
Examples
exquo(g, auto=True)
Computes polynomial exact quotient of f by g.
Examples
exquo_ground(coeff)
Exact quotient of f by a an element of the ground domain.
Examples
factor_list()
Returns a list of irreducible factors of f.
Examples
>>> Poly(f).factor_list()
(2, [(Poly(x + y, x, y, domain='ZZ'), 1),
(Poly(x**2 + 1, x, y, domain='ZZ'), 2)])
factor_list_include()
Returns a list of irreducible factors of f.
Examples
>>> Poly(f).factor_list_include()
[(Poly(2*x + 2*y, x, y, domain='ZZ'), 1),
(Poly(x**2 + 1, x, y, domain='ZZ'), 2)]
free_symbols
Free symbols of a polynomial expression.
Examples
free_symbols_in_domain
Free symbols of the domain of self.
Examples
Examples
gcdex(g, auto=True)
Extended Euclidean algorithm of f and g.
Returns (s, t, h) such that h = gcd(f, g) and s*f + t*g = h.
Examples
>>> Poly(f).gcdex(Poly(g))
(Poly(-1/5*x + 3/5, x, domain='QQ'),
Poly(1/5*x**2 - 6/5*x + 2, x, domain='QQ'),
Poly(x + 1, x, domain='QQ'))
gen
Return the principal generator.
Examples
get_domain()
Get the ground domain of f.
get_modulus()
Get the modulus of f.
Examples
gff_list()
Computes greatest factorial factorization of f.
Examples
>>> Poly(f).gff_list()
[(Poly(x, x, domain='ZZ'), 1), (Poly(x + 2, x, domain='ZZ'), 4)]
ground_roots()
Compute roots of f by factorization in the ground domain.
Examples
half_gcdex(g, auto=True)
Half extended Euclidean algorithm of f and g.
Returns (s, h) such that h = gcd(f, g) and s*f = h (mod g).
Examples
>>> Poly(f).half_gcdex(Poly(g))
(Poly(-1/5*x + 3/5, x, domain='QQ'), Poly(x + 1, x, domain='QQ'))
has_only_gens(*gens)
Return True if Poly(f, *gens) retains ground domain.
Examples
homogeneous_order()
Returns the homogeneous order of f.
A homogeneous polynomial is a polynomial whose all monomials with non-zero co-
efficients have the same total degree. This degree is the homogeneous order of
f. If you only want to check if a polynomial is homogeneous, then use Poly.
is_homogeneous() (page 867).
Examples
homogenize(s)
Returns the homogeneous polynomial of f.
A homogeneous polynomial is a polynomial whose all monomials with non-zero co-
efficients have the same total degree. If you only want to check if a polynomial is
homogeneous, then use Poly.is_homogeneous() (page 867). If you want not only
to check if a polynomial is homogeneous but also compute its homogeneous order,
then use Poly.homogeneous_order() (page 864).
Examples
inject(front=False)
Inject ground domain generators into f.
Examples
>>> f.inject()
Poly(x**2*y + x*y**3 + x*y + 1, x, y, domain='ZZ')
>>> f.inject(front=True)
Poly(y**3*x + y*x**2 + y*x + 1, y, x, domain='ZZ')
integrate(*specs, **args)
Computes indefinite integral of f.
Examples
References
Examples
invert(g, auto=True)
Invert f modulo g when possible.
Examples
is_cyclotomic
Returns True if f is a cyclotomic polnomial.
Examples
>>> Poly(f).is_cyclotomic
False
>>> Poly(g).is_cyclotomic
True
is_ground
Returns True if f is an element of the ground domain.
Examples
is_homogeneous
Returns True if f is a homogeneous polynomial.
A homogeneous polynomial is a polynomial whose all monomials with non-zero co-
efficients have the same total degree. If you want not only to check if a poly-
nomial is homogeneous but also compute its homogeneous order, then use Poly.
homogeneous_order() (page 864).
Examples
is_irreducible
Returns True if f has no factors over its domain.
Examples
is_linear
Returns True if f is linear in all its variables.
Examples
is_monic
Returns True if the leading coefficient of f is one.
Examples
is_monomial
Returns True if f is zero or has only one term.
Examples
is_multivariate
Returns True if f is a multivariate polynomial.
Examples
is_one
Returns True if f is a unit polynomial.
Examples
is_primitive
Returns True if GCD of the coefficients of f is one.
Examples
is_quadratic
Returns True if f is quadratic in all its variables.
Examples
is_sqf
Returns True if f is a square-free polynomial.
Examples
is_univariate
Returns True if f is a univariate polynomial.
Examples
is_zero
Returns True if f is a zero polynomial.
Examples
l1_norm()
Returns l1 norm of f.
Examples
lcm(g)
Returns polynomial LCM of f and g.
Examples
length()
Returns the number of non-zero terms in f.
Examples
lift()
Convert algebraic coefficients to rationals.
Examples
ltrim(gen)
Remove dummy generators from f that are to the left of specified gen in the gener-
ators as ordered. When gen is an integer, it refers to the generator located at that
position within the tuple of generators of f.
Examples
max_norm()
Returns maximum norm of f.
Examples
monic(auto=True)
Divides all coefficients by LC(f).
Examples
monoms(order=None)
Returns all non-zero monomials from f in lex order.
See also:
all_monoms (page 850)
Examples
mul(g)
Multiply two polynomials f and g.
Examples
mul_ground(coeff)
Multiply f by a an element of the ground domain.
Examples
neg()
Negate all coefficients in f.
Examples
>>> -Poly(x**2 - 1, x)
Poly(-x**2 + 1, x, domain='ZZ')
Examples
Examples
nth(*N)
Returns the n-th coefficient of f where N are the exponents of the generators in the
term of interest.
See also:
coeff_monomial (page 852)
Examples
nth_power_roots_poly(n)
Construct a polynomial with n-th powers of roots of f.
Examples
>>> f.nth_power_roots_poly(2)
Poly(x**4 - 2*x**3 + 3*x**2 - 2*x + 1, x, domain='ZZ')
>>> f.nth_power_roots_poly(3)
Poly(x**4 + 2*x**2 + 1, x, domain='ZZ')
>>> f.nth_power_roots_poly(4)
Poly(x**4 + 2*x**3 + 3*x**2 + 2*x + 1, x, domain='ZZ')
>>> f.nth_power_roots_poly(12)
Poly(x**4 - 4*x**3 + 6*x**2 - 4*x + 1, x, domain='ZZ')
one
Return one polynomial with self’s properties.
pdiv(g)
Polynomial pseudo-division of f by g.
Examples
Examples
>>> a = Poly(x**2 + 1)
pexquo(g)
Polynomial exact pseudo-quotient of f by g.
Examples
pow(n)
Raise f to a non-negative power n.
Examples
pquo(g)
Polynomial pseudo-quotient of f by g.
See the Caveat note in the function prem(f, g).
Examples
prem(g)
Polynomial pseudo-remainder of f by g.
Caveat: The function prem(f, g, x) can be safely used to compute in Z[x]
_only_ subresultant polynomial remainder sequences (prs’s).
To safely compute Euclidean and Sturmian prs’s in Z[x] employ anyone of the cor-
responding functions found in the module sympy.polys.subresultants_qq_zz. The
functions in the module with suffix _pg compute prs’s in Z[x] employing rem(f,
g, x), whereas the functions with suffix _amv compute prs’s in Z[x] employing
rem_z(f, g, x).
The function rem_z(f, g, x) differs from prem(f, g, x) in that to compute the re-
mainder polynomials in Z[x] it premultiplies the divident times the absolute value
of the leading coefficient of the divisor raised to the power degree(f, x) - degree(g,
x) + 1.
Examples
primitive()
Returns the content and a primitive form of f.
Examples
quo(g, auto=True)
Computes polynomial quotient of f by g.
Examples
quo_ground(coeff)
Quotient of f by a an element of the ground domain.
Examples
rat_clear_denoms(g)
Clear denominators in a rational function f/g.
Examples
>>> f = Poly(x**2/y + 1, x)
>>> g = Poly(x**3 + y, x)
>>> p, q = f.rat_clear_denoms(g)
>>> p
Poly(x**2 + y, x, domain='ZZ[y]')
>>> q
Poly(y*x**3 + y**2, x, domain='ZZ[y]')
real_roots(multiple=True, radicals=True)
Return a list of real roots with multiplicities.
Examples
Examples
rem(g, auto=True)
Computes the polynomial remainder of f by g.
Examples
reorder(*gens, **args)
Efficiently apply new order of generators.
Examples
replace(x, y=None)
Replace x with y in generators list.
Examples
resultant(g, includePRS=False)
Computes the resultant of f and g via PRS.
If includePRS=True, it includes the subresultant PRS in the result. Because the PRS
is used to calculate the resultant, this is more efficient than calling subresultants()
(page 833) separately.
Examples
>>> f = Poly(x**2 + 1, x)
retract(field=None)
Recalculate the ground domain of a polynomial.
Examples
>>> f.retract()
Poly(x**2 + 1, x, domain='ZZ')
>>> f.retract(field=True)
Poly(x**2 + 1, x, domain='QQ')
revert(n)
Compute f**(-1) mod x**n.
Examples
root(index, radicals=True)
Get an indexed root of a polynomial.
Examples
>>> f.root(0)
-1/2
>>> f.root(1)
2
>>> f.root(2)
2
>>> f.root(3)
Traceback (most recent call last):
...
IndexError: root index out of [-3, 2] range, got 3
set_domain(domain)
Set the ground domain of f.
set_modulus(modulus)
Set the modulus of f.
Examples
shift(a)
Efficiently compute Taylor shift f(x + a).
Examples
slice(x, m, n=None)
Take a continuous subsequence of terms of f.
sqf_list(all=False)
Returns a list of square-free factors of f.
Examples
>>> Poly(f).sqf_list()
(2, [(Poly(x + 1, x, domain='ZZ'), 2),
(Poly(x + 2, x, domain='ZZ'), 3)])
>>> Poly(f).sqf_list(all=True)
(2, [(Poly(1, x, domain='ZZ'), 1),
(Poly(x + 1, x, domain='ZZ'), 2),
(Poly(x + 2, x, domain='ZZ'), 3)])
sqf_list_include(all=False)
Returns a list of square-free factors of f.
Examples
>>> Poly(f).sqf_list_include()
[(Poly(2, x, domain='ZZ'), 1),
(Poly(x + 1, x, domain='ZZ'), 3),
(Poly(x, x, domain='ZZ'), 4)]
>>> Poly(f).sqf_list_include(all=True)
[(Poly(2, x, domain='ZZ'), 1),
(Poly(1, x, domain='ZZ'), 2),
(Poly(x + 1, x, domain='ZZ'), 3),
(Poly(x, x, domain='ZZ'), 4)]
sqf_norm()
Computes square-free norm of f.
Returns s, f, r, such that g(x) = f(x-sa) and r(x) = Norm(g(x)) is a square-free
polynomial over K, where a is the algebraic extension of the ground domain.
Examples
>>> s
1
>>> f
Poly(x**2 - 2*sqrt(3)*x + 4, x, domain='QQ<sqrt(3)>')
>>> r
Poly(x**4 - 4*x**2 + 16, x, domain='QQ')
sqf_part()
Computes square-free part of f.
Examples
sqr()
Square a polynomial f.
Examples
sturm(auto=True)
Computes the Sturm sequence of f.
Examples
sub(g)
Subtract two polynomials f and g.
Examples
sub_ground(coeff)
Subtract an element of the ground domain from f.
Examples
subresultants(g)
Computes the subresultant PRS of f and g.
Examples
terms(order=None)
Returns all non-zero terms from f in lex order.
See also:
all_terms (page 851)
Examples
terms_gcd()
Remove GCD of terms from the polynomial f.
Examples
Examples
to_exact()
Make the ground domain exact.
Examples
to_field()
Make the ground domain a field.
Examples
to_ring()
Make the ground domain a ring.
Examples
total_degree()
Returns the total degree of f.
Examples
transform(p, q)
Efficiently evaluate the functional transformation q**n * f(p/q).
Examples
trunc(p)
Reduce f modulo a constant p.
Examples
unify(g)
Make f and g belong to the same domain.
Examples
>>> f
Poly(1/2*x + 1, x, domain='QQ')
>>> g
Poly(2*x + 1, x, domain='ZZ')
>>> F, G = f.unify(g)
>>> F
Poly(1/2*x + 1, x, domain='QQ')
>>> G
Poly(2*x + 1, x, domain='QQ')
unit
Return unit polynomial with self’s properties.
zero
Return zero polynomial with self’s properties.
class sympy.polys.polytools.PurePoly
Class for representing pure polynomials.
free_symbols
Free symbols of a polynomial.
Examples
class sympy.polys.polytools.GroebnerBasis
Represents a reduced Groebner basis.
contains(poly)
Check if poly belongs the ideal generated by self.
Examples
>>> G.contains(f)
True
>>> G.contains(f + 1)
False
fglm(order)
Convert a Groebner basis from one ordering to another.
The FGLM algorithm converts reduced Groebner bases of zero-dimensional ideals
from one ordering to another. This method is often used when it is infeasible to
compute a Groebner basis with respect to a particular ordering directly.
References
Examples
>>> list(G.fglm('lex'))
[2*x - y**2 - y + 1, y**4 + 2*y**3 - 3*y**2 - 16*y + 7]
>>> list(groebner(F, x, y, order='lex'))
[2*x - y**2 - y + 1, y**4 + 2*y**3 - 3*y**2 - 16*y + 7]
is_zero_dimensional
Checks if the ideal generated by a Groebner basis is zero-dimensional.
The algorithm checks if the set of monomials not divisible by the leading monomial
of any element of F is bounded.
References
David A. Cox, John B. Little, Donal O’Shea. Ideals, Varieties and Algorithms, 3rd
edition, p. 230
reduce(expr, auto=True)
Reduces a polynomial modulo a Groebner basis.
Given a polynomial f and a set of polynomials G = (g_1, ..., g_n), computes a
set of quotients q = (q_1, ..., q_n) and the remainder r such that f = q_1*f_1
+ ... + q_n*f_n + r, where r vanishes or r is a completely reduced polynomial
with respect to G.
Examples
>>> G.reduce(f)
([2*x, 1], x**2 + y**2 + y)
>>> Q, r = _
Examples
References
[1] - http://en.wikipedia.org/wiki/Horner_scheme
Examples
sympy.polys.polyfuncs.interpolate(data, x)
Construct an interpolating polynomial for the data points.
Examples
The (x, y) coordinates can also be given as keys and values of a dictionary (and the points
need not be equispaced):
Examples
Domain constructors
sympy.polys.constructor.construct_domain(obj, **args)
Construct a minimal domain for the list of coefficients.
Notes
Examples
>>> minimal_polynomial(sqrt(2), x)
x**2 - 2
>>> minimal_polynomial(sqrt(2), x, domain=QQ.algebraic_field(sqrt(2)))
x - sqrt(2)
>>> minimal_polynomial(sqrt(2) + sqrt(3), x)
x**4 - 10*x**2 + 1
>>> minimal_polynomial(solve(x**3 + x + 3)[0], x)
x**3 + x + 3
>>> minimal_polynomial(sqrt(y), x)
x**2 - y
Notes
Examples
>>> minimal_polynomial(sqrt(2), x)
x**2 - 2
>>> minimal_polynomial(sqrt(2), x, domain=QQ.algebraic_field(sqrt(2)))
x - sqrt(2)
>>> minimal_polynomial(sqrt(2) + sqrt(3), x)
x**4 - 10*x**2 + 1
(continues on next page)
Examples
sympy.polys.monomials.monomial_count(V, N)
Computes the number of monomials.
The number of monomials is given by the following formula:
(#V + N )!
#V !N !
where N is a total degree and V is a set of variables.
Examples
>>> monomial_count(2, 2)
6
Orderings of monomials
class sympy.polys.orderings.LexOrder
Lexicographic order of monomials.
class sympy.polys.orderings.GradedLexOrder
Graded lexicographic order of monomials.
class sympy.polys.orderings.ReversedGradedLexOrder
Reversed graded lexicographic order of monomials.
Examples
Whether roots involving radicals are returned or not depends on whether the radicals
flag is true (which is set to True with rootof):
>>> CRootOf(x**2 - 3, 0)
CRootOf(x**2 - 3, 0)
>>> CRootOf(x**2 - 3, 0, radicals=True)
-sqrt(3)
>>> rootof(x**2 - 3, 0)
-sqrt(3)
The root bounds can be seen, however, and they are used by the evaluation methods to
get numerical approximations for the root.
>>> interval = r._get_interval(); interval
(-1, 0)
>>> r.evalf(2)
-0.98
The evalf method refines the width of the root bounds until it guarantees that any decimal
approximation within those bounds will satisfy the desired precision. It then stores the
refined interval so subsequent requests at or below the requested precision will not have
to recompute the root bounds and will return very quickly.
Before evaluation above, the interval was
>>> interval
(-1, 0)
The evala pprox method will also find the root to a given precision but the interval is not
modified unless the search for the root fails to converge within the root bounds. And
the secant method is used to find the root. (The evalf method uses bisection and will
always update the interval.)
>>> r.eval_approx(2)
-0.98
The evalf_rational will compute a rational approximation of the root to the desired
accuracy or precision.
>>> r.eval_rational(n=2)
-69629/71318
class sympy.polys.rootoftools.RootSum
Represents a sum of all roots of a univariate polynomial.
References
1. http://en.wikipedia.org/wiki/Cubic_function#Trigonometric_.28and_hyperbolic.29_
method
Examples
>>> roots(x**2 - 1, x)
{-1: 1, 1: 1}
>>> p = Poly(x**2-1, x)
>>> roots(p)
{-1: 1, 1: 1}
>>> p = Poly(x**2-y, x, y)
>>> roots(x**2 - y, x)
{-sqrt(y): 1, sqrt(y): 1}
Special polynomials
n : int
n decides the order of polynomial
inf
Lower limit of range in which coefficients lie
sup
Upper limit of range in which coefficients lie
domain : optional
Decides what ring the coefficients are supposed to belong. Default is
set to Integers.
polys : bool, optional
polys=True returns an expression, otherwise (default) returns an ex-
pression.
Orthogonal polynomials
sympy.polys.rationaltools.together(expr, deep=False)
Denest and combine rational expressions using symbolic methods.
This function takes an expression or a container of expressions and puts it (them) to-
gether by denesting and combining rational subexpressions. No heroic measures are
taken to minimize degree of the resulting numerator and denominator. To obtain com-
pletely reduced expression use cancel(). However, together() (page 901) can pre-
serve as much as possible of the structure of the input expression in the output (no
expansion is performed).
A wide variety of objects can be put together including lists, tuples, sets, relational ob-
jects, integrals and others. It is also possible to transform interior of function applica-
tions, by setting deep flag to True.
By definition, together() (page 901) is a complement to apart(), so
apart(together(expr)) should return expr unchanged. Note however, that to-
gether() (page 901) uses only symbolic methods, so it might be necessary to use
cancel() to perform algebraic simplification and minimize degree of the numerator and
denominator.
Examples
Examples
The undetermined coefficients method does not provide a result when the denominators
roots are not rational:
return value is highly structured and perfectly suited for further algorithmic treatment
rather than being human-readable. The function returns a tuple holding three elements:
• The first item is the common coefficient, free of the variable x used for decomposi-
tion. (It is an element of the base field K.)
• The second item is the polynomial part of the decomposition. This can be the zero
polynomial. (It is an element of K[x].)
• The third part itself is a list of quadruples. Each quadruple has the following ele-
ments in this order:
– The (not necessarily irreducible) polynomial D whose roots wi appear in the lin-
ear denominator of a bunch of related fraction terms. (This item can also be a
list of explicit roots. However, at the moment apart_list never returns a result
this way, but the related assemble_partfrac_list function accepts this format
as input.)
– The numerator of the fraction, written as a function of the root w
– The linear denominator of the fraction excluding its power exponent, written as
a function of the root w.
– The power to which the denominator has to be raised.
On can always rebuild a plain expression by using the function assem-
ble_partfrac_list.
See also:
apart (page 902), assemble_partfrac_list (page 904)
References
1. [Bronstein93]
[Bronstein93]
Examples
A first example:
>>> assemble_partfrac_list(pfd)
2*x + 4 + 4/(x - 1)
Second example:
>>> assemble_partfrac_list(pfd)
-2/3 - 2/(x - 2)
>>> assemble_partfrac_list(pfd)
RootSum(_w**2 + _w + t, Lambda(_a, (-2*_a*t/(4*t - 1) - t/(4*t - 1))/(-_a + x)))
>>> assemble_partfrac_list(pfd)
-4/(x + 1) - 3/(x + 1)**2 - 9/(x - 1)**2 + 4/(x - 2)
sympy.polys.partfrac.assemble_partfrac_list(partial_list)
Reassemble a full partial fraction decomposition from a structured result obtained by
the function apart_list.
See also:
apart (page 902), apart_list (page 902)
Examples
>>> assemble_partfrac_list(pfd)
-4/(x + 1) - 3/(x + 1)**2 - 9/(x - 1)**2 + 4/(x - 2)
If we happen to know some roots we can provide them easily inside the structure:
>>> pfda.doit()
-sqrt(2)/(2*(x + sqrt(2))) + sqrt(2)/(2*(x - sqrt(2)))
>>> assemble_partfrac_list(pfd)
-sqrt(2)/(2*(x + sqrt(2))) + sqrt(2)/(2*(x - sqrt(2)))
Dispersion of Polynomials
References
1. [ManWright94]
2. [Koepf98]
3. [Abramov71]
4. [Man93]
Examples
We can even perform the computations for polynomials having symbolic coefficients:
and for a single polynomial dis(f ) := dis(f, f ). Note that we make the definition max{} :=
−∞.
See also:
dispersionset (page 835)
References
1. [ManWright94]
2. [Koepf98]
3. [Abramov71]
4. [Man93]
Examples
We can even perform the computations for polynomials having symbolic coefficients:
Introduction
Reference
In this section we document the usage of the AGCA module. For convenience of the reader,
some definitions and examples/explanations are interspersed.
Base Rings
Almost all computations in commutative algebra are relative to a “base ring”. (For example,
when asking questions about an ideal, the base ring is the ring the ideal is a subset of.) In
principle all polys “domains” can be used as base rings. However, useful functionality is only
implemented for polynomial rings over fields, and various localizations and quotients thereof.
As demonstrated in the examples below, the most convenient method to create objects you
are interested in is to build them up from the ground field, and then use the various methods
to create new objects from old. For example, in order to create the local ring of the nodal
cubic y 2 = x3 at the origin, over Q, you do:
Note how the python list notation can be used as a short cut to express ideals. You can use
the convert method to return ordinary sympy objects into objects understood by the AGCA
module (although in many cases this will be done automatically – for example the list was
automatically turned into an ideal, and in the process the symbols x and y were automatically
converted into other representations). For example:
When no localisation is needed, a more mathematical notation can be used. For example, let
us create the coordinate ring of three-dimensional affine space A3 :
For more details, refer to the following class documentation. Note that the base rings, being
domains, are the main point of overlap between the AGCA module and the rest of the polys
module. All domains are documented in detail in the polys reference, so we show here only
an abridged version, with the methods most pertinent to the AGCA module.
class sympy.polys.domains.ring.Ring
Represents a ring domain.
Attributes
alias
dtype
one
rep
zero
free_module(rank)
Generate a free module of rank rank over self.
ideal(*gens)
Generate an ideal of self.
quotient_ring(e)
Form a quotient ring of self.
Here e can be an ideal or an iterable.
>>> QQ.old_poly_ring(x)/[x**2]
QQ[x]/<x**2>
sympy.polys.domains.polynomialring.PolynomialRing(domain_or_ring, sym-
bols=None, order=None)
A class for representing multivariate polynomial rings.
Attributes
alias
domain
dtype
gens
ngens
rep
symbols
>>> QQ.old_poly_ring(x)/I
QQ[x]/<x**3 + 1>
>>> QQ.old_poly_ring(x)/[x**3 + 1]
QQ[x]/<x**3 + 1>
Attributes:
• ring - the base ring
• base_ideal - the ideal used to form the quotient
Attributes
alias
one
rep
zero
The ring A itself has a natural A-module structure where addition and multiplication in the
module coincide with addition and multiplication in the ring. This A-module is also written as
A. An A-submodule of A is called an ideal of A. Ideals come up very naturally in algebraic ge-
ometry. More general modules can be seen as a technically convenient “elbow room” beyond
talking only about ideals.
If M , N are A-modules, then there is a natural (componentwise) A-module structure on M × N .
Similarly there are A-module structures on cartesian products of more components. (For
the categorically inclined: the cartesian product of finitely many A-modules, with this A-
module structure, is the finite biproduct in the category of all A-modules. With infinitely
many components, it is the direct product (but the infinite direct sum has to be constructed
differently).) As usual, repeated product of the A-module M is denoted M, M 2 , M 3 . . ., or M I
for arbitrary index sets I.
An A-module M is called free if it is isomorphic to the A-module AI for some (not necessarily
finite) index set I (refer to the next section for a definition of isomorphism). The cardinality
of I is called the rank of M ; one may prove this is well-defined. In general, the AGCA module
only works with free modules of finite rank, and other closely related modules. The easiest
way to create modules is to use member methods of the objects they are made up from. For
example, let us create a free module of rank 4 over the coordinate ring of A2 we created above,
together with a submodule:
>>> F = ar.free_module(4) ; F
4
ℚ[x, y, z]
Note how python lists can be used as a short-cut notation for module elements (vectors). As
usual, the convert method can be used to convert sympy/python objects into the internal
AGCA representation (see detailed reference below).
Here is the detailed documentation of the classes for modules, free modules, and submodules:
class sympy.polys.agca.modules.Module(ring)
Abstract base class for modules.
Do not instantiate - use ring explicit constructors instead:
Attributes:
• dtype - type of elements
• ring - containing ring
Non-implemented methods:
• submodule
• quotient_module
• is_zero
• is_submodule
• multiply_ideal
The method convert likely needs to be changed in subclasses.
contains(elem)
Return True if elem is an element of this module.
convert(elem, M=None)
Convert elem into internal representation of this module.
If M is not None, it should be a module containing it.
identity_hom()
Return the identity homomorphism on self.
is_submodule(other)
Returns True if other is a submodule of self.
is_zero()
Returns True if self is a zero module.
multiply_ideal(other)
Multiply self by the ideal other.
quotient_module(other)
Generate a quotient module.
submodule(*gens)
Generate a submodule.
subset(other)
Returns True if other is is a subset of self.
convert(elem, M=None)
Convert elem into the internal representation.
This method is called implicitly whenever computations involve elements not in the
internal representation.
dtype
alias of FreeModuleElement
identity_hom()
Return the identity homomorphism on self.
is_submodule(other)
Returns True if other is a submodule of self.
is_zero()
Returns True if self is a zero module.
(If, as this implementation assumes, the coefficient ring is not the zero ring, then
this is equivalent to the rank being zero.)
multiply_ideal(other)
Multiply self by the ideal other.
quotient_module(submodule)
Return a quotient module.
identity_hom()
Return the identity homomorphism on self.
in_terms_of_generators(e)
Express element e of self in terms of the generators.
inclusion_hom()
Return a homomorphism representing the inclusion map of self.
That is, the natural map from self to self.container.
intersect(other, **options)
Returns the intersection of self with submodule other.
The above result says: the intersection module is generated by the single element
(−xy, −xy) = −y(x, x) = −x(y, y), where (x, x) and (y, y) respectively are the unique
generators of the two modules being intersected.
is_full_module()
Return True if self is the entire free module.
is_submodule(other)
Returns True if other is a submodule of self.
is_zero()
Return True if self is a zero module.
module_quotient(other, **options)
Returns the module quotient of self by submodule other.
That is, if self is the module M and other is N , then return the ideal {f ∈ R|f N ⊂ M }.
Some implementations allow further options to be passed. Currently, the only one
implemented is relations=True, which may only be passed if other is principal. In
this case the function will return a pair (res, rel) where res is the ideal, and rel
is a list of coefficient vectors, expressing the generators of the ideal, multiplied by
the generator of other in terms of generators of self.
This means that the quotient ideal is generated by the single element y, and that
y(x, x) = 1(xy, xy), (x, x) and (xy, xy) being the generators of T and S, respectively.
multiply_ideal(I)
Multiply self by the ideal I.
quotient_module(other, **opts)
Return a quotient module.
This is the same as taking a submodule of a quotient of the containing module.
reduce_element(x)
Reduce the element x of our ring modulo the ideal self.
Here “reduce” has no specific meaning, it could return a unique normal form, sim-
plify the expression a bit, or just do nothing.
submodule(*gens)
Generate a submodule.
syzygy_module(**opts)
Compute the syzygy module of the generators of self.
Suppose M is generated by f1 , . . . , fn over the ring R. Consider the homomorphism
ϕ : Rn → M , given by sending (r1 , . . . , rn ) → r1 f1 + · · · + rn fn . The syzygy module is
defined to be the kernel of ϕ.
The syzygy module is zero iff the generators generate freely a free submodule:
True
union(other)
Returns the module generated by the union of self and other.
Ideals are created very similarly to modules. For example, let’s verify that the nodal cubic is
indeed singular at the origin:
>>> I = lr.ideal(x, y)
>>> I == lr.ideal(x)
False
>>> I == lr.ideal(y)
False
We are using here the fact that a curve is non-singular at a point if and only if the maximal ideal
of the local ring is principal, and that in this case at least one of x and y must be generators.
This is the detailed documentation of the class ideal. Please note that most of the methods
regarding properties of ideals (primality etc.) are not yet implemented.
class sympy.polys.agca.ideals.Ideal(ring)
Abstract base class for ideals.
Do not instantiate - use explicit constructors in the ring class instead:
Attributes
• ring - the ring this ideal belongs to
Non-implemented methods:
• _contains_elem
• _contains_ideal
• _quotient
• _intersect
• _union
• _product
• is_whole_ring
• is_zero
• is_prime, is_maximal, is_primary, is_radical
• is_principal
• height, depth
• radical
depth()
Compute the depth of self.
height()
Compute the height of self.
intersect(J)
Compute the intersection of self with ideal J.
is_maximal()
Return True if self is a maximal ideal.
is_primary()
Return True if self is a primary ideal.
is_prime()
Return True if self is a prime ideal.
is_principal()
Return True if self is a principal ideal.
is_radical()
Return True if self is a radical ideal.
is_whole_ring()
Return True if self is the whole ring.
is_zero()
Return True if self is the zero ideal.
product(J)
Compute the ideal product of self and J.
That is, compute the ideal generated by products xy, for x an element of self and
y ∈ J.
quotient(J, **opts)
Compute the ideal quotient of self by J.
That is, if self is the ideal I, compute the set I : J = {x ∈ R|xJ ⊂ I}.
radical()
Compute the radical of self.
reduce_element(x)
Reduce the element x of our ring modulo the ideal self.
Here “reduce” has no specific meaning: it could return a unique normal form, sim-
plify the expression a bit, or just do nothing.
saturate(J)
Compute the ideal saturation of self by J.
That is, if self is the ideal I, compute the set I : J ∞ = {x ∈ R|xJ n ⊂ I for some n}.
subset(other)
Returns True if other is is a subset of self.
Here other may be an ideal.
union(J)
Compute the ideal generated by the union of self and J.
True
dtype
alias of QuotientModuleElement
identity_hom()
Return the identity homomorphism on self.
is_submodule(other)
Return True if other is a submodule of self.
is_zero()
Return True if self is a zero module.
This happens if and only if the base module is the same as the submodule being
killed.
quotient_hom()
Return the quotient homomorphism to self.
That is, return a homomorphism representing the natural map from self.base to
self.
submodule(*gens, **opts)
Generate a submodule.
This is the same as taking a quotient of a submodule of the base module.
Attributes:
• base - base module we are quotient of
• killed_module - submodule used to form the quotient
is_full_module()
Return True if self is the entire free module.
quotient_hom()
Return the quotient homomorphism to self.
That is, return the natural map from self.base to self.
the operations described above, and homomorphisms of free modules can in principle be
instantiated by hand. Since these operations are so common, there is a convenience function
homomorphism to define a homomorphism between arbitrary modules via the method outlined
above. It is essentially the only way homomorphisms need ever be created by the user.
sympy.polys.agca.homomorphisms.homomorphism(domain, codomain, matrix)
Create a homomorphism object.
This function tries to build a homomorphism from domain to codomain via the matrix
matrix.
Examples
>>> R = QQ.old_poly_ring(x)
>>> T = R.free_module(2)
>>> F = R.free_module(2)
>>> h = homomorphism(F, T, [[1, x], [x**2, 0]])
>>> h
Matrix([
[1, x**2], : QQ[x]**2 -> QQ[x]**2
[x, 0]])
>>> h([1, 0])
[1, x]
>>> h([0, 1])
[x**2, 0]
>>> h([1, 1])
[x**2 + 1, x]
>>> F = QQ.old_poly_ring(x).free_module(2)
>>> homomorphism(F, F, [[1, 0], [0, 1]])
Matrix([
[1, 0], : QQ[x]**2 -> QQ[x]**2
[0, 1]])
Attributes:
• ring - the ring over which we are considering modules
• domain - the domain module
• codomain - the codomain module
• _ker - cached kernel
• _img - cached image
Non-implemented methods:
• _kernel
• _image
• _restrict_domain
• _restrict_codomain
• _quotient_domain
• _quotient_codomain
• _apply
• _mul_scalar
• _compose
• _add
image()
Compute the image of self.
That is, if self is the homomorphism ϕ : M → N , then compute im(ϕ) = {ϕ(x)|x ∈ M }.
This is a submodule of N .
>>> F = QQ.old_poly_ring(x).free_module(2)
>>> homomorphism(F, F, [[1, 0], [x, 0]]).image() == F.submodule([1, 0])
True
is_injective()
Return True if self is injective.
That is, check if the elements of the domain are mapped to the same codomain
element.
>>> F = QQ.old_poly_ring(x).free_module(2)
>>> h = homomorphism(F, F, [[1, 0], [x, 0]])
>>> h.is_injective()
False
>>> h.quotient_domain(h.kernel()).is_injective()
True
is_isomorphism()
Return True if self is an isomorphism.
That is, check if every element of the codomain has precisely one preimage. Equiv-
alently, self is both injective and surjective.
>>> F = QQ.old_poly_ring(x).free_module(2)
>>> h = homomorphism(F, F, [[1, 0], [x, 0]])
>>> h = h.restrict_codomain(h.image())
>>> h.is_isomorphism()
False
>>> h.quotient_domain(h.kernel()).is_isomorphism()
True
is_surjective()
Return True if self is surjective.
That is, check if every element of the codomain has at least one preimage.
>>> F = QQ.old_poly_ring(x).free_module(2)
>>> h = homomorphism(F, F, [[1, 0], [x, 0]])
>>> h.is_surjective()
False
(continues on next page)
is_zero()
Return True if self is a zero morphism.
That is, check if every element of the domain is mapped to zero under self.
>>> F = QQ.old_poly_ring(x).free_module(2)
>>> h = homomorphism(F, F, [[1, 0], [x, 0]])
>>> h.is_zero()
False
>>> h.restrict_domain(F.submodule()).is_zero()
True
>>> h.quotient_codomain(h.image()).is_zero()
True
kernel()
Compute the kernel of self.
That is, if self is the homomorphism ϕ : M → N , then compute ker(ϕ) = {x ∈ M |ϕ(x) =
0}. This is a submodule of M .
>>> F = QQ.old_poly_ring(x).free_module(2)
>>> homomorphism(F, F, [[1, 0], [x, 0]]).kernel()
<[x, -1]>
quotient_codomain(sm)
Return self with codomain replaced by codomain/sm.
Here sm must be a submodule of self.codomain.
>>> F = QQ.old_poly_ring(x).free_module(2)
>>> h = homomorphism(F, F, [[1, 0], [x, 0]])
>>> h
Matrix([
[1, x], : QQ[x]**2 -> QQ[x]**2
[0, 0]])
>>> h.quotient_codomain(F.submodule([1, 1]))
Matrix([
[1, x], : QQ[x]**2 -> QQ[x]**2/<[1, 1]>
[0, 0]])
This is the same as composing with the quotient map on the left:
quotient_domain(sm)
Return self with domain replaced by domain/sm.
Here sm must be a submodule of self.kernel().
>>> from sympy import QQ
>>> from sympy.abc import x
>>> from sympy.polys.agca import homomorphism
>>> F = QQ.old_poly_ring(x).free_module(2)
>>> h = homomorphism(F, F, [[1, 0], [x, 0]])
>>> h
Matrix([
[1, x], : QQ[x]**2 -> QQ[x]**2
[0, 0]])
>>> h.quotient_domain(F.submodule([-x, 1]))
Matrix([
[1, x], : QQ[x]**2/<[-x, 1]> -> QQ[x]**2
[0, 0]])
restrict_codomain(sm)
Return self, with codomain restricted to to sm.
Here sm has to be a submodule of self.codomain containing the image.
>>> from sympy import QQ
>>> from sympy.abc import x
>>> from sympy.polys.agca import homomorphism
>>> F = QQ.old_poly_ring(x).free_module(2)
>>> h = homomorphism(F, F, [[1, 0], [x, 0]])
>>> h
Matrix([
[1, x], : QQ[x]**2 -> QQ[x]**2
[0, 0]])
>>> h.restrict_codomain(F.submodule([1, 0]))
Matrix([
[1, x], : QQ[x]**2 -> <[1, 0]>
[0, 0]])
restrict_domain(sm)
Return self, with the domain restricted to sm.
Here sm has to be a submodule of self.domain.
>>> from sympy import QQ
>>> from sympy.abc import x
>>> from sympy.polys.agca import homomorphism
>>> F = QQ.old_poly_ring(x).free_module(2)
>>> h = homomorphism(F, F, [[1, 0], [x, 0]])
>>> h
(continues on next page)
This is the same as just composing on the right with the submodule inclusion:
Domains
Here we document the various implemented ground domains. There are three types: abstract
domains, concrete domains, and “implementation domains”. Abstract domains cannot be
(usefully) instantiated at all, and just collect together functionality shared by many other
domains. Concrete domains are those meant to be instantiated and used in the polynomial
manipulation algorithms. In some cases, there are various possible ways to implement the
data type the domain provides. For example, depending on what libraries are available on the
system, the integers are implemented either using the python built-in integers, or using gmpy.
Note that various aliases are created automatically depending on the libraries available. As
such e.g. ZZ always refers to the most efficient implementation of the integer ring available.
Abstract Domains
class sympy.polys.domains.domain.Domain
Represents an abstract domain.
Attributes
alias
dtype
one
rep
zero
abs(a)
Absolute value of a, implies __abs__.
add(a, b)
Sum of a and b, implies __add__.
algebraic_field(*extension)
Returns an algebraic field, i.e. K(α, . . .).
almosteq(a, b, tolerance=None)
Check if a and b are almost equal.
characteristic()
Return the characteristic of this domain.
cofactors(a, b)
Returns GCD and cofactors of a and b.
convert(element, base=None)
Convert element to self.dtype.
convert_from(element, base)
Convert element to self.dtype given the base domain.
denom(a)
Returns denominator of a.
div(a, b)
Division of a and b, implies something.
evalf(a, prec=None, **options)
Returns numerical approximation of a.
exquo(a, b)
Exact quotient of a and b, implies something.
frac_field(*symbols, **kwargs)
Returns a fraction field, i.e. K(X).
from_AlgebraicField(a, K0)
Convert an algebraic number to dtype.
from_ComplexField(a, K0)
Convert a complex element to dtype.
from_ExpressionDomain(a, K0)
Convert a EX object to dtype.
from_FF_gmpy(a, K0)
Convert ModularInteger(mpz) to dtype.
from_FF_python(a, K0)
Convert ModularInteger(int) to dtype.
from_FractionField(a, K0)
Convert a rational function to dtype.
from_GlobalPolynomialRing(a, K0)
Convert a polynomial to dtype.
from_PolynomialRing(a, K0)
Convert a polynomial to dtype.
from_QQ_gmpy(a, K0)
Convert a GMPY mpq object to dtype.
from_QQ_python(a, K0)
Convert a Python Fraction object to dtype.
from_RealField(a, K0)
Convert a real element object to dtype.
from_ZZ_gmpy(a, K0)
Convert a GMPY mpz object to dtype.
from_ZZ_python(a, K0)
Convert a Python int object to dtype.
from_sympy(a)
Convert a SymPy object to dtype.
gcd(a, b)
Returns GCD of a and b.
gcdex(a, b)
Extended GCD of a and b.
get_exact()
Returns an exact domain associated with self.
get_field()
Returns a field associated with self.
get_ring()
Returns a ring associated with self.
half_gcdex(a, b)
Half extended GCD of a and b.
inject(*symbols)
Inject generators into this domain.
invert(a, b)
Returns inversion of a mod b, implies something.
is_negative(a)
Returns True if a is negative.
is_nonnegative(a)
Returns True if a is non-negative.
is_nonpositive(a)
Returns True if a is non-positive.
is_one(a)
Returns True if a is one.
is_positive(a)
Returns True if a is positive.
is_zero(a)
Returns True if a is zero.
lcm(a, b)
Returns LCM of a and b.
log(a, b)
Returns b-base logarithm of a.
map(seq)
Rersively apply self to all elements of seq.
mul(a, b)
Product of a and b, implies __mul__.
n(a, prec=None, **options)
Returns numerical approximation of a.
neg(a)
Returns a negated, implies __neg__.
numer(a)
Returns numerator of a.
of_type(element)
Check if a is of type dtype.
old_frac_field(*symbols, **kwargs)
Returns a fraction field, i.e. K(X).
old_poly_ring(*symbols, **kwargs)
Returns a polynomial ring, i.e. K[X].
poly_ring(*symbols, **kwargs)
Returns a polynomial ring, i.e. K[X].
pos(a)
Returns a positive, implies __pos__.
pow(a, b)
Raise a to power b, implies __pow__.
quo(a, b)
Quotient of a and b, implies something.
rem(a, b)
Remainder of a and b, implies __mod__.
revert(a)
Returns a**(-1) if possible.
sqrt(a)
Returns square root of a.
sub(a, b)
Difference of a and b, implies __sub__.
to_sympy(a)
Convert a to a SymPy object.
unify(K1, symbols=None)
Construct a minimal domain that contains elements of K0 and K1.
Known domains (from smallest to largest):
• GF(p)
• ZZ
• QQ
• RR(prec, tol)
• CC(prec, tol)
• ALG(a, b, c)
• K[x, y, z]
• K(x, y, z)
• EX
class sympy.polys.domains.field.Field
Represents a field domain.
Attributes
alias
dtype
one
rep
zero
div(a, b)
Division of a and b, implies __div__.
exquo(a, b)
Exact quotient of a and b, implies __div__.
gcd(a, b)
Returns GCD of a and b.
This definition of GCD over fields allows to clear denominators in primitive().
get_field()
Returns a field associated with self.
get_ring()
Returns a ring associated with self.
lcm(a, b)
Returns LCM of a and b.
>>> from sympy.polys.domains import QQ
>>> from sympy import S, lcm
quo(a, b)
Quotient of a and b, implies __div__.
rem(a, b)
Remainder of a and b, implies nothing.
revert(a)
Returns a**(-1) if possible.
class sympy.polys.domains.ring.Ring
Represents a ring domain.
Attributes
alias
dtype
one
rep
zero
denom(a)
Returns denominator of a.
div(a, b)
Division of a and b, implies __divmod__.
exquo(a, b)
Exact quotient of a and b, implies __floordiv__.
free_module(rank)
Generate a free module of rank rank over self.
get_ring()
Returns a ring associated with self.
ideal(*gens)
Generate an ideal of self.
invert(a, b)
Returns inversion of a mod b.
numer(a)
Returns numerator of a.
quo(a, b)
Quotient of a and b, implies __floordiv__.
quotient_ring(e)
Form a quotient ring of self.
Here e can be an ideal or an iterable.
>>> QQ.old_poly_ring(x)/[x**2]
QQ[x]/<x**2>
rem(a, b)
Remainder of a and b, implies __mod__.
revert(a)
Returns a**(-1) if possible.
class sympy.polys.domains.simpledomain.SimpleDomain
Base class for simple domains, e.g. ZZ, QQ.
Attributes
alias
dtype
one
rep
zero
inject(*gens)
Inject generators into this domain.
class sympy.polys.domains.compositedomain.CompositeDomain
Base class for composite domains, e.g. ZZ[x], ZZ(X).
Attributes
alias
domain
dtype
gens
ngens
one
rep
symbols
zero
inject(*symbols)
Inject generators into this domain.
Concrete Domains
Attributes
alias
dom
dtype
mod
one
zero
characteristic()
Return the characteristic of this domain.
from_FF_gmpy(a, K0=None)
Convert ModularInteger(mpz) to dtype.
from_FF_python(a, K0=None)
Convert ModularInteger(int) to dtype.
from_QQ_gmpy(a, K0=None)
Convert GMPY’s mpq to dtype.
from_QQ_python(a, K0=None)
Convert Python’s Fraction to dtype.
from_RealField(a, K0)
Convert mpmath’s mpf to dtype.
from_ZZ_gmpy(a, K0=None)
Convert GMPY’s mpz to dtype.
from_ZZ_python(a, K0=None)
Convert Python’s int to dtype.
from_sympy(a)
Convert SymPy’s Integer to SymPy’s Integer.
get_field()
Returns a field associated with self.
to_sympy(a)
Convert a to a SymPy object.
class sympy.polys.domains.IntegerRing
General class for integer rings.
Attributes
alias
dtype
one
zero
algebraic_field(*extension)
Returns an algebraic field, i.e. Q(α, . . .).
from_AlgebraicField(a, K0)
Convert a ANP object to dtype.
get_field()
Returns a field associated with self.
log(a, b)
Returns b-base logarithm of a.
class sympy.polys.domains.PolynomialRing(domain_or_ring, symbols=None, or-
der=None)
A class for representing multivariate polynomial rings.
Attributes
alias
domain
dtype
gens
ngens
rep
symbols
factorial(a)
Returns factorial of a.
from_AlgebraicField(a, K0)
Convert an algebraic number to dtype.
from_FractionField(a, K0)
Convert a rational function to dtype.
from_PolynomialRing(a, K0)
Convert a polynomial to dtype.
from_QQ_gmpy(a, K0)
Convert a GMPY mpq object to dtype.
from_QQ_python(a, K0)
Convert a Python F raction object to dtype.
from_RealField(a, K0)
Convert a mpmath mpf object to dtype.
from_ZZ_gmpy(a, K0)
Convert a GMPY mpz object to dtype.
from_ZZ_python(a, K0)
Convert a Python int object to dtype.
from_sympy(a)
Convert SymPy’s expression to dtype.
gcd(a, b)
Returns GCD of a and b.
gcdex(a, b)
Extended GCD of a and b.
get_field()
Returns a field associated with self .
is_negative(a)
Returns True if LC(a) is negative.
is_nonnegative(a)
Returns True if LC(a) is non-negative.
is_nonpositive(a)
Returns True if LC(a) is non-positive.
is_positive(a)
Returns True if LC(a) is positive.
lcm(a, b)
Returns LCM of a and b.
to_sympy(a)
Convert a to a SymPy object.
class sympy.polys.domains.RationalField
General class for rational fields.
Attributes
alias
dtype
one
zero
algebraic_field(*extension)
Returns an algebraic field, i.e. Q(α, . . .).
from_AlgebraicField(a, K0)
Convert a ANP object to dtype.
class sympy.polys.domains.AlgebraicField(dom, *ext)
A class for representing algebraic number fields.
Attributes
alias
one
rep
zero
algebraic_field(*extension)
Returns an algebraic field, i.e. Q(α, . . .).
denom(a)
Returns denominator of a.
dtype
alias of sympy.polys.polyclasses.ANP (page 951)
from_QQ_gmpy(a, K0)
Convert a GMPY mpq object to dtype.
from_QQ_python(a, K0)
Convert a Python Fraction object to dtype.
from_RealField(a, K0)
Convert a mpmath mpf object to dtype.
from_ZZ_gmpy(a, K0)
Convert a GMPY mpz object to dtype.
from_ZZ_python(a, K0)
Convert a Python int object to dtype.
from_sympy(a)
Convert SymPy’s expression to dtype.
get_ring()
Returns a ring associated with self.
is_negative(a)
Returns True if a is negative.
is_nonnegative(a)
Returns True if a is non-negative.
is_nonpositive(a)
Returns True if a is non-positive.
is_positive(a)
Returns True if a is positive.
numer(a)
Returns numerator of a.
to_sympy(a)
Convert a to a SymPy object.
class sympy.polys.domains.FractionField(domain_or_field, symbols=None, or-
der=None)
A class for representing multivariate rational function fields.
Attributes
alias
domain
dtype
gens
ngens
rep
symbols
denom(a)
Returns denominator of a.
factorial(a)
Returns factorial of a.
from_AlgebraicField(a, K0)
Convert an algebraic number to dtype.
from_FractionField(a, K0)
Convert a rational function to dtype.
from_PolynomialRing(a, K0)
Convert a polynomial to dtype.
from_QQ_gmpy(a, K0)
Convert a GMPY mpq object to dtype.
from_QQ_python(a, K0)
Convert a Python F raction object to dtype.
from_RealField(a, K0)
Convert a mpmath mpf object to dtype.
from_ZZ_gmpy(a, K0)
Convert a GMPY mpz object to dtype.
from_ZZ_python(a, K0)
Convert a Python int object to dtype.
from_sympy(a)
Convert SymPy’s expression to dtype.
get_ring()
Returns a field associated with self .
is_negative(a)
Returns True if LC(a) is negative.
is_nonnegative(a)
Returns True if LC(a) is non-negative.
is_nonpositive(a)
Returns True if LC(a) is non-positive.
is_positive(a)
Returns True if LC(a) is positive.
numer(a)
Returns numerator of a.
to_sympy(a)
Convert a to a SymPy object.
class sympy.polys.domains.RealField(prec=53, dps=None, tol=None)
Real numbers up to the given precision.
Attributes
alias
dtype
one
zero
almosteq(a, b, tolerance=None)
Check if a and b are almost equal.
from_sympy(expr)
Convert SymPy’s number to dtype.
gcd(a, b)
Returns GCD of a and b.
get_exact()
Returns an exact domain associated with self.
get_ring()
Returns a ring associated with self.
lcm(a, b)
Returns LCM of a and b.
to_rational(element, limit=True)
Convert a real number to rational number.
to_sympy(element)
Convert element to SymPy number.
class sympy.polys.domains.ExpressionDomain
A class for arbitrary expressions.
Attributes
alias
class Expression(ex)
An arbitrary expression.
denom(a)
Returns denominator of a.
dtype
alias of ExpressionDomain.Expression (page 943)
from_ExpressionDomain(a, K0)
Convert a EX object to dtype.
from_FractionField(a, K0)
Convert a DMF object to dtype.
from_PolynomialRing(a, K0)
Convert a DMP object to dtype.
from_QQ_gmpy(a, K0)
Convert a GMPY mpq object to dtype.
from_QQ_python(a, K0)
Convert a Python Fraction object to dtype.
from_RealField(a, K0)
Convert a mpmath mpf object to dtype.
from_ZZ_gmpy(a, K0)
Convert a GMPY mpz object to dtype.
from_ZZ_python(a, K0)
Convert a Python int object to dtype.
from_sympy(a)
Convert SymPy’s expression to dtype.
get_field()
Returns a field associated with self.
get_ring()
Returns a ring associated with self.
is_negative(a)
Returns True if a is negative.
is_nonnegative(a)
Returns True if a is non-negative.
is_nonpositive(a)
Returns True if a is non-positive.
is_positive(a)
Returns True if a is positive.
numer(a)
Returns numerator of a.
to_sympy(a)
Convert a to a SymPy object.
Implementation Domains
Attributes
dom
dtype
mod
one
zero
Attributes
dom
dtype
mod
one
zero
class sympy.polys.domains.PythonIntegerRing
Integer ring based on Python’s int type.
class sympy.polys.domains.GMPYIntegerRing
Integer ring based on GMPY’s mpz type.
class sympy.polys.domains.PythonRationalField
Rational field based on Python rational number type.
class sympy.polys.domains.GMPYRationalField
Rational field based on GMPY mpq class.
Level One
degree(j=0)
Returns the leading degree of f in x_j.
degree_list()
Returns a list of degrees of f.
diff(m=1, j=0)
Computes the m-th order derivative of f in x_j.
discriminant()
Computes discriminant of f.
div(g)
Polynomial division with remainder of f and g.
eject(dom, front=False)
Eject selected generators into the ground domain.
eval(a, j=0)
Evaluates f at the given point a in x_j.
exclude()
Remove useless generators from f.
Returns the removed generators and the new excluded f.
Examples
exquo(g)
Computes polynomial exact quotient of f and g.
exquo_ground(c)
Exact quotient of f by a an element of the ground domain.
factor_list()
Returns a list of irreducible factors of f.
factor_list_include()
Returns a list of irreducible factors of f.
classmethod from_dict(rep, lev, dom)
Construct and instance of cls from a dict representation.
classmethod from_list(rep, lev, dom)
Create an instance of cls given a list of native coefficients.
classmethod from_sympy_list(rep, lev, dom)
Create an instance of cls given a list of SymPy coefficients.
gcd(g)
Returns polynomial GCD of f and g.
gcdex(g)
Extended Euclidean algorithm, if univariate.
gff_list()
Computes greatest factorial factorization of f.
half_gcdex(g)
Half extended Euclidean algorithm, if univariate.
homogeneous_order()
Returns the homogeneous order of f.
homogenize(s)
Return homogeneous polynomial of f
inject(front=False)
Inject ground domain generators into f.
integrate(m=1, j=0)
Computes the m-th order indefinite integral of f in x_j.
intervals(all=False, eps=None, inf=None, sup=None, fast=False, sqf=False)
Compute isolating intervals for roots of f.
invert(g)
Invert f modulo g, if possible.
is_cyclotomic
Returns True if f is a cyclotomic polynomial.
is_ground
Returns True if f is an element of the ground domain.
is_homogeneous
Returns True if f is a homogeneous polynomial.
is_irreducible
Returns True if f has no factors over its domain.
is_linear
Returns True if f is linear in all its variables.
is_monic
Returns True if the leading coefficient of f is one.
is_monomial
Returns True if f is zero or has only one term.
is_one
Returns True if f is a unit polynomial.
is_primitive
Returns True if the GCD of the coefficients of f is one.
is_quadratic
Returns True if f is quadratic in all its variables.
is_sqf
Returns True if f is a square-free polynomial.
is_zero
Returns True if f is a zero polynomial.
l1_norm()
Returns l1 norm of f.
lcm(g)
Returns polynomial LCM of f and g.
lift()
Convert algebraic coefficients to rationals.
max_norm()
Returns maximum norm of f.
monic()
Divides all coefficients by LC(f).
monoms(order=None)
Returns all non-zero monomials from f in lex order.
mul(g)
Multiply two multivariate polynomials f and g.
mul_ground(c)
Multiply f by a an element of the ground domain.
neg()
Negate all coefficients in f.
norm()
Computes Norm(f).
nth(*N)
Returns the n-th coefficient of f.
pdiv(g)
Polynomial pseudo-division of f and g.
per(rep, dom=None, kill=False, ring=None)
Create a DMP out of the given representation.
permute(P)
Returns a polynomial in K[xP (1) , ..., xP (n) ].
Examples
pexquo(g)
Polynomial exact pseudo-quotient of f and g.
pow(n)
Raise f to a non-negative power n.
pquo(g)
Polynomial pseudo-quotient of f and g.
prem(g)
Polynomial pseudo-remainder of f and g.
primitive()
Returns content and a primitive form of f.
quo(g)
Computes polynomial quotient of f and g.
quo_ground(c)
Quotient of f by a an element of the ground domain.
refine_root(s, t, eps=None, steps=None, fast=False)
Refine an isolating interval to the given precision.
eps should be a rational number.
rem(g)
Computes polynomial remainder of f and g.
resultant(g, includePRS=False)
Computes resultant of f and g via PRS.
revert(n)
Compute f**(-1) mod x**n.
shift(a)
Efficiently compute Taylor shift f(x + a).
slice(m, n, j=0)
Take a continuous subsequence of terms of f.
sqf_list(all=False)
Returns a list of square-free factors of f.
sqf_list_include(all=False)
Returns a list of square-free factors of f.
sqf_norm()
Computes square-free norm of f.
sqf_part()
Computes square-free part of f.
sqr()
Square a multivariate polynomial f.
sturm()
Computes the Sturm sequence of f.
sub(g)
Subtract two multivariate polynomials f and g.
sub_ground(c)
Subtract an element of the ground domain from f.
subresultants(g)
Computes subresultant PRS sequence of f and g.
terms(order=None)
Returns all non-zero terms from f in lex order.
terms_gcd()
Remove GCD of terms from the polynomial f.
to_dict(zero=False)
Convert f to a dict representation with native coefficients.
to_exact()
Make the ground domain exact.
to_field()
Make the ground domain a field.
to_list()
Convert f to a list representation with native coefficients.
to_ring()
Make the ground domain a ring.
to_sympy_dict(zero=False)
Convert f to a dict representation with SymPy coefficients.
to_sympy_list()
Convert f to a list representation with SymPy coefficients.
to_tuple()
Convert f to a tuple representation with native coefficients.
This is needed for hashing.
total_degree()
Returns the total degree of f.
transform(p, q)
Evaluate functional transformation q**n * f(p/q).
trunc(p)
Reduce f modulo a constant p.
unify(g)
Unify representations of two multivariate polynomials.
class sympy.polys.polyclasses.DMF(rep, dom, lev=None, ring=None)
Dense Multivariate Fractions over K.
add(g)
Add two multivariate fractions f and g.
cancel()
Remove common factors from f.num and f.den.
denom()
Returns the denominator of f.
exquo(g)
Computes quotient of fractions f and g.
frac_unify(g)
Unify representations of two multivariate fractions.
half_per(rep, kill=False)
Create a DMP out of the given representation.
invert(check=True)
Computes inverse of a fraction f.
is_one
Returns True if f is a unit fraction.
is_zero
Returns True if f is a zero fraction.
mul(g)
Multiply two multivariate fractions f and g.
neg()
Negate all coefficients in f.
numer()
Returns the numerator of f.
per(num, den, cancel=True, kill=False, ring=None)
Create a DMF out of the given representation.
poly_unify(g)
Unify a multivariate fraction and a polynomial.
pow(n)
Raise f to a non-negative power n.
quo(g)
Computes quotient of fractions f and g.
sub(g)
Subtract two multivariate fractions f and g.
class sympy.polys.polyclasses.ANP(rep, mod, dom)
Dense Algebraic Number Polynomials over a field.
LC()
Returns the leading coefficient of f.
TC()
Returns the trailing coefficient of f.
is_ground
Returns True if f is an element of the ground domain.
is_one
Returns True if f is a unit algebraic number.
is_zero
Returns True if f is a zero algebraic number.
pow(n)
Raise f to a non-negative power n.
to_dict()
Convert f to a dict representation with native coefficients.
to_list()
Convert f to a list representation with native coefficients.
to_sympy_dict()
Convert f to a dict representation with SymPy coefficients.
to_sympy_list()
Convert f to a list representation with SymPy coefficients.
to_tuple()
Convert f to a tuple representation with native coefficients.
This is needed for hashing.
unify(g)
Unify representations of two algebraic numbers.
Level Zero
Level zero contains the bulk code of the polynomial manipulation module.
These functions can be used to manipulate polynomials in K[X0 , . . . , Xu ]. Functions for manip-
ulating multivariate polynomials in the dense representation have the prefix dmp_. Functions
which only apply to univariate polynomials (i.e. u = 0) have the prefix dup__. The ground do-
main K has to be passed explicitly. For many multivariate polynomial manipulation functions
also the level u, i.e. the number of generators minus one, has to be passed. (Note that, in
many cases, dup_ versions of functions are available, which may be slightly more efficient.)
Basic manipulation:
sympy.polys.densebasic.dmp_LC(f, K)
Return leading coefficient of f.
Examples
sympy.polys.densebasic.dmp_TC(f, K)
Return trailing coefficient of f.
Examples
sympy.polys.densebasic.dmp_ground_LC(f, u, K)
Return the ground leading coefficient.
Examples
sympy.polys.densebasic.dmp_ground_TC(f, u, K)
Return the ground trailing coefficient.
Examples
sympy.polys.densebasic.dmp_true_LT(f, u, K)
Return the leading term c * x_1**n_1 ... x_k**n_k.
Examples
sympy.polys.densebasic.dmp_degree(f, u)
Return the leading degree of f in x_0 in K[X].
Note that the degree of 0 is negative infinity (the SymPy object -oo).
Examples
>>> dmp_degree([[[]]], 2)
-oo
>>> dmp_degree(f, 1)
1
sympy.polys.densebasic.dmp_degree_in(f, j, u)
Return the leading degree of f in x_j in K[X].
Examples
>>> dmp_degree_in(f, 0, 1)
1
>>> dmp_degree_in(f, 1, 1)
2
sympy.polys.densebasic.dmp_degree_list(f, u)
Return a list of degrees of f in K[X].
Examples
>>> dmp_degree_list(f, 1)
(1, 2)
sympy.polys.densebasic.dmp_strip(f, u)
Remove leading zeros from f in K[X].
Examples
sympy.polys.densebasic.dmp_validate(f, K=None)
Return the number of levels in f and recursively strip it.
Examples
sympy.polys.densebasic.dup_reverse(f)
Compute x**n * f(1/x), i.e.: reverse f in K[x].
Examples
>>> dup_reverse(f)
[3, 2, 1]
sympy.polys.densebasic.dmp_copy(f, u)
Create a new copy of a polynomial f in K[X].
Examples
>>> dmp_copy(f, 1)
[[1], [1, 2]]
sympy.polys.densebasic.dmp_to_tuple(f, u)
Convert f into a nested tuple of tuples.
This is needed for hashing. This is similar to dmp_copy().
Examples
>>> dmp_to_tuple(f, 1)
((1,), (1, 2))
sympy.polys.densebasic.dmp_normal(f, u, K)
Normalize a multivariate polynomial in the given domain.
Examples
Examples
sympy.polys.densebasic.dmp_from_sympy(f, u, K)
Convert the ground domain of f from SymPy to K.
Examples
sympy.polys.densebasic.dmp_nth(f, n, u, K)
Return the n-th coefficient of f in K[x].
Examples
sympy.polys.densebasic.dmp_ground_nth(f, N, u, K)
Return the ground n-th coefficient of f in K[x].
Examples
sympy.polys.densebasic.dmp_zero_p(f, u)
Return True if f is zero in K[X].
Examples
>>> dmp_zero_p([[[[[]]]]], 4)
True
>>> dmp_zero_p([[[[[1]]]]], 4)
False
sympy.polys.densebasic.dmp_zero(u)
Return a multivariate zero.
Examples
>>> dmp_zero(4)
[[[[[]]]]]
sympy.polys.densebasic.dmp_one_p(f, u, K)
Return True if f is one in K[X].
Examples
sympy.polys.densebasic.dmp_one(u, K)
Return a multivariate one over K.
Examples
sympy.polys.densebasic.dmp_ground_p(f, c, u)
Return True if f is constant in K[X].
Examples
>>> dmp_ground_p([[[3]]], 3, 2)
True
>>> dmp_ground_p([[[4]]], None, 2)
True
sympy.polys.densebasic.dmp_ground(c, u)
Return a multivariate constant.
Examples
>>> dmp_ground(3, 5)
[[[[[[3]]]]]]
>>> dmp_ground(1, -1)
1
sympy.polys.densebasic.dmp_zeros(n, u, K)
Return a list of multivariate zeros.
Examples
sympy.polys.densebasic.dmp_grounds(c, n, u)
Return a list of multivariate constants.
Examples
>>> dmp_grounds(ZZ(4), 3, 2)
[[[[4]]], [[[4]]], [[[4]]]]
>>> dmp_grounds(ZZ(4), 3, -1)
[4, 4, 4]
sympy.polys.densebasic.dmp_negative_p(f, u, K)
Return True if LC(f) is negative.
Examples
sympy.polys.densebasic.dmp_positive_p(f, u, K)
Return True if LC(f) is positive.
Examples
sympy.polys.densebasic.dmp_from_dict(f, u, K)
Create a K[X] polynomial from a dict.
Examples
>>> dmp_from_dict({(0, 0): ZZ(3), (0, 1): ZZ(2), (2, 1): ZZ(1)}, 1, ZZ)
[[1, 0], [], [2, 3]]
>>> dmp_from_dict({}, 0, ZZ)
[]
Examples
sympy.polys.densebasic.dmp_swap(f, i, j, u, K)
Transform K[..x_i..x_j..] to K[..x_j..x_i..].
Examples
sympy.polys.densebasic.dmp_permute(f, P, u, K)
Return a polynomial in K[x_{P(1)},..,x_{P(n)}].
Examples
sympy.polys.densebasic.dmp_nest(f, l, K)
Return a multivariate value nested l-levels.
Examples
sympy.polys.densebasic.dmp_raise(f, l, u, K)
Return a multivariate polynomial raised l-levels.
Examples
sympy.polys.densebasic.dmp_deflate(f, u, K)
Map x_i**m_i to y_i in a polynomial in K[X].
Examples
sympy.polys.densebasic.dmp_multi_deflate(polys, u, K)
Map x_i**m_i to y_i in a set of polynomials in K[X].
Examples
sympy.polys.densebasic.dmp_inflate(f, M, u, K)
Map y_i to x_i**k_i in a polynomial in K[X].
Examples
sympy.polys.densebasic.dmp_exclude(f, u, K)
Exclude useless levels from f.
Return the levels excluded, the new excluded f, and the new u.
Examples
sympy.polys.densebasic.dmp_include(f, J, u, K)
Include useless levels in f.
Examples
sympy.polys.densebasic.dmp_inject(f, u, K, front=False)
Convert f from K[X][Y] to K[X,Y].
Examples
sympy.polys.densebasic.dmp_eject(f, u, K, front=False)
Convert f from K[X,Y] to K[X][Y].
Examples
sympy.polys.densebasic.dmp_terms_gcd(f, u, K)
Remove GCD of terms from f in K[X].
Examples
sympy.polys.densebasic.dmp_list_terms(f, u, K, order=None)
List all non-zero terms from f in the given order order.
Examples
sympy.polys.densebasic.dmp_apply_pairs(f, g, h, args, u, K)
Apply h to pairs of coefficients of f and g.
Examples
sympy.polys.densebasic.dmp_slice(f, m, n, u, K)
Take a continuous subsequence of terms of f in K[X].
sympy.polys.densebasic.dup_random(n, a, b, K)
Return a polynomial of degree n with coefficients in [a, b].
Examples
Arithmetic operations:
sympy.polys.densearith.dmp_add_term(f, c, i, u, K)
Add c(x_2..x_u)*x_0**i to f in K[X].
Examples
>>> R.dmp_add_term(x*y + 1, 2, 2)
2*x**2 + x*y + 1
sympy.polys.densearith.dmp_sub_term(f, c, i, u, K)
Subtract c(x_2..x_u)*x_0**i from f in K[X].
Examples
sympy.polys.densearith.dmp_mul_term(f, c, i, u, K)
Multiply f by c(x_2..x_u)*x_0**i in K[X].
Examples
sympy.polys.densearith.dmp_add_ground(f, c, u, K)
Add an element of the ground domain to f.
Examples
sympy.polys.densearith.dmp_sub_ground(f, c, u, K)
Subtract an element of the ground domain from f.
Examples
sympy.polys.densearith.dmp_mul_ground(f, c, u, K)
Multiply f by a constant value in K[X].
Examples
sympy.polys.densearith.dmp_quo_ground(f, c, u, K)
Quotient by a constant in K[X].
Examples
sympy.polys.densearith.dmp_exquo_ground(f, c, u, K)
Exact quotient by a constant in K[X].
Examples
sympy.polys.densearith.dup_lshift(f, n, K)
Efficiently multiply f by x**n in K[x].
Examples
>>> R.dup_lshift(x**2 + 1, 2)
x**4 + x**2
sympy.polys.densearith.dup_rshift(f, n, K)
Efficiently divide f by x**n in K[x].
Examples
sympy.polys.densearith.dmp_abs(f, u, K)
Make all coefficients positive in K[X].
Examples
>>> R.dmp_abs(x**2*y - x)
x**2*y + x
sympy.polys.densearith.dmp_neg(f, u, K)
Negate a polynomial in K[X].
Examples
>>> R.dmp_neg(x**2*y - x)
-x**2*y + x
sympy.polys.densearith.dmp_add(f, g, u, K)
Add dense polynomials in K[X].
Examples
sympy.polys.densearith.dmp_sub(f, g, u, K)
Subtract dense polynomials in K[X].
Examples
sympy.polys.densearith.dmp_add_mul(f, g, h, u, K)
Returns f + g*h where f, g, h are in K[X].
Examples
>>> R.dmp_add_mul(x**2 + y, x, x + 2)
2*x**2 + 2*x + y
sympy.polys.densearith.dmp_sub_mul(f, g, h, u, K)
Returns f - g*h where f, g, h are in K[X].
Examples
>>> R.dmp_sub_mul(x**2 + y, x, x + 2)
-2*x + y
sympy.polys.densearith.dmp_mul(f, g, u, K)
Multiply dense polynomials in K[X].
Examples
>>> R.dmp_mul(x*y + 1, x)
x**2*y + x
sympy.polys.densearith.dmp_sqr(f, u, K)
Square dense polynomials in K[X].
Examples
sympy.polys.densearith.dmp_pow(f, n, u, K)
Raise f to the n-th power in K[X].
Examples
>>> R.dmp_pow(x*y + 1, 3)
x**3*y**3 + 3*x**2*y**2 + 3*x*y + 1
sympy.polys.densearith.dmp_pdiv(f, g, u, K)
Polynomial pseudo-division in K[X].
Examples
sympy.polys.densearith.dmp_prem(f, g, u, K)
Polynomial pseudo-remainder in K[X].
Examples
sympy.polys.densearith.dmp_pquo(f, g, u, K)
Polynomial exact pseudo-quotient in K[X].
Examples
>>> R.dmp_pquo(f, g)
2*x
>>> R.dmp_pquo(f, h)
2*x + 2*y - 2
sympy.polys.densearith.dmp_pexquo(f, g, u, K)
Polynomial pseudo-quotient in K[X].
Examples
>>> R.dmp_pexquo(f, g)
2*x
>>> R.dmp_pexquo(f, h)
Traceback (most recent call last):
...
ExactQuotientFailed: [[2], [2]] does not divide [[1], [1, 0], []]
sympy.polys.densearith.dmp_rr_div(f, g, u, K)
Multivariate division with remainder over a ring.
Examples
sympy.polys.densearith.dmp_ff_div(f, g, u, K)
Polynomial division with remainder over a field.
Examples
sympy.polys.densearith.dmp_div(f, g, u, K)
Polynomial division with remainder in K[X].
Examples
sympy.polys.densearith.dmp_rem(f, g, u, K)
Returns polynomial remainder in K[X].
Examples
sympy.polys.densearith.dmp_quo(f, g, u, K)
Returns exact polynomial quotient in K[X].
Examples
sympy.polys.densearith.dmp_exquo(f, g, u, K)
Returns polynomial quotient in K[X].
Examples
>>> R.dmp_exquo(f, g)
x
>>> R.dmp_exquo(f, h)
Traceback (most recent call last):
...
ExactQuotientFailed: [[2], [2]] does not divide [[1], [1, 0], []]
sympy.polys.densearith.dmp_max_norm(f, u, K)
Returns maximum norm of a polynomial in K[X].
Examples
>>> R.dmp_max_norm(2*x*y - x - 3)
3
sympy.polys.densearith.dmp_l1_norm(f, u, K)
Returns l1 norm of a polynomial in K[X].
Examples
>>> R.dmp_l1_norm(2*x*y - x - 3)
6
sympy.polys.densearith.dmp_expand(polys, u, K)
Multiply together several polynomials in K[X].
Examples
Further tools:
sympy.polys.densetools.dmp_integrate(f, m, u, K)
Computes the indefinite integral of f in x_0 in K[X].
Examples
sympy.polys.densetools.dmp_integrate_in(f, m, j, u, K)
Computes the indefinite integral of f in x_j in K[X].
Examples
sympy.polys.densetools.dmp_diff(f, m, u, K)
m-th order derivative in x_0 of a polynomial in K[X].
Examples
>>> R.dmp_diff(f, 1)
y**2 + 2*y + 3
>>> R.dmp_diff(f, 2)
0
sympy.polys.densetools.dmp_diff_in(f, m, j, u, K)
m-th order derivative in x_j of a polynomial in K[X].
Examples
>>> R.dmp_diff_in(f, 1, 0)
y**2 + 2*y + 3
>>> R.dmp_diff_in(f, 1, 1)
2*x*y + 2*x + 4*y + 3
sympy.polys.densetools.dmp_eval(f, a, u, K)
Evaluate a polynomial at x_0 = a in K[X] using the Horner scheme.
Examples
sympy.polys.densetools.dmp_eval_in(f, a, j, u, K)
Evaluate a polynomial at x_j = a in K[X] using the Horner scheme.
Examples
>>> R.dmp_eval_in(f, 2, 0)
5*y + 8
>>> R.dmp_eval_in(f, 2, 1)
7*x + 4
sympy.polys.densetools.dmp_eval_tail(f, A, u, K)
Evaluate a polynomial at x_j = a_j, ... in K[X].
Examples
sympy.polys.densetools.dmp_diff_eval_in(f, m, a, j, u, K)
Differentiate and evaluate a polynomial in x_j at a in K[X].
Examples
>>> R.dmp_diff_eval_in(f, 1, 2, 0)
y**2 + 2*y + 3
>>> R.dmp_diff_eval_in(f, 1, 2, 1)
6*x + 11
sympy.polys.densetools.dmp_trunc(f, p, u, K)
Reduce a K[X] polynomial modulo a polynomial p in K[Y].
Examples
>>> R.dmp_trunc(f, g)
11*x**2 + 11*x + 5
sympy.polys.densetools.dmp_ground_trunc(f, p, u, K)
Reduce a K[X] polynomial modulo a constant p in K.
Examples
sympy.polys.densetools.dup_monic(f, K)
Divide all coefficients by LC(f) in K[x].
Examples
sympy.polys.densetools.dmp_ground_monic(f, u, K)
Divide all coefficients by LC(f) in K[X].
Examples
>>> R.dmp_ground_monic(f)
x**2*y + 2*x**2 + x*y + 3*y + 1
>>> R.dmp_ground_monic(f)
x**2*y + 8/3*x**2 + 5/3*x*y + 2*x + 2/3*y + 1
sympy.polys.densetools.dup_content(f, K)
Compute the GCD of coefficients of f in K[x].
Examples
>>> R.dup_content(f)
2
>>> R.dup_content(f)
2
sympy.polys.densetools.dmp_ground_content(f, u, K)
Compute the GCD of coefficients of f in K[X].
Examples
>>> R.dmp_ground_content(f)
2
>>> R.dmp_ground_content(f)
2
sympy.polys.densetools.dup_primitive(f, K)
Compute content and the primitive form of f in K[x].
Examples
>>> R.dup_primitive(f)
(2, 3*x**2 + 4*x + 6)
>>> R.dup_primitive(f)
(2, 3*x**2 + 4*x + 6)
sympy.polys.densetools.dmp_ground_primitive(f, u, K)
Compute content and the primitive form of f in K[X].
Examples
>>> R.dmp_ground_primitive(f)
(2, x*y + 3*x + 2*y + 6)
>>> R.dmp_ground_primitive(f)
(2, x*y + 3*x + 2*y + 6)
sympy.polys.densetools.dup_extract(f, g, K)
Extract common content from a pair of polynomials in K[x].
Examples
sympy.polys.densetools.dmp_ground_extract(f, g, u, K)
Extract common content from a pair of polynomials in K[X].
Examples
sympy.polys.densetools.dup_real_imag(f, K)
Return bivariate polynomials f1 and f2, such that f = f1 + f2*I.
Examples
sympy.polys.densetools.dup_mirror(f, K)
Evaluate efficiently the composition f(-x) in K[x].
Examples
sympy.polys.densetools.dup_scale(f, a, K)
Evaluate efficiently composition f(a*x) in K[x].
Examples
sympy.polys.densetools.dup_shift(f, a, K)
Evaluate efficiently Taylor shift f(x + a) in K[x].
Examples
sympy.polys.densetools.dup_transform(f, p, q, K)
Evaluate functional transformation q**n * f(p/q) in K[x].
Examples
sympy.polys.densetools.dmp_compose(f, g, u, K)
Evaluate functional composition f(g) in K[X].
Examples
sympy.polys.densetools.dup_decompose(f, K)
Computes functional decomposition of f in K[x].
Given a univariate polynomial f with coefficients in a field of characteristic zero, returns
list [f_1, f_2, ..., f_n], where:
and f_2, ..., f_n are monic and homogeneous polynomials of at least second degree.
Unlike factorization, complete functional decompositions of polynomials are not unique,
consider examples:
1. f o g = f(x + b) o (g - b)
2. x**n o x**m = x**m o x**n
3. T_n o T_m = T_m o T_n
where T_n and T_m are Chebyshev polynomials.
References
1. [Kozen89]
[Kozen89]
Examples
sympy.polys.densetools.dmp_lift(f, u, K)
Convert algebraic coefficients to integers in K[X].
Examples
>>> K = QQ.algebraic_field(I)
>>> R, x = ring(”x”, K)
>>> R.dmp_lift(f)
x**8 + 2*x**6 + 9*x**4 - 8*x**2 + 16
sympy.polys.densetools.dup_sign_variations(f, K)
Compute the number of sign variations of f in K[x].
Examples
Examples
sympy.polys.densetools.dmp_revert(f, g, u, K)
Compute f**(-1) mod x**n using Newton iteration.
Examples
Functions in this module carry the prefix gf_, referring to the classical name “Galois Fields”
for finite fields. Note that many polynomial factorization algorithms work by reduction to the
finite field case, so having special implementations for this case is justified both by perfor-
mance, and by the necessity of certain methods which do not even make sense over general
fields.
sympy.polys.galoistools.gf_crt(U, M, K=None)
Chinese Remainder Theorem.
Given a set of integer residues u_0,...,u_n and a set of co-prime integer moduli m_0,.
..,m_n, returns an integer u, such that u = u_i mod m_i for i = ``0,...,n.
As an example consider a set of residues U = [49, 76, 65] and a set of moduli M =
[99, 97, 95]. Then we have:
Examples
sympy.polys.galoistools.gf_crt2(U, M, p, E, S, K)
Second part of the Chinese Remainder Theorem.
Examples
sympy.polys.galoistools.gf_int(a, p)
Coerce a mod p to an integer in the range [-p/2, p/2].
Examples
>>> gf_int(2, 7)
2
>>> gf_int(5, 7)
-2
sympy.polys.galoistools.gf_degree(f)
Return the leading degree of f.
Examples
sympy.polys.galoistools.gf_LC(f, K)
Return the leading coefficient of f.
Examples
sympy.polys.galoistools.gf_TC(f, K)
Return the trailing coefficient of f.
Examples
sympy.polys.galoistools.gf_strip(f)
Remove leading zeros from f.
Examples
sympy.polys.galoistools.gf_trunc(f, p)
Reduce all coefficients modulo p.
Examples
sympy.polys.galoistools.gf_normal(f, p, K)
Normalize all coefficients in K.
Examples
sympy.polys.galoistools.gf_from_dict(f, p, K)
Create a GF(p)[x] polynomial from a dict.
Examples
sympy.polys.galoistools.gf_to_dict(f, p, symmetric=True)
Convert a GF(p)[x] polynomial to a dict.
Examples
sympy.polys.galoistools.gf_from_int_poly(f, p)
Create a GF(p)[x] polynomial from Z[x].
Examples
sympy.polys.galoistools.gf_to_int_poly(f, p, symmetric=True)
Convert a GF(p)[x] polynomial to Z[x].
Examples
sympy.polys.galoistools.gf_neg(f, p, K)
Negate a polynomial in GF(p)[x].
Examples
sympy.polys.galoistools.gf_add_ground(f, a, p, K)
Compute f + a where f in GF(p)[x] and a in GF(p).
Examples
sympy.polys.galoistools.gf_sub_ground(f, a, p, K)
Compute f - a where f in GF(p)[x] and a in GF(p).
Examples
sympy.polys.galoistools.gf_mul_ground(f, a, p, K)
Compute f * a where f in GF(p)[x] and a in GF(p).
Examples
sympy.polys.galoistools.gf_quo_ground(f, a, p, K)
Compute f/a where f in GF(p)[x] and a in GF(p).
Examples
sympy.polys.galoistools.gf_add(f, g, p, K)
Add polynomials in GF(p)[x].
Examples
sympy.polys.galoistools.gf_sub(f, g, p, K)
Subtract polynomials in GF(p)[x].
Examples
sympy.polys.galoistools.gf_mul(f, g, p, K)
Multiply polynomials in GF(p)[x].
Examples
sympy.polys.galoistools.gf_sqr(f, p, K)
Square polynomials in GF(p)[x].
Examples
sympy.polys.galoistools.gf_add_mul(f, g, h, p, K)
Returns f + g*h where f, g, h in GF(p)[x].
Examples
sympy.polys.galoistools.gf_sub_mul(f, g, h, p, K)
Compute f - g*h where f, g, h in GF(p)[x].
Examples
sympy.polys.galoistools.gf_expand(F, p, K)
Expand results of factor() in GF(p)[x].
Examples
>>> gf_expand([([3, 2, 4], 1), ([2, 2], 2), ([3, 1], 3)], 5, ZZ)
[4, 3, 0, 3, 0, 1, 4, 1]
sympy.polys.galoistools.gf_div(f, g, p, K)
Division with remainder in GF(p)[x].
Given univariate polynomials f and g with coefficients in a finite field with p elements,
returns polynomials q and r (quotient and remainder) such that f = q*g + r.
Consider polynomials x**3 + x + 1 and x**2 + x in GF(2):
References
1. [Monagan93]
2. [Gathen99]
[Monagan93], [Gathen99]
sympy.polys.galoistools.gf_rem(f, g, p, K)
Compute polynomial remainder in GF(p)[x].
Examples
sympy.polys.galoistools.gf_quo(f, g, p, K)
Compute exact quotient in GF(p)[x].
Examples
sympy.polys.galoistools.gf_exquo(f, g, p, K)
Compute polynomial quotient in GF(p)[x].
Examples
sympy.polys.galoistools.gf_lshift(f, n, K)
Efficiently multiply f by x**n.
Examples
sympy.polys.galoistools.gf_rshift(f, n, K)
Efficiently divide f by x**n.
Examples
sympy.polys.galoistools.gf_pow(f, n, p, K)
Compute f**n in GF(p)[x] using repeated squaring.
Examples
sympy.polys.galoistools.gf_pow_mod(f, n, g, p, K)
Compute f**n in GF(p)[x]/(g) using repeated squaring.
Given polynomials f and g in GF(p)[x] and a non-negative integer n, efficiently computes
f**n (mod g) i.e. the remainder of f**n from division by g, using the repeated squaring
algorithm.
References
1. [Gathen99]
[Gathen99]
Examples
sympy.polys.galoistools.gf_gcd(f, g, p, K)
Euclidean Algorithm in GF(p)[x].
Examples
sympy.polys.galoistools.gf_lcm(f, g, p, K)
Compute polynomial LCM in GF(p)[x].
Examples
sympy.polys.galoistools.gf_cofactors(f, g, p, K)
Compute polynomial GCD and cofactors in GF(p)[x].
Examples
sympy.polys.galoistools.gf_gcdex(f, g, p, K)
Extended Euclidean Algorithm in GF(p)[x].
Given polynomials f and g in GF(p)[x], computes polynomials s, t and h, such that h
= gcd(f, g) and s*f + t*g = h. The typical application of EEA is solving polynomial
diophantine equations.
Consider polynomials f = (x + 7) (x + 1), g = (x + 7) (x**2 + 1) in GF(11)[x].
Application of Extended Euclidean Algorithm gives:
References
1. [Gathen99]
[Gathen99]
sympy.polys.galoistools.gf_monic(f, p, K)
Compute LC and a monic polynomial in GF(p)[x].
Examples
sympy.polys.galoistools.gf_diff(f, p, K)
Differentiate polynomial in GF(p)[x].
Examples
sympy.polys.galoistools.gf_eval(f, a, p, K)
Evaluate f(a) in GF(p) using Horner scheme.
Examples
sympy.polys.galoistools.gf_multi_eval(f, A, p, K)
Evaluate f(a) for a in [a_1, ..., a_n].
Examples
sympy.polys.galoistools.gf_compose(f, g, p, K)
Compute polynomial composition f(g) in GF(p)[x].
Examples
sympy.polys.galoistools.gf_compose_mod(g, h, f, p, K)
Compute polynomial composition g(h) in GF(p)[x]/(f).
Examples
sympy.polys.galoistools.gf_trace_map(a, b, c, n, f, p, K)
Compute polynomial trace map in GF(p)[x]/(f).
Given a polynomial f in GF(p)[x], polynomials a, b, c in the quotient ring GF(p)[x]/(f)
such that b = c**t (mod f) for some positive power t of p, and a positive integer n,
returns a mapping:
In factorization context, b = x**p mod f and c = x mod f. This way we can efficiently
compute trace polynomials in equal degree factorization routine, much faster than with
other methods, like iterated Frobenius algorithm, for large degrees.
References
1. [Gathen92]
[Gathen92]
Examples
>>> gf_trace_map([1, 2], [4, 4], [1, 1], 4, [3, 2, 4], 5, ZZ)
([1, 3], [1, 3])
sympy.polys.galoistools.gf_random(n, p, K)
Generate a random polynomial in GF(p)[x] of degree n.
Examples
sympy.polys.galoistools.gf_irreducible(n, p, K)
Generate random irreducible polynomial of degree n in GF(p)[x].
Examples
sympy.polys.galoistools.gf_irreducible_p(f, p, K)
Test irreducibility of a polynomial f in GF(p)[x].
Examples
sympy.polys.galoistools.gf_sqf_p(f, p, K)
Return True if f is square-free in GF(p)[x].
Examples
sympy.polys.galoistools.gf_sqf_part(f, p, K)
Return square-free part of a GF(p)[x] polynomial.
Examples
sympy.polys.galoistools.gf_sqf_list(f, p, K, all=False)
Return the square-free decomposition of a GF(p)[x] polynomial.
Given a polynomial f in GF(p)[x], returns the leading coefficient of f and a square-free
decomposition f_1**e_1 f_2**e_2 ... f_k**e_k such that all f_i are monic polyno-
mials and (f_i, f_j) for i != j are co-prime and e_1 ... e_k are given in increasing
order. All trivial terms (i.e. f_i = 1) aren’t included in the output.
Consider polynomial f = x**11 + 1 over GF(11)[x]:
This phenomenon doesn’t happen in characteristic zero. However we can still compute
square-free decomposition of f using gf_sqf():
References
1. [Geddes92]
[Geddes92]
sympy.polys.galoistools.gf_Qmatrix(f, p, K)
Calculate Berlekamp’s Q matrix.
Examples
sympy.polys.galoistools.gf_Qbasis(Q, p, K)
Compute a basis of the kernel of Q.
Examples
sympy.polys.galoistools.gf_berlekamp(f, p, K)
Factor a square-free f in GF(p)[x] for small p.
Examples
sympy.polys.galoistools.gf_zassenhaus(f, p, K)
Factor a square-free f in GF(p)[x] for medium p.
Examples
sympy.polys.galoistools.gf_shoup(f, p, K)
Factor a square-free f in GF(p)[x] for large p.
Examples
sympy.polys.galoistools.gf_factor_sqf(f, p, K, method=None)
Factor a square-free polynomial f in GF(p)[x].
Examples
sympy.polys.galoistools.gf_factor(f, p, K)
Factor (non square-free) polynomials in GF(p)[x].
Given a possibly non square-free polynomial f in GF(p)[x], returns its complete factor-
ization into irreducibles:
where each f_i is a monic polynomial and gcd(f_i, f_j) == 1, for i != j. The result
is given as a tuple consisting of the leading coefficient of f and a list of factors of f with
their multiplicities.
The algorithm proceeds by first computing square-free decomposition of f and then it-
eratively factoring each of square-free factors.
Consider a non square-free polynomial f = (7*x + 1) (x + 2)**2 in GF(11)[x]. We
obtain its factorization into irreducibles as follows:
References
1. [Gathen99]
[Gathen99]
sympy.polys.galoistools.gf_value(f, a)
Value of polynomial ‘f’ at ‘a’ in field R.
Examples
sympy.polys.galoistools.gf_csolve(f, n)
To solve f(x) congruent 0 mod(n).
n is divided into canonical factors and f(x) cong 0 mod(p**e) will be solved for each factor.
Applying the Chinese Remainder Theorem to the results returns the final answers.
References
[1] ‘An introduction to the Theory of Numbers’ 5th Edition by Ivan Niven,
Zuckerman and Montgomery.
Examples
Dense representations quickly require infeasible amounts of storage and computation time if
the number of variables increases. For this reason, there is code to manipulate polynomials
in a sparse representation.
Sparse polynomials are represented as dictionaries.
sympy.polys.rings.ring(symbols, domain, order=LexOrder())
Construct a polynomial ring returning (ring, x_1, ..., x_n).
Parameters symbols : str, Symbol/Expr or sequence of str, Symbol/Expr (non-
empty)
domain : Domain or coercible
order : Order or coercible, optional, defaults to lex
Examples
Examples
Examples
Examples
>>> x, y, z = symbols(”x,y,z”)
>>> R, f = sring(x + 2*y + 3*z)
>>> R
Polynomial ring in x, y, z over ZZ with lex order
>>> f
x + 2*y + 3*z
>>> type(_)
<class 'sympy.polys.rings.PolyElement'>
class sympy.polys.rings.PolyRing
Multivariate distributed polynomial ring.
Attributes
domain
gens
ngens
order
symbols
add(*objs)
Add a sequence of polynomials or containers of polynomials.
Examples
add_gens(symbols)
Add the elements of symbols as generators to self
compose(other)
Add the generators of other to self
drop(*gens)
Remove specified generators from this ring.
drop_to_ground(*gens)
Remove specified generators from the ring and inject them into its domain.
index(gen)
Compute index of gen in self.gens.
monomial_basis(i)
Return the ith-basis element.
mul(*objs)
Multiply a sequence of polynomials or containers of polynomials.
Examples
class sympy.polys.rings.PolyElement
Element of multivariate distributed polynomial ring.
almosteq(p2, tolerance=None)
Approximate equality test for polynomials.
cancel(g)
Cancel common factors in a rational function f/g.
Examples
coeff(element)
Returns the coefficient that stands next to the given monomial.
Parameters element : PolyElement (with is_monomial = True) or 1
Examples
>>> f.coeff(x**2*y)
3
>>> f.coeff(x*y)
0
>>> f.coeff(1)
23
coeffs(order=None)
Ordered list of polynomial coefficients.
Parameters order : Order or coercible, optional
Examples
>>> f.coeffs()
[2, 1]
>>> f.coeffs(grlex)
[1, 2]
const()
Returns the constant coeffcient.
content()
Returns GCD of polynomial’s coefficients.
copy()
Return a copy of polynomial self.
Polynomials are mutable; if one is interested in preserving a polynomial, and one
plans to use inplace operations, one can copy the polynomial. This method makes a
shallow copy.
Examples
degree(x=None)
The leading degree in x or the main variable.
Note that the degree of 0 is negative infinity (the SymPy object -oo).
degrees()
A tuple containing leading degrees in all variables.
Note that the degree of 0 is negative infinity (the SymPy object -oo)
diff(x)
Computes partial derivative in x.
Examples
div(fv)
Division algorithm, see [CLO] p64.
fv array of polynomials return qv, r such that self = sum(fv[i]*qv[i]) + r
All polynomials are required not to be Laurent polynomials.
Examples
imul_num(c)
multiply inplace the polynomial p by an element in the coefficient ring, provided p
is not one of the generators; else multiply not inplace
Examples
itercoeffs()
Iterator over coefficients of a polynomial.
itermonoms()
Iterator over monomials of a polynomial.
iterterms()
Iterator over terms of a polynomial.
leading_expv()
Leading monomial tuple according to the monomial ordering.
Examples
leading_monom()
Leading monomial as a polynomial element.
Examples
leading_term()
Leading term as a polynomial element.
Examples
listcoeffs()
Unordered list of polynomial coefficients.
listmonoms()
Unordered list of polynomial monomials.
listterms()
Unordered list of polynomial terms.
monic()
Divides all coefficients by the leading coefficient.
monoms(order=None)
Ordered list of polynomial monomials.
Parameters order : Order or coercible, optional
Examples
>>> f.monoms()
[(2, 3), (1, 7)]
>>> f.monoms(grlex)
[(1, 7), (2, 3)]
primitive()
Returns content and a primitive polynomial.
square()
square of a polynomial
Examples
strip_zero()
Eliminate monomials with zero coefficient.
tail_degree(x=None)
The tail degree in x or the main variable.
Note that the degree of 0 is negative infinity (the SymPy object -oo)
tail_degrees()
A tuple containing tail degrees in all variables.
Note that the degree of 0 is negative infinity (the SymPy object -oo)
terms(order=None)
Ordered list of polynomial terms.
Parameters order : Order or coercible, optional
Examples
>>> f.terms()
[((2, 3), 2), ((1, 7), 1)]
>>> f.terms(grlex)
[((1, 7), 1), ((2, 3), 2)]
In commutative algebra, one often studies not only polynomials, but also modules over poly-
nomial rings. The polynomial manipulation module provides rudimentary low-level support
for finitely generated free modules. This is mainly used for Groebner basis computations (see
there), so manipulation functions are only provided to the extend needed. They carry the
prefix sdm_. Note that in examples, the generators of the free module are called f1 , f2 , . . ..
sympy.polys.distributedmodules.sdm_monomial_mul(M, X)
Multiply tuple X representing a monomial of K[X] into the tuple M representing a mono-
mial of F .
Examples
sympy.polys.distributedmodules.sdm_monomial_deg(M)
Return the total degree of M.
Examples
sympy.polys.distributedmodules.sdm_monomial_divides(A, B)
Does there exist a (polynomial) monomial X such that XA = B?
Examples
Positive examples:
In the following examples, the monomial is given in terms of x, y and the generator(s),
f_1, f_2 etc. The tuple form of that monomial is used in the call to sdm_monomial_divides.
Note: the generator appears last in the expression but first in the tuple and other factors
appear in the same order that they appear in the monomial expression.
A = f1 divides B = f1
A = f1 divides B = x2 yf1
Negative examples:
A = f1 does not divide B = f2
sympy.polys.distributedmodules.sdm_LC(f, K)
Returns the leading coeffcient of f.
sympy.polys.distributedmodules.sdm_to_dict(f)
Make a dictionary from a distributed polynomial.
sympy.polys.distributedmodules.sdm_from_dict(d, O)
Create an sdm from a dictionary.
Here O is the monomial order to use.
sympy.polys.distributedmodules.sdm_add(f, g, O, K)
Add two module elements f, g.
Addition is done over the ground field K, monomials are ordered according to O.
Examples
(xyf1 ) + (−xyf1 ) = 0‘
sympy.polys.distributedmodules.sdm_LM(f)
Returns the leading monomial of f.
Only valid if f ̸= 0.
Examples
sympy.polys.distributedmodules.sdm_LT(f)
Returns the leading term of f.
Only valid if f ̸= 0.
Examples
sympy.polys.distributedmodules.sdm_mul_term(f, term, O, K)
Multiply a distributed module element f by a (polynomial) term term.
Multiplication of coefficients is done over the ground field K, and monomials are ordered
according to O.
Examples
0f1 = 0
x0 = 0
(x)(f1 ) = xf1
sympy.polys.distributedmodules.sdm_zero()
Return the zero module element.
sympy.polys.distributedmodules.sdm_deg(f)
Degree of f.
This is the maximum of the degrees of all its monomials. Invalid if f is zero.
Examples
sympy.polys.distributedmodules.sdm_from_vector(vec, O, K, **opts)
Create an sdm from an iterable of expressions.
Coefficients are created in the ground field K, and terms are ordered according to mono-
mial order O. Named arguments are passed on to the polys conversion code and can be
used to specify for example generators.
Examples
Examples
Let K be a field, and consider the ring K[X] of polynomials in a single indeterminate X with
coefficients in K. Given two elements f and g of K[X] with g ̸= 0 there are unique polynomials q
and r such that f = qg + r and deg(r) < deg(g) or r = 0. They are denoted by quo(f, g) (quotient)
and rem(f, g) (remainder), so we have the division identity
It follows that every ideal I of K[X] is a principal ideal, generated by any element ̸= 0 of
minimum degree (assuming I non-zero). In fact, if g is such a polynomial and f is any element
of I, rem(f, g) belongs to I as a linear combination of f and g, hence must be zero; therefore
f is a multiple of g.
Using this result it is possible to find a greatest common divisor (gcd) of any polynomials
f, g, . . . in K[X]. If I is the ideal formed by all linear combinations of the given polynomials
with coefficients in K[X], and d is its generator, then every common divisor of the polynomials
also divides d. On the other hand, the given polynomials are multiples of the generator d;
hence d is a gcd of the polynomials, denoted gcd(f, g, . . .).
An algorithm for the gcd of two polynomials f and g in K[X] can now be obtained as follows.
By the division identity, r = rem(f, g) is in the ideal generated by f and g, as well as f is in the
ideal generated by g and r. Hence the ideals generated by the pairs (f, g) and (g, r) are the
same. Set f0 = f , f1 = g, and define recursively fi = rem(fi−2 , fi−1 ) for i ≥ 2. The recursion
ends after a finite number of steps with fk+1 = 0, since the degrees of the polynomials are
strictly decreasing. By the above remark, all the pairs (fi−1 , fi ) generate the same ideal. In
particular, the ideal generated by f and g is generated by fk alone as fk+1 = 0. Hence d = fk is
a gcd of f and g.
The sequence of polynomials f0 , f1 , . . . , fk is called the Euclidean polynomial remainder se-
quence determined by (f, g) because of the analogy with the classical Euclidean algorithm for
the gcd of natural numbers.
The algorithm may be extended to obtain an expression for d in terms of f and g by using the
full division identities to write recursively each fi as a linear combination of f and g. This
leads to an equation
d = uf + vg (u, v ∈ K[X])
Examples
sympy.polys.euclidtools.dmp_gcdex(f, g, u, K)
Extended Euclidean algorithm in F [X].
Examples
sympy.polys.euclidtools.dmp_invert(f, g, u, K)
Compute multiplicative inverse of f modulo g in F [X].
Examples
sympy.polys.euclidtools.dmp_euclidean_prs(f, g, u, K)
Euclidean polynomial remainder sequence (PRS) in K[X].
Examples
Assume, as is usual, that the coefficient field K is the field of fractions of an integral domain
A. In this case the coefficients (numerators and denominators) of the polynomials in the
Euclidean remainder sequence tend to grow very fast.
If A is a unique factorization domain, the coefficients may be reduced by cancelling common
factors of numerators and denominators. Further reduction is possible noting that a gcd of
polynomials in K[X] is not unique: it may be multiplied by any (non-zero) constant factor.
Any polynomial f in K[X] can be simplified by extracting the denominators and common fac-
tors of the numerators of its coefficients. This yields the representation f = cF where c ∈ K
is the content of f and F is a primitive polynomial, i.e., a polynomial in A[X] with coprime
coefficients.
It is possible to start the algorithm by replacing the given polynomials f and g with their
primitive parts. This will only modify rem(f, g) by a constant factor. Replacing it with its
primitive part and continuing recursively we obtain all the primitive parts of the polynomials
in the Euclidean remainder sequence, including the primitive gcd(f, g).
This sequence is the primitive polynomial remainder sequence. It is an example of general
polynomial remainder sequences where the computed remainders are modified by constant
multipliers (or divisors) in order to simplify the results.
sympy.polys.euclidtools.dmp_primitive_prs(f, g, u, K)
Primitive polynomial remainder sequence (PRS) in K[X].
Examples
Subresultant sequence
The coefficients of the primitive polynomial sequence do not grow exceedingly, but the com-
putation of the primitive parts requires extra processing effort. Besides, the method only
works with fraction fields of unique factorization domains, excluding, for example, the gen-
eral number fields.
Collins [Collins67] realized that the so-called subresultant polynomials of a pair of polynomi-
als also form a generalized remainder sequence. The coefficients of these polynomials are
expressible as determinants in the coefficients of the given polynomials. Hence (the logarithm
of) their size only grows linearly. In addition, if the coefficients of the given polynomials are in
the subdomain A, so are those of the subresultant polynomials. This means that the subresul-
tant sequence is comparable to the primitive remainder sequence without relying on unique
factorization in A.
To see how subresultants are associated with remainder sequences recall that all polynomials
h in the sequence are linear combinations of the given polynomials f and g
h = uf + vg
with polynomials u and v in K[X]. Moreover, as is seen from the extended Euclidean algorithm,
the degrees of u and v are relatively low, with limited growth from step to step.
Let n = deg(f ), and m = deg(g), and assume n ≥ m. If deg(h) = j < m, the coefficients of the
powers X k (k > j) in the products uf and vg cancel each other. In particular, the products must
have the same degree, say, l. Then deg(u) = l − n and deg(v) = l − m with a total of 2l − n − m + 2
coefficients to be determined.
On the other hand, the equality h = uf + vg implies that l − j linear combinations of the
coefficients are zero, those associated with the powers X i (j < i ≤ l), and one has a given
non-zero value, namely the leading coefficient of h.
To satisfy these l − j + 1 linear equations the total number of coefficients to be determined
cannot be lower than l − j + 1, in general. This leads to the inequality l ≥ n + m − j − 1. Taking
l = n + m − j − 1, we obtain deg(u) = m − j − 1 and deg(v) = n − j − 1.
In the case j = 0 the matrix of the resulting system of linear equations is the Sylvester matrix
S(f, g) associated to f and g, an (n + m) × (n + m) matrix with coefficients of f and g as entries.
Its determinant is the resultant res(f, g) of the pair (f, g). It is non-zero if and only if f and g
are relatively prime.
For any j in the interval from 0 to m the matrix of the linear system is an (n+m−2j)×(n+m−2j)
submatrix of the Sylvester matrix. Its determinant sj (f, g) is called the j th scalar subresultant
of f and g.
If sj (f, g) is not zero, the associated equation h = uf +vg has a unique solution where deg(h) = j
and the leading coefficient of h has any given value; the one with leading coefficient sj (f, g) is
the j th subresultant polynomial or, briefly, subresultant of the pair (f, g), and denoted Sj (f, g).
This choice guarantees that the remainining coefficients are also certain subdeterminants of
the Sylvester matrix. In particular, if f and g are in A[X], so is Sj (f, g) as well. This construction
of subresultants applies to any j between 0 and m regardless of the value of sj (f, g); if it is zero,
then deg(Sj (f, g)) < j.
The properties of subresultants are as follows. Let n0 = deg(f ), n1 = deg(g), n2 , . . . , nk be the
decreasing sequence of degrees of polynomials in a remainder sequence. Let 0 ≤ j ≤ n1 ; then
• sj (f, g) ̸= 0 if and only if j = ni for some i.
• Sj (f, g) ̸= 0 if and only if j = ni or j = ni − 1 for some i.
Normally, ni−1 − ni = 1 for 1 < i ≤ k. If ni−1 − ni > 1 for some i (the abnormal case), then
Sni−1 −1 (f, g) and Sni (f, g) are constant multiples of each other. Hence either one could be
included in the polynomial remainder sequence. The former is given by smaller determinants,
so it is expected to have smaller coefficients.
Collins defined the subresultant remainder sequence by setting
In the normal case, these are the same as the Sni (f, g). He also derived expressions for the
constants γi in the remainder formulas
γi fi = rem(fi−2 , fi−1 )
Brown and Traub [BrownTraub71] later developed a recursive procedure for computing the
coefficients γi . Their algorithm deals with elements of the domain A exclusively (assuming
f, g ∈ A[X]). However, in the abnormal case there was a problem, a division in A which could
only be conjectured to be exact.
This was subsequently justified by Brown [Brown78] who showed that the result of the division
is, in fact, a scalar subresultant. More specifically, the constant appearing in the computation
of fi is sni−2 (f, g) (Theorem 3). The implication of this discovery is that the scalar subresultants
are computed as by-products of the algorithm, all but snk (f, g) which is not needed after finding
fk+1 = 0. Completing the last step we obtain all non-zero scalar subresultants, including the
last one which is the resultant if this does not vanish.
sympy.polys.euclidtools.dmp_inner_subresultants(f, g, u, K)
Subresultant PRS algorithm in K[X].
Examples
sympy.polys.euclidtools.dmp_subresultants(f, g, u, K)
Computes subresultant PRS of two polynomials in K[X].
Examples
sympy.polys.euclidtools.dmp_prs_resultant(f, g, u, K)
Resultant algorithm in K[X] using subresultant PRS.
Examples
sympy.polys.euclidtools.dmp_zz_modular_resultant(f, g, p, u, K)
Compute resultant of f and g modulo a prime p.
Examples
>>> f = x + y + 2
>>> g = 2*x*y + x + 3
>>> R.dmp_zz_modular_resultant(f, g, 5)
-2*y**2 + 1
sympy.polys.euclidtools.dmp_zz_collins_resultant(f, g, u, K)
Collins’s modular resultant algorithm in Z[X].
Examples
>>> f = x + y + 2
>>> g = 2*x*y + x + 3
>>> R.dmp_zz_collins_resultant(f, g)
-2*y**2 - 5*y + 1
sympy.polys.euclidtools.dmp_qq_collins_resultant(f, g, u, K0)
Collins’s modular resultant algorithm in Q[X].
Examples
>>> R.dmp_qq_collins_resultant(f, g)
-2*y**2 - 7/3*y + 5/6
sympy.polys.euclidtools.dmp_resultant(f, g, u, K, includePRS=False)
Computes resultant of two polynomials in K[X].
Examples
>>> R.dmp_resultant(f, g)
-3*y**10 - 12*y**7 + y**6 - 54*y**4 + 8*y**3 + 729*y**2 - 216*y + 16
sympy.polys.euclidtools.dmp_discriminant(f, u, K)
Computes discriminant of a polynomial in K[X].
Examples
sympy.polys.euclidtools.dmp_rr_prs_gcd(f, g, u, K)
Computes polynomial GCD using subresultants over a ring.
Returns (h, cff, cfg) such that a = gcd(f, g), cff = quo(f, h), and cfg = quo(g,
h).
Examples
>>> R.dmp_rr_prs_gcd(f, g)
(x + y, x + y, x)
sympy.polys.euclidtools.dmp_ff_prs_gcd(f, g, u, K)
Computes polynomial GCD using subresultants over a field.
Returns (h, cff, cfg) such that a = gcd(f, g), cff = quo(f, h), and cfg = quo(g,
h).
Examples
>>> R.dmp_ff_prs_gcd(f, g)
(x + y, 1/2*x + 1/2*y, x)
sympy.polys.euclidtools.dmp_zz_heu_gcd(f, g, u, K)
Heuristic polynomial GCD in Z[X].
Given univariate polynomials f and g in Z[X], returns their GCD and cofactors, i.e. poly-
nomials h, cff and cfg such that:
The algorithm is purely heuristic which means it may fail to compute the GCD. This will
be signaled by raising an exception. In this case you will need to switch to another GCD
method.
The algorithm computes the polynomial GCD by evaluating polynomials f and g at certain
points and computing (fast) integer GCD of those evaluations. The polynomial GCD is
recovered from the integer image by interpolation. The evaluation process reduces f and
g variable by variable into a large integer. The final step is to verify if the interpolated
polynomial is the correct GCD. This gives cofactors of the input polynomials as a side
effect.
References
1. [Liao95]
[Liao95]
Examples
>>> R.dmp_zz_heu_gcd(f, g)
(x + y, x + y, x)
sympy.polys.euclidtools.dmp_qq_heu_gcd(f, g, u, K0)
Heuristic polynomial GCD in Q[X].
Returns (h, cff, cfg) such that a = gcd(f, g), cff = quo(f, h), and cfg = quo(g,
h).
Examples
>>> R.dmp_qq_heu_gcd(f, g)
(x + 2*y, 1/4*x + 1/2*y, 1/2*x)
sympy.polys.euclidtools.dmp_inner_gcd(f, g, u, K)
Computes polynomial GCD and cofactors of f and g in K[X].
Returns (h, cff, cfg) such that a = gcd(f, g), cff = quo(f, h), and cfg = quo(g,
h).
Examples
>>> R.dmp_inner_gcd(f, g)
(x + y, x + y, x)
sympy.polys.euclidtools.dmp_gcd(f, g, u, K)
Computes polynomial GCD of f and g in K[X].
Examples
>>> R.dmp_gcd(f, g)
x + y
sympy.polys.euclidtools.dmp_lcm(f, g, u, K)
Computes polynomial LCM of f and g in K[X].
Examples
>>> R.dmp_lcm(f, g)
x**3 + 2*x**2*y + x*y**2
sympy.polys.euclidtools.dmp_content(f, u, K)
Returns GCD of multivariate coefficients.
Examples
sympy.polys.euclidtools.dmp_primitive(f, u, K)
Returns multivariate content and a primitive polynomial.
Examples
sympy.polys.euclidtools.dmp_cancel(f, g, u, K, include=True)
Cancel common factors in a rational function f /g.
Examples
References
1. [Gathen99]
[Gathen99]
sympy.polys.factortools.dup_zz_hensel_lift(p, f, f_list, l, K)
Multifactor Hensel lifting in Z[x].
Given a prime p, polynomial f over Z[x] such that lc(f ) is a unit modulo p, monic pair-wise
coprime polynomials fi over Z[x] satisfying:
f = lc(f) f_1 ... f_r (mod p)
References
1. [Gathen99]
[Gathen99]
sympy.polys.factortools.dup_zz_zassenhaus(f, K)
Factor primitive square-free polynomials in Z[x].
sympy.polys.factortools.dup_zz_irreducible_p(f, K)
Test irreducibility using Eisenstein’s criterion.
sympy.polys.factortools.dup_cyclotomic_p(f, K, irreducible=False)
Efficiently test if f is a cyclotomic polnomial.
Examples
sympy.polys.factortools.dup_zz_cyclotomic_poly(n, K)
Efficiently generate n-th cyclotomic polnomial.
sympy.polys.factortools.dup_zz_cyclotomic_factor(f, K)
Efficiently factor polynomials x ∗ ∗n − 1 and x ∗ ∗n + 1 in Z[x].
Given a univariate polynomial f in Z[x] returns a list of factors of f , provided that f is in
the form x ∗ ∗n − 1 or x ∗ ∗n + 1 for n >= 1. Otherwise returns None.
Factorization is performed using using cyclotomic decomposition of f , which makes this
method much faster that any other direct factorization approach (e.g. Zassenhaus’s).
References
1. [Weisstein09]
[Weisstein09]
sympy.polys.factortools.dup_zz_factor_sqf(f, K)
Factor square-free (non-primitive) polyomials in Z[x].
sympy.polys.factortools.dup_zz_factor(f, K)
Factor (non square-free) polynomials in Z[x].
Given a univariate polynomial f in Z[x] computes its complete factorization f1 , ..., fn into
irreducibles over integers:
The factorization is computed by reducing the input polynomial into a primitive square-
free polynomial and factoring it using Zassenhaus algorithm. Trial division is used to
recover the multiplicities of factors.
The result is returned as a tuple consisting of:
Consider polynomial f = 2 ∗ x ∗ ∗4 − 2:
>>> R.dup_zz_factor(2*x**4 - 2)
(2, [(x - 1, 1), (x + 1, 1), (x**2 + 1, 1)])
f = 2 (x - 1) (x + 1) (x**2 + 1)
Note that this is a complete factorization over integers, however over Gaussian integers
we can factor the last term.
By default, polynomials x∗∗n−1 and x∗∗n+1 are factored using cyclotomic decomposition
to speedup computations. To disable this behaviour set cyclotomic=False.
References
1. [Gathen99]
[Gathen99]
sympy.polys.factortools.dmp_zz_wang_non_divisors(E, cs, ct, K)
Wang/EEZ: Compute a set of valid divisors.
sympy.polys.factortools.dmp_zz_wang_test_points(f, T, ct, A, u, K)
Wang/EEZ: Test evaluation points for suitability.
sympy.polys.factortools.dmp_zz_wang_lead_coeffs(f, T, cs, E, H, A, u, K)
Wang/EEZ: Compute correct leading coefficients.
sympy.polys.factortools.dmp_zz_diophantine(F, c, A, d, p, u, K)
Wang/EEZ: Solve multivariate Diophantine equations.
sympy.polys.factortools.dmp_zz_wang_hensel_lifting(f, H, LC, A, p, u, K)
Wang/EEZ: Parallel Hensel lifting algorithm.
sympy.polys.factortools.dmp_zz_wang(f, u, K, mod=None, seed=None)
Factor primitive square-free polynomials in Z[X].
Given a multivariate polynomial f in Z[x1 , ..., xn ], which is primitive and square-free in x1 ,
computes factorization of f into irreducibles over integers.
The procedure is based on Wang’s Enhanced Extended Zassenhaus algorithm. The al-
gorithm works by viewing f as a univariate polynomial in Z[x2 , ..., xn ][x1 ], for which an
evaluation mapping is computed:
where ai , for i = 2, ..., n, are carefully chosen integers. The mapping is used to transform f
into a univariate polynomial in Z[x1 ], which can be factored efficiently using Zassenhaus
algorithm. The last step is to lift univariate factors to obtain true multivariate factors.
For this purpose a parallel Hensel lifting procedure is used.
The parameter seed is passed to _randint and can be used to seed randint (when an
integer) or (for testing purposes) can be a sequence of numbers.
References
1. [Wang78]
2. [Geddes92]
[Wang78], [Geddes92]
sympy.polys.factortools.dmp_zz_factor(f, u, K)
Factor (non square-free) polynomials in Z[X].
Given a multivariate polynomial f in Z[x] computes its complete factorization f1 , ..., fn into
irreducibles over integers:
The factorization is computed by reducing the input polynomial into a primitive square-
free polynomial and factoring it using Enhanced Extended Zassenhaus (EEZ) algorithm.
Trial division is used to recover the multiplicities of factors.
The result is returned as a tuple consisting of:
f = 2 (x - y) (x + y)
References
1. [Gathen99]
[Gathen99]
sympy.polys.factortools.dmp_ext_factor(f, u, K)
Factor multivariate polynomials over algebraic number fields.
sympy.polys.factortools.dup_gf_factor(f, K)
Factor univariate polynomials over finite fields.
sympy.polys.factortools.dmp_factor_list(f, u, K0)
Factor polynomials into irreducibles in K[X].
sympy.polys.factortools.dmp_factor_list_include(f, u, K)
Factor polynomials into irreducibles in K[X].
sympy.polys.factortools.dmp_irreducible_p(f, u, K)
Returns True if f has no factors over its domain.
Groebner bases can be used to answer many problems in computational commutative algebra.
Their computation in rather complicated, and very performance-sensitive. We present here
various low-level implementations of Groebner basis computation algorithms; please see the
previous section of the manual for usage.
sympy.polys.groebnertools.groebner(seq, ring, method=None)
Computes Groebner basis for a set of polynomials in K[X].
Wrapper around the (default) improved Buchberger and the other algorithms for com-
puting Groebner bases. The choice of algorithm can be changed via method argument
or setup() from sympy.polys.polyconfig, where method can be either buchberger or
f5b.
sympy.polys.groebnertools.spoly(p1, p2, ring)
Compute LCM(LM(p1), LM(p2))/LM(p1)*p1 - LCM(LM(p1), LM(p2))/LM(p2)*p2 This is
the S-poly provided p1 and p2 are monic
sympy.polys.groebnertools.red_groebner(G, ring)
Compute reduced Groebner basis, from BeckerWeispfenning93, p. 216
Selects a subset of generators, that already generate the ideal and computes a reduced
Groebner basis for them.
sympy.polys.groebnertools.is_groebner(G, ring)
Check if G is a Groebner basis.
sympy.polys.groebnertools.is_minimal(G, ring)
Checks if G is a minimal Groebner basis.
sympy.polys.groebnertools.is_reduced(G, ring)
Checks if G is a reduced Groebner basis.
sympy.polys.fglmtools.matrix_fglm(F, ring, O_to)
Converts the reduced Groebner basis F of a zero-dimensional ideal w.r.t. O_from to a
reduced Groebner basis w.r.t. O_to.
References
If phantom is not None, it should be a pair of module elements on which to perform the
same operation(s) as on f and g. The in this case both results are returned.
Examples
sympy.polys.distributedmodules.sdm_ecart(f)
Compute the ecart of f.
This is defined to be the difference of the total degree of f and the total degree of the
leading monomial of f [SCA, defn 2.3.7].
Invalid if f is zero.
Examples
sympy.polys.distributedmodules.sdm_nf_mora(f, G, O, K, phantom=None)
Compute a weak normal form of f with respect to G and order O.
The ground field is assumed to be K, and monomials ordered according to O.
Weak normal forms are defined in [SCA, defn 2.3.3]. They are not unique. This function
deterministically computes a weak normal form, depending on the order of G.
The most important property of a weak normal form is the following: if R is the ring asso-
ciated with the monomial ordering (if the ordering is global, we just have R = K[x1 , . . . , xn ],
otherwise it is a certain localization thereof), I any ideal of R and G a standard basis for
I, then for any f ∈ R, we have f ∈ I if and only if N F (f |G) = 0.
This is the generalized Mora algorithm for computing weak normal forms with respect
to arbitrary monomial orders [SCA, algorithm 2.3.9].
If phantom is not None, it should be a pair of “phantom” arguments on which to perform
the same computations as on f, G, both results are then returned.
sympy.polys.distributedmodules.sdm_groebner(G, NF, O, K, extended=False)
Compute a minimal standard basis of G with respect to order O.
The algorithm uses a normal form NF, for example sdm_nf_mora. The ground field is
assumed to be K, and monomials ordered according to O.
Exceptions
Reference
Modular GCD
sympy.polys.modulargcd.modgcd_univariate(f, g)
Computes the GCD of two polynomials in Z[x] using a modular algorithm.
The algorithm computes the GCD of two univariate integer polynomials f and g by com-
puting the GCD in Zp [x] for suitable primes p and then reconstructing the coefficients
with the Chinese Remainder Theorem. Trial division is only made for candidates which
are very likely the desired GCD.
Parameters f : PolyElement
univariate integer polynomial
g : PolyElement
univariate integer polynomial
Returns h : PolyElement
GCD of the polynomials f and g
cff : PolyElement
f
cofactor of f , i.e. h
cfg : PolyElement
g
cofactor of g, i.e. h
References
1. [Monagan00]
[Monagan00]
Examples
>>> f = x**5 - 1
>>> g = x - 1
>>> cff * h == f
True
>>> cfg * h == g
True
>>> f = 6*x**2 - 6
>>> g = 2*x**2 + 4*x + 2
>>> cff * h == f
True
>>> cfg * h == g
True
sympy.polys.modulargcd.modgcd_bivariate(f, g)
Computes the GCD of two polynomials in Z[x, y] using a modular algorithm.
The algorithm computes the GCD of two bivariate integer polynomials f and g by calcu-
lating the GCD in Zp [x, y] for suitable primes p and then reconstructing the coefficients
with the Chinese Remainder Theorem. To compute the bivariate GCD over Zp , the poly-
nomials f mod p and g mod p are evaluated at y = a for certain a ∈ Zp and then their
univariate GCD in Zp [x] is computed. Interpolating those yields the bivariate GCD in
Zp [x, y]. To verify the result in Z[x, y], trial division is done, but only for candidates which
are very likely the desired GCD.
Parameters f : PolyElement
bivariate integer polynomial
g : PolyElement
bivariate integer polynomial
Returns h : PolyElement
GCD of the polynomials f and g
cff : PolyElement
f
cofactor of f , i.e. h
cfg : PolyElement
g
cofactor of g, i.e. h
References
1. [Monagan00]
[Monagan00]
Examples
>>> cff * h == f
True
>>> cfg * h == g
True
>>> cff * h == f
True
>>> cfg * h == g
True
sympy.polys.modulargcd.modgcd_multivariate(f, g)
Compute the GCD of two polynomials in Z[x0 , . . . , xk−1 ] using a modular algorithm.
The algorithm computes the GCD of two multivariate integer polynomials f and g by
calculating the GCD in Zp [x0 , . . . , xk−1 ] for suitable primes p and then reconstructing the
coefficients with the Chinese Remainder Theorem. To compute the multivariate GCD
over Zp the recursive subroutine _modgcd_multivariate_p is used. To verify the result
in Z[x0 , . . . , xk−1 ], trial division is done, but only for candidates which are very likely the
desired GCD.
Parameters f : PolyElement
multivariate integer polynomial
g : PolyElement
multivariate integer polynomial
Returns h : PolyElement
GCD of the polynomials f and g
cff : PolyElement
f
cofactor of f , i.e. h
cfg : PolyElement
g
cofactor of g, i.e. h
See also:
_modgcd_multivariate_p
References
1. [Monagan00]
2. [Brown71]
[Monagan00], [Brown71]
Examples
>>> cff * h == f
True
>>> cfg * h == g
True
>>> cff * h == f
True
>>> cfg * h == g
True
sympy.polys.modulargcd.func_field_modgcd(f, g)
Compute the GCD of two polynomials f and g in Q(α)[x0 , . . . , xn−1 ] using a modular algo-
rithm.
The algorithm first computes the primitive associate m̌α (z) of the minimal polynomial
mα in Z[z] and the primitive associates of f and g in Z[x1 , . . . , xn−1 ][z]/(m̌α )[x0 ]. Then
it computes the GCD in Q(x1 , . . . , xn−1 )[z]/(mα (z))[x0 ]. This is done by calculating the
GCD in Zp (x1 , . . . , xn−1 )[z]/(m̌α (z))[x0 ] for suitable primes p and then reconstructing the
coefficients with the Chinese Remainder Theorem and Rational Reconstuction. The
GCD over Zp (x1 , . . . , xn−1 )[z]/(m̌α (z))[x0 ] is computed with a recursive subroutine, which
evaluates the polynomials at xn−1 = a for suitable evaluation points a ∈ Zp and then
calls itself recursively until the ground domain does no longer contain any parame-
ters. For Zp [z]/(m̌α (z))[x0 ] the Euclidean Algorithm is used. The results of those re-
cursive calls are then interpolated and Rational Function Reconstruction is used to
obtain the correct coefficients. The results, both in Q(x1 , . . . , xn−1 )[z]/(mα (z))[x0 ] and
Zp (x1 , . . . , xn−1 )[z]/(m̌α (z))[x0 ], are verified by a fraction free trial division.
Apart from the above GCD computation some GCDs in Q(α)[x1 , . . . , xn−1 ] have to be cal-
culated, because treating the polynomials as univariate ones can result in a spurious
content of the GCD. For this func_field_modgcd is called recursively.
Parameters f, g : PolyElement
polynomials in Q(α)[x0 , . . . , xn−1 ]
Returns h : PolyElement
monic GCD of the polynomials f and g
cff : PolyElement
f
cofactor of f , i.e. h
cfg : PolyElement
g
cofactor of g, i.e. h
References
1. [Hoeij04]
[Hoeij04]
Examples
>>> f = x**2 - 2
>>> g = x + sqrt(2)
>>> h == x + sqrt(2)
True
>>> cff * h == f
True
(continues on next page)
>>> h == x + sqrt(2)*y
True
>>> cff * h == f
True
>>> cfg * h == g
True
>>> f = x + sqrt(2)*y
>>> g = x + y
>>> h == R.one
True
>>> cff * h == f
True
>>> cfg * h == g
True
Undocumented
Many parts of the polys module are still undocumented, and even where there is documenta-
tion it is scarce. Please contribute!
Any finite Taylor series, for all practical purposes is, in fact a polynomial. This module makes
use of the efficient representation and operations of sparse polynomials for very fast multi-
variate series manipulations. Typical speedups compared to SymPy’s series method are in
the range 20-100, with the gap widening as the series being handled gets larger.
All the functions expand any given series on some ring specified by the user. Thus, the coef-
ficients of the calculated series depend on the ring being used. For example:
QQ stands for the Rational domain. Here all coefficients are rationals. It is recommended to
use QQ with ring series as it automatically chooses the fastest Rational type.
Similarly, if a Real domain is used:
Though the definition of a polynomial limits the use of Polynomial module to Taylor series, we
extend it to allow Laurent and even Puiseux series (with fractional exponents):
>>> x**QQ(3, 4)
x**(3/4)
rs_series
Direct use of elementary ring series functions does give more control, but is limiting at the
same time. Creating an appropriate ring for the desired series expansion and knowing which
ring series function to call, are things not everyone might be familiar with.
rs_series is a function that takes an arbitrary Expr and returns its expansion by calling the ap-
propriate ring series functions. The returned series is a polynomial over the simplest (almost)
possible ring that does the job. It recursively builds the ring as it parses the given expression,
adding generators to the ring when it needs them. Some examples:
It can expand complicated multivariate expressions involving multiple functions and most
importantly, it does so blazingly fast:
rs_series is over 300 times faster. Given an expression to expand, there is some fixed overhead
to parse it. Thus, for larger orders, the speed improvement becomes more prominent:
To figure out the right ring for a given expression, rs_series uses the sring function, which
in turn uses other functions of polys. As explained above, non-natural exponents are not
allowed. But the restriction is on exponents and not generators. So, polys allows all sorts of
symbolic terms as generators to make sure that the exponent is a natural number:
In the above example, 1/a and a ∗ ∗(1/7) will be treated as completely different atoms. For all
practical purposes, we could let b = 1/a and c = a ∗ ∗(1/7) and do the manipulations. Effectively,
expressions involving 1/a and a ∗ ∗(1/7) (and their powers) will never simplify:
>>> expr*R(1/a)
(1/a)**2 + (1/a)*(a**(1/7))**3
This leads to similar issues with manipulating Laurent and Puiseux series as faced earlier.
Fortunately, this time we have an elegant solution and are able to isolate the series and polys
behaviour from one another. We introduce a boolean flag series in the list of allowed Options
for polynomials (see sympy.polys.polyoptions.Options). Thus, when we want sring to
allow rational exponents we supply a series=True flag to sring:
Contribute
rs_series is not fully implemented yet. As of now, it supports only multivariate Taylor ex-
pansions of expressions involving sin, cos, exp and tan. Adding the remaining functions is
not at all difficult and they will be gradually added. If you are interested in helping, read
the comments in ring_series.py. Currently, it does not support Puiseux series (though the
elementary functions do). This is expected to be fixed soon.
You can also add more functions to ring_series.py. Only elementary functions are supported
currently. The long term goal is to replace SymPy’s current series method with rs_series.
Functions in this module carry the prefix rs_, standing for “ring series”. They manipulate
finite power series in the sparse representation provided by polys.ring.ring.
Elementary functions
sympy.polys.ring_series.rs_log(p, x, prec)
The Logarithm of p modulo O(x**prec).
Notes
Examples
sympy.polys.ring_series.rs_LambertW(p, x, prec)
Calculate the series expansion of the principal branch of the Lambert W function.
See also:
LambertW
Examples
sympy.polys.ring_series.rs_exp(p, x, prec)
Exponentiation of a series modulo O(x**prec)
Examples
sympy.polys.ring_series.rs_atan(p, x, prec)
The arctangent of a series
Return the series expansion of the atan of p, about 0.
See also:
atan
Examples
sympy.polys.ring_series.rs_asin(p, x, prec)
Arcsine of a series
Return the series expansion of the asin of p, about 0.
See also:
asin
Examples
sympy.polys.ring_series.rs_tan(p, x, prec)
Tangent of a series.
Return the series expansion of the tan of p, about 0.
See also:
_tan1, tan
Examples
sympy.polys.ring_series.rs_cot(p, x, prec)
Cotangent of a series
Return the series expansion of the cot of p, about 0.
See also:
cot
Examples
sympy.polys.ring_series.rs_sin(p, x, prec)
Sine of a series
Return the series expansion of the sin of p, about 0.
See also:
sin
Examples
sympy.polys.ring_series.rs_cos(p, x, prec)
Cosine of a series
Return the series expansion of the cos of p, about 0.
See also:
cos
Examples
sympy.polys.ring_series.rs_cos_sin(p, x, prec)
Return the tuple (rs_cos(p, x, prec), rs_sin(p, x, prec)).
Is faster than calling rs_cos and rs_sin separately
sympy.polys.ring_series.rs_atanh(p, x, prec)
Hyperbolic arctangent of a series
Return the series expansion of the atanh of p, about 0.
See also:
atanh
Examples
sympy.polys.ring_series.rs_sinh(p, x, prec)
Hyperbolic sine of a series
Return the series expansion of the sinh of p, about 0.
See also:
sinh
Examples
sympy.polys.ring_series.rs_cosh(p, x, prec)
Hyperbolic cosine of a series
Return the series expansion of the cosh of p, about 0.
See also:
cosh
Examples
sympy.polys.ring_series.rs_tanh(p, x, prec)
Hyperbolic tangent of a series
Return the series expansion of the tanh of p, about 0.
See also:
tanh
Examples
sympy.polys.ring_series.rs_hadamard_exp(p1, inverse=False)
Return sum f_i/i!*x**i from sum f_i*x**i, where x is the first variable.
If invers=True return sum f_i*i!*x**i
Examples
Operations
sympy.polys.ring_series.rs_mul(p1, p2, x, prec)
Return the product of the given two series, modulo O(x**prec).
x is the series variable or its position in the generators.
Examples
sympy.polys.ring_series.rs_square(p1, x, prec)
Square the series modulo O(x**prec)
Examples
sympy.polys.ring_series.rs_pow(p1, n, x, prec)
Return p1**n modulo O(x**prec)
Examples
sympy.polys.ring_series.rs_series_inversion(p, x, prec)
Multivariate series inversion 1/p modulo O(x**prec).
Examples
sympy.polys.ring_series.rs_series_reversion(p, x, n, y)
Reversion of a series.
p is a series with O(x**n) of the form p = a ∗ x + f (x) where a is a number different from
0.
f (x) = sum(a_k ∗ x_k, kinrange(2, n))
a_k : Can depend polynomially on other variables, not indicated. x : Variable
with name x. y : Variable with name y.
Solve p = y, that is, given a ∗ x + f (x) − y = 0, find the solution x = r(y) up to O(y**n)
Algorithm:
If r_i is the solution at order i, then: a ∗ r_i + f (r_i) − y = O(y ∗ ∗(i + 1))
and if r_(i + 1) is the solution at order i + 1, then: a ∗ r_(i + 1) + f (r_(i + 1)) − y = O(y ∗ ∗(i + 2))
We have, r_(i + 1) = r_i + e, such that, a ∗ e + f (r_i) = O(y ∗ ∗(i + 2)) or e = −f (r_i)/a
So we use the recursion relation: r_(i + 1) = r_i − f (r_i)/a with the boundary condition:
r_1 = y
Examples
sympy.polys.ring_series.rs_nth_root(p, n, x, prec)
Multivariate series expansion of the nth root of p.
Parameters n : p ∗ ∗(1/n) is returned.
x : PolyElement
prec : Order of the expanded series.
Notes
The result of this function is dependent on the ring over which the polynomial has been
defined. If the answer involves a root of a constant, make sure that the polynomial is
over a real field. It can not yet handle roots of symbols.
Examples
sympy.polys.ring_series.rs_trunc(p1, x, prec)
Truncate the series in the x variable with precision prec, that is, modulo O(x**prec)
Examples
Examples
sympy.polys.ring_series.rs_diff(p, x)
Return partial derivative of p with respect to x.
Parameters x : PolyElement with respect to which p is differentiated.
Examples
sympy.polys.ring_series.rs_integrate(p, x)
Integrate p with respect to x.
Parameters x : PolyElement with respect to which p is integrated.
Examples
sympy.polys.ring_series.rs_newton(p, x, prec)
Compute the truncated Newton sum of the polynomial p
Examples
sympy.polys.ring_series.rs_compose_add(p1, p2)
compute the composed sum prod(p2(x - beta) for beta root of p1)
References
A. Bostan, P. Flajolet, B. Salvy and E. Schost “Fast Computation with Two Algebraic Num-
bers”, (2002) Research Report 4579, Institut National de Recherche en Informatique et
en Automatique
Examples
Utility functions
sympy.polys.ring_series.rs_is_puiseux(p, x)
Test if p is Puiseux series in x.
Raise an exception if it has a negative power in x.
Examples
sympy.polys.ring_series.rs_puiseux(f, p, x, prec)
Return the puiseux series for f (p, x, prec).
To be used when function f is implemented only for regular series.
Examples
sympy.polys.ring_series.rs_puiseux2(f, p, q, x, prec)
Return the puiseux series for f (p, q, x, prec).
To be used when function f is implemented only for regular series.
sympy.polys.ring_series.rs_series_from_list(p, c, x, prec, concur=1)
Return a series sumc[n] ∗ p ∗ ∗n modulo O(x ∗ ∗prec).
It reduces the number of multiplications by summing concurrently.
ax = [1, p, p ∗ ∗2, .., p ∗ ∗(J − 1)] s = sum(c[i] ∗ ax[i] for i in range(r, (r + 1) ∗ J)) ∗ p ∗ ∗((K − 1) ∗ J)
with K >= (n + 1)/J
See also:
sympy.polys.ring.compose
Examples
sympy.polys.ring_series.rs_fun(p, f, *args)
Function of a multivariate series computed by substitution.
The case with f method name is used to compute rs_tan and rs_nth_root of a multivariate
series:
rs_f un(p, tan, iv, prec)
tan series is first computed for a dummy variable _x, i.e, rs_tan(_x, iv, prec). Then
we substitute _x with p to get the desired series
Examples
sympy.polys.ring_series.mul_xin(p, i, n)
Return p ∗ xi ∗ ∗n.
x_i is the ith variable in p.
sympy.polys.ring_series.pow_xin(p, i, n)
Literature
See the Printing (page 19) section in Tutorial for introduction into printing.
This guide documents the printing system in SymPy and how it works internally.
The whole printing process is started by calling .doprint(expr) on the printer which you
want to use. This method looks for an appropriate method which can print the given expres-
sion in the given style that the printer defines. While looking for the method, it follows these
steps:
1. Let the object print itself if it knows how.
The printer looks for a specific method in every object. The name of that method
depends on the specific printer and is defined under Printer.printmethod. For
example, StrPrinter calls _sympystr and LatexPrinter calls _latex. Look at the
documentation of the printer that you want to use. The name of the method is
specified there.
This was the original way of doing printing in sympy. Every class had its own
latex, mathml, str and repr methods, but it turned out that it is hard to produce
a high quality printer, if all the methods are spread out that far. Therefore all
printing code was combined into the different printers, which works great for
built-in sympy objects, but not that good for user defined classes where it is
inconvenient to patch the printers.
2. Take the best fitting method defined in the printer.
The printer loops through expr classes (class + its bases), and tries to dispatch
the work to _print_<EXPR_CLASS>
e.g., suppose we have the following class hierarchy:
Basic
|
Atom
|
Number
|
Rational
then, for expr=Rational(...), the Printer will try to call printer methods in the
order as shown in the figure below:
p._print(expr)
|
|-- p._print_Rational(expr)
|
|-- p._print_Number(expr)
|
|-- p._print_Atom(expr)
|
`-- p._print_Basic(expr)
if ._print_Rational method exists in the printer, then it is called, and the result
is returned back. Otherwise, the printer tries to call ._print_Number and so on.
3. As a fall-back use the emptyPrinter method for the printer.
As fall-back self.emptyPrinter will be called with the expression. If not de-
fined in the Printer subclass this will be the same as str(expr).
In the example below, we have a printer which prints the derivative of a function in a shorter
form.
class MyLatexPrinter(LatexPrinter):
”””Print derivative of a function of symbols in a shorter form.
”””
def _print_Derivative(self, expr):
function, *vars = expr.args
if not isinstance(type(function), UndefinedFunction) or \
not all(isinstance(i, Symbol) for i in vars):
return super()._print_Derivative(expr)
def print_my_latex(expr):
””” Most of the printers define their own wrappers for print().
These wrappers usually take printer settings. Our printer does not have
any settings.
”””
print(MyLatexPrinter().doprint(expr))
y = Symbol(”y”)
x = Symbol(”x”)
f = Function(”f”)
expr = f(x, y).diff(x, y)
# Print the expression using the normal latex printer and our custom
# printer.
print_latex(expr)
print_my_latex(expr)
In the example below, the latex printing of the modulo operator is modified. This is done by
overriding the method _latex of Mod.
class ModOp(Mod):
def _latex(self, printer=None):
# Always use printer.doprint() otherwise nested expressions won't
# work. See the example of ModOpWrong.
a, b = [printer.doprint(i) for i in self.args]
return r”\operatorname{Mod}{\left( %s,%s \right)}” % (a,b)
class ModOpWrong(Mod):
def _latex(self, printer=None):
a, b = [str(i) for i in self.args]
return r”\operatorname{Mod}{\left( %s,%s \right)}” % (a,b)
x = Symbol('x')
m = Symbol('m')
print_latex(ModOp(x, m))
print_latex(Mod(x, m))
# Nested modulo.
print_latex(ModOp(ModOp(x, m), Integer(7)))
print_latex(ModOpWrong(ModOpWrong(x, m), Integer(7)))
The main class responsible for printing is Printer (see also its source code):
class sympy.printing.printer.Printer(settings=None)
Generic printer
Its job is to provide infrastructure for implementing new printers easily.
If you want to define your custom Printer or your custom printing method for your custom
class then see the example above: printer_example (page 1048) .
Attributes
printmethod
printmethod = None
_print(expr, **kwargs)
Internal dispatcher
Tries the following concepts to print an expression:
1. Let the object print itself if it knows how.
This class implements C code printing, i.e. it converts Python expressions to strings of C code
(see also C89CodePrinter).
Usage:
Attributes
math_macros
printmethod = '_ccode'
indent_code(code)
Accepts a string of code or a list of code lines
class sympy.printing.ccode.C99CodePrinter(settings={})
Attributes
math_macros
printmethod = '_ccode'
sympy.printing.ccode.ccode(expr, assign_to=None, standard=’c99’, **settings)
Converts an expr to a string of c code
Examples
>>> from sympy import ccode, symbols, Rational, sin, ceiling, Abs, Function
>>> x, tau = symbols(”x, tau”)
>>> expr = (2*tau)**Rational(7, 2)
>>> ccode(expr)
'8*M_SQRT2*pow(tau, 7.0/2.0)'
>>> ccode(expr, math_macros={})
'8*sqrt(2)*pow(tau, 7.0/2.0)'
(continues on next page)
Simple custom printing can be defined for certain types by passing a dictionary of {“type”
: “function”} to the user_functions kwarg. Alternatively, the dictionary value can be a
list of tuples i.e. [(argument_test, cfunction_string)].
>>> custom_functions = {
... ”ceiling”: ”CEIL”,
... ”Abs”: [(lambda x: not x.is_integer, ”fabs”),
... (lambda x: x.is_integer, ”ABS”)],
... ”func”: ”f”
... }
>>> func = Function('func')
>>> ccode(func(Abs(x) + ceiling(x)), standard='C89', user_functions=custom_
,→functions)
'f(fabs(x) + CEIL(x))'
Support for loops is provided through Indexed types. With contract=True these expres-
sions will be turned into loops, whereas contract=False will just print the assignment
expression that should be looped over:
Matrices are also supported, but a MatrixSymbol of the same dimensions must be pro-
vided to assign_to. Note that any expression that can be generated normally can also
exist inside a Matrix:
sympy.printing.ccode.print_ccode(expr, **settings)
Prints C representation of the given expression.
This module contains printers for C++ code, i.e. functions to convert SymPy expressions to
strings of C++ code.
Usage:
class sympy.printing.cxxcode.CXX98CodePrinter(settings=None)
Attributes
math_macros
printmethod = '_cxxcode'
class sympy.printing.cxxcode.CXX11CodePrinter(settings=None)
Attributes
math_macros
printmethod = '_cxxcode'
sympy.printing.cxxcode.cxxcode(expr, assign_to=None, standard=’c++11’, **set-
tings)
C++ equivalent of sympy.ccode().
5.20.5 RCodePrinter
This class implements R code printing (i.e. it converts Python expressions to strings of R
code).
Usage:
Examples
>>> from sympy import rcode, symbols, Rational, sin, ceiling, Abs, Function
>>> x, tau = symbols(”x, tau”)
>>> rcode((2*tau)**Rational(7, 2))
'8*sqrt(2)*tau^(7.0/2.0)'
>>> rcode(sin(x), assign_to=”s”)
's = sin(x);'
Simple custom printing can be defined for certain types by passing a dictionary of {“type”
: “function”} to the user_functions kwarg. Alternatively, the dictionary value can be a
list of tuples i.e. [(argument_test, cfunction_string)].
>>> custom_functions = {
... ”ceiling”: ”CEIL”,
... ”Abs”: [(lambda x: not x.is_integer, ”fabs”),
... (lambda x: x.is_integer, ”ABS”)],
... ”func”: ”f”
... }
>>> func = Function('func')
>>> rcode(func(Abs(x) + ceiling(x)), user_functions=custom_functions)
'f(fabs(x) + CEIL(x))'
Support for loops is provided through Indexed types. With contract=True these expres-
sions will be turned into loops, whereas contract=False will just print the assignment
expression that should be looped over:
>>> from sympy import Eq, IndexedBase, Idx
>>> len_y = 5
>>> y = IndexedBase('y', shape=(len_y,))
>>> t = IndexedBase('t', shape=(len_y,))
>>> Dy = IndexedBase('Dy', shape=(len_y-1,))
(continues on next page)
Matrices are also supported, but a MatrixSymbol of the same dimensions must be pro-
vided to assign_to. Note that any expression that can be generated normally can also
exist inside a Matrix:
sympy.printing.rcode.print_rcode(expr, **settings)
Prints R representation of the given expression.
The fcode function translates a sympy expression into Fortran code. The main purpose is
to take away the burden of manually translating long mathematical expressions. Therefore
the resulting expression should also require no (or very little) manual tweaking to make it
compilable. The optional arguments of fcode can be used to fine-tune the behavior of fcode
in such a way that manual changes in the result are no longer needed.
sympy.printing.fcode.fcode(expr, assign_to=None, **settings)
Converts an expr to a string of fortran code
Parameters expr : Expr
A sympy expression to be converted.
assign_to : optional
When given, the argument is used as the name of the variable to
which the expression is assigned. Can be a string, Symbol, Ma-
trixSymbol, or Indexed type. This is helpful in case of line-wrapping,
or for expressions that generate multi-line statements.
precision : integer, optional
DEPRECATED. Use type_mappings instead. The precision for num-
bers such as pi [default=17].
user_functions : dict, optional
A dictionary where keys are FunctionClass instances and values are
their string representations. Alternatively, the dictionary value can
be a list of tuples i.e. [(argument_test, cfunction_string)]. See below
for examples.
human : bool, optional
If True, the result is a single string that may contain some
constant declarations for the number symbols. If False, the
Examples
>>> from sympy import fcode, symbols, Rational, sin, ceiling, floor
>>> x, tau = symbols(”x, tau”)
>>> fcode((2*tau)**Rational(7, 2))
' 8*sqrt(2.0d0)*tau**(7.0d0/2.0d0)'
>>> fcode(sin(x), assign_to=”s”)
' s = sin(x)'
Custom printing can be defined for certain types by passing a dictionary of “type” :
“function” to the user_functions kwarg. Alternatively, the dictionary value can be a list
of tuples i.e. [(argument_test, cfunction_string)].
>>> custom_functions = {
... ”ceiling”: ”CEIL”,
... ”floor”: [(lambda x: not x.is_integer, ”FLOOR1”),
... (lambda x: x.is_integer, ”FLOOR2”)]
... }
>>> fcode(floor(x) + ceiling(x), user_functions=custom_functions)
' CEIL(x) + FLOOR1(x)'
Support for loops is provided through Indexed types. With contract=True these expres-
sions will be turned into loops, whereas contract=False will just print the assignment
expression that should be looped over:
Matrices are also supported, but a MatrixSymbol of the same dimensions must be pro-
vided to assign_to. Note that any expression that can be generated normally can also
exist inside a Matrix:
sympy.printing.fcode.print_fcode(expr, **settings)
Prints the Fortran representation of the given expression.
See fcode for the meaning of the optional arguments.
class sympy.printing.fcode.FCodePrinter(settings={})
A printer to convert sympy expressions to strings of Fortran code
printmethod = '_fcode'
indent_code(code)
Accepts a string of code or a list of code lines
Two basic examples:
In case of line wrapping, it is handy to include the assignment so that lines are wrapped
properly when the assignment part is added.
Note that by default only top-level piecewise functions are supported due to the lack of a
conditional operator in Fortran 77. Inline conditionals can be supported using the merge
function introduced in Fortran 95 by setting of the kwarg standard=95:
Loops are generated if there are Indexed objects in the expression. This also requires use of
the assign_to option.
Repeated indices in an expression with Indexed objects are interpreted as summation. For
instance, code for the trace of a matrix can be generated with
By default, number symbols such as pi and E are detected and defined as Fortran parame-
ters. The precision of the constants can be tuned with the precision argument. Parameter
definitions are easily avoided using the N function.
When some functions are not part of the Fortran standard, it might be desirable to introduce
the names of user-defined functions in the Fortran expression.
However, when the user_functions argument is not provided, fcode will generate code which
assumes that a function of the same name will be provided by the user. A comment will be
added to inform the user of the issue:
By default the output is human readable code, ready for copy and paste. With the option
human=False, the return value is suitable for post-processing with source code generators
that write routines with multiple instructions. The return value is a three-tuple containing:
(i) a set of number symbols that must be defined as ‘Fortran parameters’, (ii) a list functions
that cannot be translated in pure Fortran and (iii) a string of Fortran code. A few examples:
doprint(expr)
Returns printer’s representation for expr (as a string)
sympy.printing.mathematica.mathematica_code(expr, **settings)
Converts an expr to a string of the Wolfram Mathematica code
Examples
Examples
>>> from sympy import jscode, symbols, Rational, sin, ceiling, Abs
>>> x, tau = symbols(”x, tau”)
>>> jscode((2*tau)**Rational(7, 2))
'8*Math.sqrt(2)*Math.pow(tau, 7/2)'
>>> jscode(sin(x), assign_to=”s”)
's = Math.sin(x);'
Custom printing can be defined for certain types by passing a dictionary of “type” :
“function” to the user_functions kwarg. Alternatively, the dictionary value can be a list
of tuples i.e. [(argument_test, js_function_string)].
>>> custom_functions = {
... ”ceiling”: ”CEIL”,
... ”Abs”: [(lambda x: not x.is_integer, ”fabs”),
... (lambda x: x.is_integer, ”ABS”)]
... }
>>> jscode(Abs(x) + ceiling(x), user_functions=custom_functions)
'fabs(x) + CEIL(x)'
Support for loops is provided through Indexed types. With contract=True these expres-
sions will be turned into loops, whereas contract=False will just print the assignment
expression that should be looped over:
Matrices are also supported, but a MatrixSymbol of the same dimensions must be pro-
vided to assign_to. Note that any expression that can be generated normally can also
exist inside a Matrix:
Examples
Note that element-wise (Hadamard) operations are used by default between symbols.
This is because its possible in Julia to write “vectorized” code. It is harmless if the
values are scalars.
>>> julia_code(sin(pi*x*y), assign_to=”s”)
's = sin(pi*x.*y)'
If you need a matrix product “*” or matrix power “^”, you can specify the symbol as a
MatrixSymbol.
>>> from sympy import Symbol, MatrixSymbol
>>> n = Symbol('n', integer=True, positive=True)
>>> A = MatrixSymbol('A', n, n)
>>> julia_code(3*pi*A**3)
'(3*pi)*A^3'
This class uses several rules to decide which symbol to use a product. Pure numbers
use “*”, Symbols use “.*” and MatrixSymbols use “*”. A HadamardProduct can be used
to specify componentwise multiplication “.*” of two MatrixSymbols. There is currently
there is no easy way to specify scalar symbols, so sometimes the code might have some
minor cosmetic issues. For example, suppose x and y are scalars and A is a Matrix, then
while a human programmer might write “(x^2*y)*A^3”, we generate:
>>> julia_code(x**2*y*A**3)
'(x.^2.*y)*A^3'
Matrices are supported using Julia inline notation. When using assign_to with matrices,
the name can be specified either as a string or as a MatrixSymbol. The dimensions must
align in the latter case.
Note that any expression that can be generated normally can also exist inside a Matrix:
Custom printing can be defined for certain types by passing a dictionary of “type” :
“function” to the user_functions kwarg. Alternatively, the dictionary value can be a list
of tuples i.e., [(argument_test, cfunction_string)]. This can be used to call a custom Julia
function.
Support for loops is provided through Indexed types. With contract=True these expres-
sions will be turned into loops, whereas contract=False will just print the assignment
expression that should be looped over:
Examples
Note that element-wise (Hadamard) operations are used by default between symbols.
This is because its very common in Octave to write “vectorized” code. It is harmless if
the values are scalars.
If you need a matrix product “*” or matrix power “^”, you can specify the symbol as a
MatrixSymbol.
This class uses several rules to decide which symbol to use a product. Pure numbers
use “*”, Symbols use “.*” and MatrixSymbols use “*”. A HadamardProduct can be used
to specify componentwise multiplication “.*” of two MatrixSymbols. There is currently
there is no easy way to specify scalar symbols, so sometimes the code might have some
minor cosmetic issues. For example, suppose x and y are scalars and A is a Matrix, then
while a human programmer might write “(x^2*y)*A^3”, we generate:
>>> octave_code(x**2*y*A**3)
'(x.^2.*y)*A^3'
Matrices are supported using Octave inline notation. When using assign_to with matri-
ces, the name can be specified either as a string or as a MatrixSymbol. The dimensions
must align in the latter case.
Note that any expression that can be generated normally can also exist inside a Matrix:
Custom printing can be defined for certain types by passing a dictionary of “type” :
“function” to the user_functions kwarg. Alternatively, the dictionary value can be a
list of tuples i.e., [(argument_test, cfunction_string)]. This can be used to call a custom
Octave function.
Support for loops is provided through Indexed types. With contract=True these expres-
sions will be turned into loops, whereas contract=False will just print the assignment
expression that should be looped over:
Examples
>>> from sympy import rust_code, symbols, Rational, sin, ceiling, Abs, Function
>>> x, tau = symbols(”x, tau”)
>>> rust_code((2*tau)**Rational(7, 2))
'8*1.4142135623731*tau.powf(7_f64/2.0)'
>>> rust_code(sin(x), assign_to=”s”)
's = x.sin();'
Simple custom printing can be defined for certain types by passing a dictionary of {“type”
: “function”} to the user_functions kwarg. Alternatively, the dictionary value can be a
list of tuples i.e. [(argument_test, cfunction_string)].
>>> custom_functions = {
... ”ceiling”: ”CEIL”,
... ”Abs”: [(lambda x: not x.is_integer, ”fabs”, 4),
... (lambda x: x.is_integer, ”ABS”, 4)],
... ”func”: ”f”
... }
(continues on next page)
Support for loops is provided through Indexed types. With contract=True these expres-
sions will be turned into loops, whereas contract=False will just print the assignment
expression that should be looped over:
Matrices are also supported, but a MatrixSymbol of the same dimensions must be pro-
vided to assign_to. Note that any expression that can be generated normally can also
exist inside a Matrix:
5.20.13 Gtk
You can print to a gtkmathview widget using the function print_gtk located in sympy.
printing.gtk (it requires to have installed gtkmathview and libgtkmathview-bin in some
systems).
GtkMathView accepts MathML, so this rendering depends on the MathML representation of
the expression.
Usage:
sympy.printing.gtk.print_gtk(x, start_viewer=True)
Print to Gtkmathview, a gtk widget capable of rendering MathML.
Needs libgtkmathview-bin
5.20.14 LambdaPrinter
This classes implements printing to strings that can be used by the sympy.utilities.
lambdify.lambdify() (page 1479) function.
class sympy.printing.lambdarepr.LambdaPrinter(settings=None)
This printer converts expressions into strings that can be used by lambdify.
Attributes
modules
printmethod = '_lambdacode'
sympy.printing.lambdarepr.lambdarepr(expr, **settings)
Returns a string usable for lambdifying.
5.20.15 LatexPrinter
printmethod = '_latex'
sympy.printing.latex.latex(expr, fold_frac_powers=False,
fold_func_brackets=False, fold_short_frac=None,
inv_trig_style=’abbreviated’, itex=False,
ln_notation=False, long_frac_ratio=None,
mat_delim=’[’, mat_str=None, mode=’plain’,
mul_symbol=None, order=None, sym-
bol_names=None)
Convert the given expression to LaTeX string representation.
Parameters fold_frac_powers : boolean, optional
Emit ^{p/q} instead of ^{\frac{p}{q}} for fractional powers.
fold_func_brackets : boolean, optional
Fold function brackets where applicable.
fold_short_frac : boolean, optional
Emit p / q instead of \frac{p}{q} when the denominator is simple
enough (at most two terms and no powers). The default value is True
for inline mode, False otherwise.
inv_trig_style : string, optional
How inverse trig functions should be displayed. Can be one of ab-
breviated, full, or power. Defaults to abbreviated.
itex : boolean, optional
Specifies if itex-specific syntax is used, including emitting $$...$$.
ln_notation : boolean, optional
If set to True, \ln is used instead of default \log.
long_frac_ratio : float or None, optional
The allowed ratio of the width of the numerator to the width of the
denominator before the printer breaks off long fractions. If None (the
default value), long fractions are not broken up.
mat_delim : string, optional
The delimiter to wrap around matrices. Can be one of [, (, or the
empty string. Defaults to [.
mat_str : string, optional
Which matrix environment string to emit. smallmatrix, matrix, ar-
ray, etc. Defaults to smallmatrix for inline mode, matrix for matri-
ces of no more than 10 columns, and array otherwise.
mode: string, optional
Specifies how the generated code will be delimited. mode can be one
of plain, inline, equation or equation*. If mode is set to plain,
then the resulting code will not be delimited at all (this is the de-
fault). If mode is set to inline then inline LaTeX $...$ will be used.
If mode is set to equation or equation*, the resulting code will be
enclosed in the equation or equation* environment (remember to
import amsmath for equation*), unless the itex option is set. In the
latter case, the $$...$$ syntax is used.
Notes
Not using a print statement for printing, results in double backslashes for latex com-
mands since that’s the way Python escapes backslashes in strings.
Examples
>>> from sympy import latex, pi, sin, asin, Integral, Matrix, Rational, log
>>> from sympy.abc import x, y, mu, r, tau
Basic usage:
>>> print(latex((2*tau)**Rational(7,2)))
8 \sqrt{2} \tau^{\frac{7}{2}}
Fraction options:
Multiplication options:
Trig options:
>>> print(latex(asin(Rational(7,2))))
\operatorname{asin}{\left (\frac{7}{2} \right )}
>>> print(latex(asin(Rational(7,2)), inv_trig_style=”full”))
\arcsin{\left (\frac{7}{2} \right )}
>>> print(latex(asin(Rational(7,2)), inv_trig_style=”power”))
\sin^{-1}{\left (\frac{7}{2} \right )}
Matrix options:
Logarithms:
>>> print(latex(log(10)))
\log{\left (10 \right )}
>>> print(latex(log(10), ln_notation=True))
\ln{\left (10 \right )}
latex() also supports the builtin container types list, tuple, and dictionary.
sympy.printing.latex.print_latex(expr, **settings)
Prints LaTeX representation of the given expression. Takes the same settings as latex().
5.20.16 MathMLPrinter
Attributes
printmethod
class sympy.printing.mathml.MathMLContentPrinter(settings=None)
Prints an expression to the Content MathML markup language.
References: https://www.w3.org/TR/MathML2/chapter4.html
printmethod = '_mathml_content'
mathml_tag(e)
Returns the MathML tag for an expression.
class sympy.printing.mathml.MathMLPresentationPrinter(settings=None)
Prints an expression to the Presentation MathML markup language.
References: https://www.w3.org/TR/MathML2/chapter3.html
printmethod = '_mathml_presentation'
mathml_tag(e)
Returns the MathML tag for an expression.
sympy.printing.mathml.mathml(expr, printer=’content’, **settings)
Returns the MathML representation of expr. If printer is presentation then prints Pre-
sentation MathML else prints content MathML.
sympy.printing.mathml.print_mathml(expr, printer=’content’, **settings)
Prints a pretty representation of the MathML code for expr. If printer is presentation
then prints Presentation MathML else prints content MathML.
Examples
>>> ##
>>> from sympy.printing.mathml import print_mathml
>>> from sympy.abc import x
>>> print_mathml(x+1) #doctest: +NORMALIZE_WHITESPACE
<apply>
<plus/>
<ci>x</ci>
<cn>1</cn>
</apply>
>>> print_mathml(x+1, printer='presentation')
<mrow>
<mi>x</mi>
<mo>+</mo>
<mn>1</mn>
</mrow>
5.20.17 PythonCodePrinter
Attributes
modules
class sympy.printing.pycode.NumPyPrinter(settings=None)
Numpy printer which handles vectorized piecewise functions, logical operators, etc.
Attributes
modules
sympy.printing.pycode.pycode(expr, **settings)
Converts an expr to a string of Python code
Parameters expr : Expr
A SymPy expression.
fully_qualified_modules : bool
Whether or not to write out full module names of functions (math.sin
vs. sin). default: True.
Examples
5.20.18 PythonPrinter
5.20.19 srepr
This printer generates executable code. This code satisfies the identity eval(srepr(expr))
== expr.
srepr() gives more low level textual output than repr()
Example:
srepr() gives the repr form, which is what repr() would normally give but for SymPy we
don’t actually use srepr() for __repr__ because it’s is so verbose, it is unlikely that anyone
would want it called by default. Another reason is that lists call repr on their elements, like
print([a, b, c]) calls repr(a), repr(b), repr(c). So if we used srepr for ‘‘ __repr__‘‘ any
list with SymPy objects would include the srepr form, even if we used str() or print().
class sympy.printing.repr.ReprPrinter(settings=None)
printmethod = '_sympyrepr'
emptyPrinter(expr)
The fallback printer.
reprify(args, sep)
Prints each item in args and joins them with sep.
sympy.printing.repr.srepr(expr, **settings)
return expr in repr form
5.20.20 StrPrinter
printmethod = '_sympystr'
sympy.printing.str.sstrrepr(expr, **settings)
return expr in mixed str/repr form
i.e. strings are returned in repr form with quotes, and everything else is returned in str
form.
This function could be useful for hooking into sys.displayhook
Examples
sympy.printing.tree.print_node(node)
Returns information about the “node”.
This includes class name, string representation and assumptions.
sympy.printing.tree.tree(node)
Returns a tree representation of “node” as a string.
It uses print_node() together with pprint_nodes() on node.args recursively.
See also: print_tree()
sympy.printing.tree.print_tree(node)
Prints a tree representation of “node”.
Examples
5.20.22 Preview
The ‘png’ format is considered special. For all other formats the rules are slightly differ-
ent. As an example we will take ‘dvi’ output format. If you would run
then ‘view’ will look for available ‘dvi’ viewers on your system (predefined in the function,
so it will try evince, first, then kdvi and xdvi). If nothing is found you will need to set the
viewer explicitly.
This will skip auto-detection and will run user specified ‘superior-dvi-viewer’. If ‘view’
fails to find it on your system it will gracefully raise an exception.
You may also enter ‘file’ for the viewer argument. Doing so will cause this function to
return a file object in read-only mode, if ‘filename’ is unset. However, if it was set, then
‘preview’ writes the genereted file to this filename instead.
There is also support for writing to a BytesIO like object, which needs to be passed to
the ‘outputbuffer’ argument.
The LaTeX preamble can be customized by setting the ‘preamble’ keyword argument.
This can be used, e.g., to set a different font size, use a custom documentclass or import
certain set of LaTeX packages.
If the value of ‘output’ is different from ‘dvi’ then command line options can be set (‘dviop-
tions’ argument) for the execution of the ‘dvi’+output conversion tool. These options
have to be in the form of a list of strings (see subprocess.Popen).
Additional keyword args will be passed to the latex call, e.g., the symbol_names flag.
For post-processing the generated TeX File can be written to a file by passing the desired
filename to the ‘outputTexFile’ keyword argument. To write the TeX code to a file named
“sample.tex” and run the default png viewer to display the resulting bitmap, do
sympy.printing.conventions.split_super_sub(text)
Split a symbol name into a name, superscripts and subscripts
The first part of the symbol name is considered to be its actual ‘name’, followed by
super- and subscripts. Each superscript is preceded with a “^” character or by “__”.
Each subscript is preceded by a “_” character. The three return values are the actual
name, a list with superscripts and a list with subscripts.
CodePrinter
This class is a base class for other classes that implement code-printing functionality, and
additionally lists a number of functions that cannot be easily translated to C or Fortran.
class sympy.printing.codeprinter.Assignment
Represents variable assignment for code generation.
Parameters lhs : Expr
Sympy object representing the lhs of the expression. These should
be singular objects, such as one would use in writing code. Notable
types include Symbol, MatrixSymbol, MatrixElement, and Indexed.
Types that subclass these types are also supported.
rhs : Expr
Sympy object representing the rhs of the expression. This can be any
type, provided its shape corresponds to that of the lhs. For example,
a Matrix type can be assigned to MatrixSymbol, but not to Symbol,
as the dimensions will not align.
Examples
class sympy.printing.codeprinter.CodePrinter(settings=None)
The base class for code-printing subclasses.
printmethod = '_sympystr'
exception sympy.printing.codeprinter.AssignmentError
Raised if an assignment variable for a loop is missing.
Precedence
sympy.printing.pretty.pretty_symbology.U(name)
unicode character by name or None if not found
sympy.printing.pretty.pretty_symbology.pretty_use_unicode(flag=None)
Set whether pretty-printer should use unicode by default
sympy.printing.pretty.pretty_symbology.pretty_try_use_unicode()
See if unicode output is available and leverage it if possible
sympy.printing.pretty.pretty_symbology.xstr(*args)
call str or unicode depending on current mode
The following two functions return the Unicode version of the inputted Greek letter.
sympy.printing.pretty.pretty_symbology.g(l)
sympy.printing.pretty.pretty_symbology.G(l)
sympy.printing.pretty.pretty_symbology.greek_letters = ['alpha', 'beta', 'gamma', 'delta',
list() -> new empty list list(iterable) -> new list initialized from iterable’s items
sympy.printing.pretty.pretty_symbology.digit_2txt = {'0': 'ZERO', '1': 'ONE', '2': 'TWO', '
sympy.printing.pretty.pretty_symbology.symb_2txt = {'(': 'LEFT PARENTHESIS', ')': 'RIGHT PA
The following functions return the Unicode subscript/superscript version of the character.
sympy.printing.pretty.pretty_symbology.sub = {'(': '₍', ')': '₎', '+': '₊', '-': '₋', '0':
sympy.printing.pretty.pretty_symbology.sup = {'(': '⁽', ')': '⁾', '+': '⁺', '-': '⁻', '0':
The following functions return Unicode vertical objects.
sympy.printing.pretty.pretty_symbology.xobj(symb, length)
Construct spatial object of given length.
return: [] of equal-length strings
sympy.printing.pretty.pretty_symbology.vobj(symb, height)
Construct vertical object of a given height
see: xobj
sympy.printing.pretty.pretty_symbology.hobj(symb, width)
Construct horizontal object of a given width
see: xobj
The following constants are for rendering roots and fractions.
sympy.printing.pretty.pretty_symbology.root = {2: '√', 3: '∛', 4: '∜'}
sympy.printing.pretty.pretty_symbology.VF(txt)
sympy.printing.pretty.pretty_symbology.frac = {(1, 2): '½', (1, 3): '⅓', (1, 4): '¼', (1, 5
The following constants/functions are for rendering atoms and symbols.
sympy.printing.pretty.pretty_symbology.xsym(sym)
get symbology for a ‘character’
sympy.printing.pretty.pretty_symbology.atoms_table = {'Complexes': 'ℂ', 'EmptySet': '∅', 'E
sympy.printing.pretty.pretty_symbology.pretty_atom(atom_name, de-
fault=None)
return pretty representation of an atom
sympy.printing.pretty.pretty_symbology.pretty_symbol(symb_name)
return pretty representation of a symbol
sympy.printing.pretty.pretty_symbology.annotated(letter)
Return a stylised drawing of the letter letter, together with information on how to put
annotations (super- and subscripts to the left and to the right) on it.
See pretty.py functions _print_meijerg, _print_hyper on how to use this information.
Prettyprinter by Jurjen Bos. (I hate spammers: mail me at pietjepuk314 at the reverse of
ku.oc.oohay). All objects have a method that create a “stringPict”, that can be used in the str
method for pretty printing.
Updates by Jason Gedge (email <my last name> at cs mun ca)
• terminal_string() method
• minor fixes and changes (mostly to prettyForm)
TODO:
• Allow left/center/right alignment options for above/below and top/center/bottom
alignment options for left/right
class sympy.printing.pretty.stringpict.stringPict(s, baseline=0)
An ASCII picture. The pictures are represented as a list of equal length strings.
above(*args)
Put pictures above this picture. Returns string, baseline arguments for stringPict.
Baseline is baseline of bottom picture.
below(*args)
Put pictures under this picture. Returns string, baseline arguments for stringPict.
Baseline is baseline of top picture
Examples
height()
The height of the picture in characters.
left(*args)
Put pictures (left to right) at left. Returns string, baseline arguments for stringPict.
leftslash()
Precede object by a slash of the proper size.
static next(*args)
Put a string of stringPicts next to each other. Returns string, baseline arguments
for stringPict.
parens(left=’(’, right=’)’, ifascii_nougly=False)
Put parentheses around self. Returns string, baseline arguments for stringPict.
left or right can be None or empty string which means ‘no paren from that side’
render(*args, **kwargs)
Return the string form of self.
Unless the argument line_break is set to False, it will break the expression in a form
that can be printed on the terminal without being broken up.
right(*args)
Put pictures next to this one. Returns string, baseline arguments for stringPict.
(Multiline) strings are allowed, and are given a baseline of 0.
Examples
root(n=None)
Produce a nice root symbol. Produces ugly results for big n inserts.
static stack(*args)
Put pictures on top of each other, from top to bottom. Returns string, baseline argu-
ments for stringPict. The baseline is the baseline of the second picture. Everything
is centered. Baseline is the baseline of the second picture. Strings are allowed. The
special value stringPict.LINE is a row of ‘-‘ extended to the width.
terminal_width()
Return the terminal width if possible, otherwise return 0.
width()
The width of the picture in characters.
class sympy.printing.pretty.stringpict.prettyForm(s, baseline=0, binding=0,
unicode=None)
Extension of the stringPict class that knows about basic math applications, optimizing
double minus signs.
“Binding” is interpreted as follows:
5.20.25 dotprint
in the same object, like Pow(x, x), it will still only appear only once. Hence, with
repeat=False, the number of arrows out of an object might not equal the number of
args it has).
labelfunc: How to label leaf nodes. The default is str. Another good option is
srepr. For example with str, the leaf nodes of x + 1 are labeled, x and 1. With
srepr, they are labeled Symbol('x') and Integer(1).
Additional keyword arguments are included as styles for the graph.
Examples
5.21.1 Introduction
The plotting module allows you to make 2-dimensional and 3-dimensional plots. Presently the
plots are rendered using matplotlib as a backend. It is also possible to plot 2-dimensional
plots using a TextBackend if you don’t have matplotlib.
The plotting module has the following functions:
• plot: Plots 2D line plots.
• plot_parametric: Plots 2D parametric plots.
• plot_implicit: Plots 2D implicit and region plots.
• plot3d: Plots 3D plots of functions in two variables.
Examples
Consider two Plot objects, p1 and p2. To add the second plot’s first series object to
the first, use the append method, like so:
extend(arg)
Adds all series from another plot.
Examples
Consider two Plot objects, p1 and p2. To add the second plot to the first, use the
extend method, like so:
sympy.plotting.plot.plot(*args, **kwargs)
Plots a function of a single variable and returns an instance of the Plot class (also, see
the description of the show keyword argument below).
The plotting uses an adaptive algorithm which samples recursively to accurately plot the
plot. The adaptive algorithm uses a random point near the midpoint of two points that
has to be further sampled. Hence the same plots can appear slightly different.
See also:
Plot (page 1088), LineOver1DRangeSeries.
Examples
Single Plot
No adaptive sampling.
Usage
Single Plot
plot(expr, range, **kwargs)
If the range is not specified, then a default range of (-10, 10) is used.
Multiple plots with same range.
plot(expr1, expr2, ..., range, **kwargs)
If the range is not specified, then a default range of (-10, 10) is used.
Multiple plots with different ranges.
plot((expr1, range), (expr2, range), ..., **kwargs)
Range has to be specified for every expression.
Default range may change in the future if a more advanced default range detection al-
gorithm is implemented.
Arguments
Keyword Arguments
Aesthetics options:
line_color: float. Specifies the color for the plot. See Plot to see how to set color for
the plots.
If there are multiple plots, then the same series series are applied to all the plots. If you
want to set these options separately, you can index the Plot object returned and set it.
Arguments for Plot class:
title : str. Title of the plot. It is set to the latex representation of the expression, if the
plot has only one expression.
xlabel : str. Label for the x-axis.
ylabel : str. Label for the y-axis.
xscale: {‘linear’, ‘log’} Sets the scaling of the x-axis.
yscale: {‘linear’, ‘log’} Sets the scaling if the y-axis.
axis_center: tuple of two floats denoting the coordinates of the center or {‘center’,
‘auto’}
xlim : tuple of two floats, denoting the x-axis limits.
ylim : tuple of two floats, denoting the y-axis limits.
sympy.plotting.plot.plot_parametric(*args, **kwargs)
Plots a 2D parametric plot.
The plotting uses an adaptive algorithm which samples recursively to accurately plot the
plot. The adaptive algorithm uses a random point near the midpoint of two points that
has to be further sampled. Hence the same plots can appear slightly different.
See also:
Plot (page 1088), Parametric2DLineSeries (page 1100)
Examples
Usage
Single plot.
plot_parametric(expr_x, expr_y, range, **kwargs)
If the range is not specified, then a default range of (-10, 10) is used.
Multiple plots with same range.
plot_parametric((expr1_x, expr1_y), (expr2_x, expr2_y), range, **kwargs)
If the range is not specified, then a default range of (-10, 10) is used.
Multiple plots with different ranges.
plot_parametric((expr_x, expr_y, range), ..., **kwargs)
Range has to be specified for every expression.
Default range may change in the future if a more advanced default range detection al-
gorithm is implemented.
Arguments
Keyword Arguments
Aesthetics
line_color: function which returns a float. Specifies the color for the plot. See sympy.
plotting.Plot for more details.
If there are multiple plots, then the same Series arguments are applied to all the plots.
If you want to set these options separately, you can index the returned Plot object and
set it.
Arguments for Plot class:
xlabel : str. Label for the x-axis.
ylabel : str. Label for the y-axis.
xscale: {‘linear’, ‘log’} Sets the scaling of the x-axis.
yscale: {‘linear’, ‘log’} Sets the scaling if the y-axis.
axis_center: tuple of two floats denoting the coordinates of the center or {‘center’,
‘auto’}
xlim : tuple of two floats, denoting the x-axis limits.
ylim : tuple of two floats, denoting the y-axis limits.
sympy.plotting.plot.plot3d(*args, **kwargs)
Plots a 3D surface plot.
See also:
Plot (page 1088), SurfaceOver2DRangeSeries (page 1100)
Examples
Single plot
Usage
Single plot
Arguments
Keyword Arguments
Examples
Single plot.
Multiple plots.
Usage
Single plot:
plot3d_parametric_line(expr_x, expr_y, expr_z, range, **kwargs)
If the range is not specified, then a default range of (-10, 10) is used.
Multiple plots.
plot3d_parametric_line((expr_x, expr_y, expr_z, range), ..., **kwargs)
Ranges have to be specified for every expression.
Default range may change in the future if a more advanced default range detection al-
gorithm is implemented.
Arguments
Keyword Arguments
If there are multiple plots, then the same series arguments are applied to all the plots.
If you want to set these options separately, you can index the returned Plot object and
set it.
Arguments for Plot class.
title : str. Title of the plot.
sympy.plotting.plot.plot3d_parametric_surface(*args, **kwargs)
Plots a 3D parametric surface plot.
See also:
Plot (page 1088), ParametricSurfaceSeries (page 1101)
Examples
Single plot.
>>> plot3d_parametric_surface(cos(u + v), sin(u - v), u - v,
... (u, -5, 5), (v, -5, 5))
Plot object containing:
[0]: parametric cartesian surface: (cos(u + v), sin(u - v), u - v) for u over (-5.
,→0, 5.0) and v over (-5.0, 5.0)
Usage
Single plot.
plot3d_parametric_surface(expr_x, expr_y, expr_z, range_u, range_v,
**kwargs)
If the ranges is not specified, then a default range of (-10, 10) is used.
Multiple plots.
plot3d_parametric_surface((expr_x, expr_y, expr_z, range_u, range_v), ...
, **kwargs)
Ranges have to be specified for every expression.
Default range may change in the future if a more advanced default range detection al-
gorithm is implemented.
Arguments
Keyword Arguments
Examples
Plot expressions:
>>> from sympy import plot_implicit, cos, sin, symbols, Eq, And
>>> x, y = symbols('x y')
Plotting regions.
When plotting an expression with a single variable (y - 1, for example), specify the x or
the y variable explicitly:
Arguments
If neither x_var nor y_var are given then the free symbols in the expression will be
assigned in the order they are sorted.
The following keyword arguments can also be used:
• adaptive. Boolean. The default value is set to True. It has to be set to False
if you want to use a mesh grid.
• depth integer. The depth of recursion for adaptive mesh grid. Default value
is 0. Takes value in the range (0, 4).
• points integer. The number of points if adaptive mesh grid is not used. De-
fault value is 200.
• title string .The title for the plot.
• xlabel string. The label for the x-axis
• ylabel string. The label for the y-axis
Aesthetics options:
• line_color: float or string. Specifies the color for the plot. See Plot to see
how to set color for the plots.
plot_implicit, by default, uses interval arithmetic to plot functions. If the expression can-
not be plotted using interval arithmetic, it defaults to a generating a contour using a mesh
grid of fixed number of points. By setting adaptive to False, you can force plot_implicit
to use the mesh grid. The mesh grid method can be effective when adaptive plotting
using interval arithmetic, fails to plot with small line width.
class sympy.plotting.plot.BaseSeries
Base class for the data objects containing stuff to be plotted.
The backend should check if it supports the data series that it’s given. (eg TextBackend
supports only LineOver1DRange). It’s the backend responsibility to know how to use the
class of data series that it’s given.
Some data series classes are grouped (using a class attribute like is_2Dline) according
to the api they present (based only on convention). The backend is not obliged to use
that api (eg. The LineOver1DRange belongs to the is_2Dline group and presents the
get_points method, but the TextBackend does not use the get_points method).
class sympy.plotting.plot.Line2DBaseSeries
A base class for 2D lines.
• adding the label, steps and only_integers options
• making is_2Dline true
• defining get_segments and get_color_array
class sympy.plotting.plot.LineOver1DRangeSeries(expr, var_start_end, **kwargs)
Representation for a line consisting of a SymPy expression over a range.
get_segments()
Adaptively gets segments for plotting.
The adaptive sampling is done by recursively checking if three points are almost
collinear. If they are not collinear, then more points are added between those points.
References
References
class sympy.plotting.plot.Line3DBaseSeries
A base class for 3D lines.
Most of the stuff is derived from Line2DBaseSeries.
class sympy.plotting.plot.Parametric3DLineSeries(expr_x, expr_y, expr_z,
var_start_end, **kwargs)
Representation for a 3D line consisting of two parametric sympy expressions and a range.
class sympy.plotting.plot.SurfaceBaseSeries
A base class for 3D surfaces.
This is the documentation for the old plotting module that uses pyglet. This module has some
limitations and is not actively developed anymore. For an alternative you can look at the new
plotting module.
The pyglet plotting module can do nice 2D and 3D plots that can be controlled by console
commands as well as keyboard and mouse, with the only dependency being pyglet.
Here is the simplest usage:
$ python -i examples/pyglet_plotting.py
Camera Keys
Sensitivity Modifier SHIFT
Zoom R and F, Page Up and Down, Numpad + and -
Rotate View X,Y axis Arrow Keys, A,S,D,W, Numpad 4,6,8,2
Rotate View Z axis Q and E, Numpad 7 and 9
Rotate Ordinate Z axis Z and C, Numpad 1 and 3
View XY F1
View XZ F2
View YZ F3
View Perspective F4
Reset X, Numpad 5
Axes Keys
Toggle Visible F5
Toggle Colors F6
Window Keys
Close ESCAPE
Screenshot F8
The mouse can be used to rotate, zoom, and translate by dragging the left, middle, and right
mouse buttons respectively.
Plot supports several curvilinear coordinate modes, and they are independent for each plot-
ted function. You can specify a coordinate mode explicitly with the ‘mode’ named argument,
but it can be automatically determined for cartesian or parametric plots, and therefore must
only be specified for polar, cylindrical, and spherical modes.
Specifically, Plot(function arguments) and Plot.__setitem__(i, function arguments)
(accessed using array-index syntax on the Plot instance) will interpret your arguments as a
cartesian plot if you provide one function and a parametric plot if you provide two or three
functions. Similarly, the arguments will be interpreted as a curve is one variable is used, and
a surface if two are used.
Supported mode names by number of variables:
• 1 (curves): parametric, cartesian, polar
• 2 (surfaces): parametric, cartesian, cylindrical, spherical
Note that function parameters are given as option strings of the form “key1=value1; key2 =
value2” (spaces are truncated). Keyword arguments given directly to plot apply to the plot
itself.
The basic format for variable intervals is [var, min, max, steps]. However, the syntax is quite
flexible, and arguments not specified are taken from the defaults for the current coordinate
mode:
>>> p = Plot(visible=False)
>>> f = x**2
>>> p[1] = f
>>> p[2] = f.diff(x)
>>> p[3] = f.diff(x).diff(x)
>>> p
[1]: x**2, 'mode=cartesian'
[2]: 2*x, 'mode=cartesian'
[3]: 2, 'mode=cartesian'
>>> p.show()
>>> p.clear()
>>> p
<blank plot>
>>> p[1] = x**2+y**2
>>> p[1].style = 'solid'
>>> p[2] = -x**2-y**2
>>> p[2].style = 'wireframe'
>>> p[1].color = z, (0.4,0.4,0.9), (0.9,0.4,0.4)
>>> p[1].style = 'both'
>>> p[2].style = 'both'
>>> p.close()
The following code plots a saddle and color it by the magnitude of its gradient:
>>> fz = x**2-y**2
>>> Fx, Fy, Fz = fz.diff(x), fz.diff(y), 0
>>> p[1] = fz, 'style=solid'
>>> p[1].color = (Fx**2 + Fy**2 + Fz**2)**(0.5)
When not specified explicitly, the default color gradient is f(0.0)=(0.4,0.4,0.4) ->
f(1.0)=(0.9,0.9,0.9). In our case, everything is gray-scale because we have applied the de-
fault color gradient uniformly for each color component. When defining a color scheme in
this way, you might want to supply a color gradient as well:
The other way to specify a color scheme is to give a separate function for each component r,
g, b. With this syntax, the default color scheme is defined:
This maps z->red, y->green, and x->blue. In some cases, you might prefer to use the following
alternative syntax:
You can still use multi-step gradients with three-function color schemes.
The plotting module is capable of plotting some 2D geometric entities like line, circle and
ellipse. The following example plots a circle and a tangent line at a random point on the
ellipse.
In [1]: p = Plot(axes='label_axes=True')
In [2]: c = Circle(Point(0,0), 1)
In [3]: t = c.tangent_line(c.random_point())
In [4]: p[0] = c
In [5]: p[1] = t
Plotting polygons (Polygon, RegularPolygon, Triangle) are not supported directly. However a
polygon can be plotted through a loop as follows.
In [6]: p = Plot(axes='label_axes=True')
In [7]: t = RegularPolygon(Point(0,0), 1, 5)
5.23.1 Contents
Ask
References
[R1]
Examples
antihermitian
Antihermitian predicate.
Q.antihermitian(x) is true iff x belongs to the field of antihermitian operators, i.e.,
operators in the form x*I, where x is Hermitian.
References
[R2]
bounded
See documentation of Q.finite.
commutative
Commutative predicate.
ask(Q.commutative(x)) is true iff x commutes with any other object with respect
to multiplication operation.
complex
Complex number predicate.
Q.complex(x) is true iff x belongs to the set of complex numbers. Note that every
complex number is finite.
References
[R3]
Examples
complex_elements
Complex elements matrix predicate.
Q.complex_elements(x) is true iff all the elements of x are complex numbers.
Examples
composite
Composite number predicate.
ask(Q.composite(x)) is true iff x is a positive integer and has at least one positive
divisor other than 1 and the number itself.
Examples
diagonal
Diagonal matrix predicate.
Q.diagonal(x) is true iff x is a diagonal matrix. A diagonal matrix is a matrix in
which the entries outside the main diagonal are all zero.
References
[R4]
Examples
even
Even number predicate.
ask(Q.even(x)) is true iff x belongs to the set of even integers.
Examples
extended_real
Extended real predicate.
Q.extended_real(x) is true iff x is a real number or {−∞, ∞}.
See documentation of Q.real for more information about related facts.
Examples
finite
Finite predicate.
Q.finite(x) is true if x is neither an infinity nor a NaN. In other words, ask(Q.
finite(x)) is true for all x having a bounded absolute value.
References
[R5]
Examples
fullrank
Fullrank matrix predicate.
Q.fullrank(x) is true iff x is a full rank matrix. A matrix is full rank if all rows and
columns of the matrix are linearly independent. A square matrix is full rank iff its
determinant is nonzero.
Examples
hermitian
Hermitian predicate.
ask(Q.hermitian(x)) is true iff x belongs to the set of Hermitian operators.
References
[R6]
imaginary
Imaginary number predicate.
References
[R7]
Examples
infinite
Infinite number predicate.
Q.infinite(x) is true iff the absolute value of x is infinity.
infinitesimal
See documentation of Q.zero.
infinity
See documentation of Q.infinite.
integer
Integer predicate.
Q.integer(x) is true iff x belongs to the set of integer numbers.
References
[R8]
Examples
integer_elements
Integer elements matrix predicate.
Q.integer_elements(x) is true iff all the elements of x are integers.
Examples
invertible
Invertible matrix predicate.
Q.invertible(x) is true iff x is an invertible matrix. A square matrix is called in-
vertible only if its determinant is 0.
References
[R9]
Examples
irrational
Irrational number predicate.
Q.irrational(x) is true iff x is any real number that cannot be expressed as a ratio
of integers.
References
[R10]
Examples
is_true
Generic predicate.
ask(Q.is_true(x)) is true iff x is true. This only makes sense if x is a predicate.
Examples
lower_triangular
Lower triangular matrix predicate.
A matrix M is called lower triangular matrix if aij = 0 for i > j.
References
[R11]
Examples
negative
Negative number predicate.
Q.negative(x) is true iff x is a real number and x < 0, that is, it is in the interval
(−∞, 0). Note in particular that negative infinity is not negative.
A few important facts about negative numbers:
• Note that Q.nonnegative and ~Q.negative are not the same thing. ~Q.
negative(x) simply means that x is not negative, whereas Q.nonnegative(x)
means that x is real and not negative, i.e., Q.nonnegative(x) is logically equiv-
alent to Q.zero(x) | Q.positive(x). So for example, ~Q.negative(I) is true,
whereas Q.nonnegative(I) is false.
• See the documentation of Q.real for more information about related facts.
Examples
nonnegative
Nonnegative real number predicate.
ask(Q.nonnegative(x)) is true iff x belongs to the set of positive numbers including
zero.
• Note that Q.nonnegative and ~Q.negative are not the same thing. ~Q.
negative(x) simply means that x is not negative, whereas Q.nonnegative(x)
means that x is real and not negative, i.e., Q.nonnegative(x) is logically equiv-
alent to Q.zero(x) | Q.positive(x). So for example, ~Q.negative(I) is true,
whereas Q.nonnegative(I) is false.
Examples
nonpositive
Nonpositive real number predicate.
ask(Q.nonpositive(x)) is true iff x belongs to the set of negative numbers includ-
ing zero.
• Note that Q.nonpositive and ~Q.positive are not the same thing. ~Q.
positive(x) simply means that x is not positive, whereas Q.nonpositive(x)
means that x is real and not positive, i.e., Q.nonpositive(x) is logically equiva-
lent to Q.negative(x)|Q.zero(x). So for example, ~Q.positive(I) is true, whereas
Q.nonpositive(I) is false.
Examples
nonzero
Nonzero real number predicate.
ask(Q.nonzero(x)) is true iff x is real and x is not zero. Note in particular that
Q.nonzero(x) is false if x is not real. Use ~Q.zero(x) if you want the negation of
being zero without any real assumptions.
A few important facts about nonzero numbers:
• Q.nonzero is logically equivalent to Q.positive | Q.negative.
• See the documentation of Q.real for more information about related facts.
Examples
normal
Normal matrix predicate.
A matrix is normal if it commutes with its conjugate transpose.
References
[R12]
Examples
odd
Odd number predicate.
ask(Q.odd(x)) is true iff x belongs to the set of odd numbers.
Examples
orthogonal
Orthogonal matrix predicate.
Q.orthogonal(x) is true iff x is an orthogonal matrix. A square matrix M is an or-
thogonal matrix if it satisfies M^TM = MM^T = I where M^T is the transpose matrix of
M and I is an identity matrix. Note that an orthogonal matrix is necessarily invertible.
References
[R13]
Examples
positive
Positive real number predicate.
Q.positive(x) is true iff x is real and x > 0, that is if x is in the interval (0, ∞). In
particular, infinity is not positive.
A few important facts about positive numbers:
• Note that Q.nonpositive and ~Q.positive are not the same thing. ~Q.
positive(x) simply means that x is not positive, whereas Q.nonpositive(x)
means that x is real and not positive, i.e., Q.nonpositive(x) is logically equiva-
lent to Q.negative(x)|Q.zero(x). So for example, ~Q.positive(I) is true, whereas
Q.nonpositive(I) is false.
• See the documentation of Q.real for more information about related facts.
Examples
positive_definite
Positive definite matrix predicate.
If M is a :math:n \times n symmetric real matrix, it is said to be positive definite if
Z T M Z is positive for every non-zero column vector Z of n real numbers.
References
[R14]
Examples
prime
Prime number predicate.
ask(Q.prime(x)) is true iff x is a natural number greater than 1 that has no positive
divisors other than 1 and the number itself.
Examples
rational
Rational number predicate.
Q.rational(x) is true iff x belongs to the set of rational numbers.
References
https://en.wikipedia.org/wiki/Rational_number
Examples
real
Real number predicate.
Q.real(x) is true iff x is a real number, i.e., it is in the interval (−∞, ∞). Note that, in
particular the infinities are not real. Use Q.extended_real if you want to consider
those as well.
A few important facts about reals:
• Every real number is positive, negative, or zero. Furthermore, because these
sets are pairwise disjoint, each real number is exactly one of those three.
• Every real number is also complex.
• Every real number is finite.
• Every real number is either rational or irrational.
• Every real number is either algebraic or transcendental.
• The facts Q.negative, Q.zero, Q.positive, Q.nonnegative, Q.nonpositive, Q.
nonzero, Q.integer, Q.rational, and Q.irrational all imply Q.real, as do all
facts that imply those facts.
• The facts Q.algebraic, and Q.transcendental do not imply Q.real; they imply
Q.complex. An algebraic or transcendental number may or may not be real.
• The “non” facts (i.e., Q.nonnegative, Q.nonzero, Q.nonpositive and Q.
noninteger) are not equivalent to not the fact, but rather, not the fact and
Q.real. For example, Q.nonnegative means ~Q.negative & Q.real. So for
example, I is not nonnegative, nonzero, or nonpositive.
References
[R15]
Examples
real_elements
Real elements matrix predicate.
Q.real_elements(x) is true iff all the elements of x are real numbers.
Examples
singular
Singular matrix predicate.
A matrix is singular iff the value of its determinant is 0.
References
[R16]
Examples
square
Square matrix predicate.
Q.square(x) is true iff x is a square matrix. A square matrix is a matrix with the
same number of rows and columns.
References
[R17]
Examples
symmetric
Symmetric matrix predicate.
Q.symmetric(x) is true iff x is a square matrix and is equal to its transpose. Every
square diagonal matrix is a symmetric matrix.
References
[R18]
Examples
transcendental
Transcedental number predicate.
Q.transcendental(x) is true iff x belongs to the set of transcendental numbers. A
transcendental number is a real or complex number that is not algebraic.
triangular
Triangular matrix predicate.
Q.triangular(X) is true if X is one that is either lower triangular or upper triangular.
References
[R19]
Examples
unit_triangular
Unit triangular matrix predicate.
A unit triangular matrix is a triangular matrix with 1s on the diagonal.
Examples
unitary
Unitary matrix predicate.
Q.unitary(x) is true iff x is a unitary matrix. Unitary matrix is an analogue to
orthogonal matrix. A square matrix M with complex elements is unitary if :math:M^TM
= MM^T= I where :math:M^T is the conjugate transpose matrix of M.
References
[R20]
Examples
upper_triangular
Upper triangular matrix predicate.
A matrix M is called upper triangular matrix if Mij = 0 for i < j.
References
[R21]
Examples
zero
Zero number predicate.
ask(Q.zero(x)) is true iff the value of x is zero.
Examples
Examples
Remarks Relations in assumptions are not implemented (yet), so the following will not
give a meaningful result.
sympy.assumptions.ask.ask_full_inference(proposition, assumptions,
known_facts_cnf)
Method for inferring properties about objects.
sympy.assumptions.ask.compute_known_facts(known_facts, known_facts_keys)
Compute the various forms of knowledge compilation used by the assumptions system.
This function is typically applied to the results of the get_known_facts and
get_known_facts_keys functions defined at the bottom of this file.
sympy.assumptions.ask.register_handler(key, handler)
Register a handler in the ask system. key must be a string and handler a class inheriting
from AskHandler:
sympy.assumptions.ask.remove_handler(key, handler)
Removes a handler from the ask system. Same syntax as register_handler
Assume
class sympy.assumptions.assume.AppliedPredicate
The class of expressions resulting from applying a Predicate.
Examples
arg
Return the expression used by this assumption.
Examples
class sympy.assumptions.assume.AssumptionsContext
Set representing assumptions.
This is used to represent global assumptions, but you can also use this class to create
your own local assumptions contexts. It is basically a thin wrapper to Python’s set, so
see its documentation for advanced usage.
Examples
add(*assumptions)
Add an assumption.
class sympy.assumptions.assume.Predicate
A predicate is a function that returns a boolean value.
Predicates merely wrap their argument and remain unevaluated:
>>> from sympy import Q, ask
>>> type(Q.prime)
<class 'sympy.assumptions.assume.Predicate'>
>>> Q.prime.name
(continues on next page)
To obtain the truth value of an expression containing predicates, use the function ask:
>>> ask(Q.prime(7))
True
eval(expr, assumptions=True)
Evaluate self(expr) under the given assumptions.
This uses only direct resolution methods, not logical inference.
sympy.assumptions.assume.assuming(*assumptions)
Context manager for assumptions
Examples
Refine
sympy.assumptions.refine.refine(expr, assumptions=True)
Simplify an expression using assumptions.
Gives the form of expr that would be obtained if symbols in it were replaced by explicit
numerical expressions satisfying the assumptions.
Examples
sympy.assumptions.refine.refine_Pow(expr, assumptions)
Handler for instances of Pow.
sympy.assumptions.refine.refine_Relational(expr, assumptions)
Handler for Relational
sympy.assumptions.refine.refine_abs(expr, assumptions)
Handler for the absolute value.
Examples
sympy.assumptions.refine.refine_atan2(expr, assumptions)
Handler for the atan2 function
Examples
Handlers
Contents
Calculus
This module contains query handlers responsible for calculus queries: infinitesimal, finite,
etc.
class sympy.assumptions.handlers.calculus.AskFiniteHandler
Handler for key ‘finite’.
Test that an expression is bounded respect to all its variables.
Examples of usage:
B U ?
‘+’ ‘-‘ ‘x’ ‘+’ ‘-‘ ‘x’
B B U ?
U ‘+’ U ? ? U ? ?
‘-‘ ? U ? ? U ?
‘x’ ? ?
? ?
• ‘B’ = Bounded
• ‘U’ = Unbounded
• ‘?’ = unknown boundedness
• ‘+’ = positive sign
• ‘-‘ = negative sign
• ‘x’ = sign unknown
B U ?
s /s
B B U ?
U U U ?
? ?
• B = Bounded
• U = Unbounded
• ? = unknown boundedness
• s = signed (hence nonzero)
• /s = not signed
Examples
nTheory
Handlers for keys related to number theory: prime, even, odd, etc.
class sympy.assumptions.handlers.ntheory.AskOddHandler
Handler for key ‘odd’ Test that an expression represents an odd number
class sympy.assumptions.handlers.ntheory.AskPrimeHandler
Handler for key ‘prime’ Test that an expression represents a prime number. When the
expression is a number the result, when True, is subject to the limitations of isprime()
which is used to return the result.
static Pow(expr, assumptions)
Integer**Integer -> !Prime
Order
Examples
Sets
Syntax
• ask(proposition)
• ask(proposition, assumptions)
where proposition is any boolean expression
Examples
Remarks Relations in assumptions are not implemented (yet), so the following will not
give a meaningful result.
5.23.2 Querying
ask’s optional second argument should be a boolean expression involving assumptions about
objects in expr. Valid values include:
• Q.integer(x)
• Q.positive(x)
• Q.integer(x) & Q.positive(x)
• etc.
Q is an object holding known predicates.
See documentation for the logic module for a complete list of valid boolean expressions.
You can also define a context so you don’t have to pass that argument each time to function
ask(). This is done by using the assuming context manager from module sympy.assumptions.
5.23.3 Design
Each time ask is called, the appropriate Handler for the current key is called. This is always
a subclass of sympy.assumptions.AskHandler. Its classmethods have the names of the classes
it supports. For example, a (simplified) AskHandler for the ask ‘positive’ would look like this:
class AskPositiveHandler(CommonHandler):
def Mul(self):
# return True if all argument's in self.expr.args are positive
...
def Add(self):
for arg in self.expr.args:
if not ask(arg, positive, self.assumptions):
break
else:
# if all argument's are positive
return True
...
The .Mul() method is called when self.expr is an instance of Mul, the Add method would be
called when self.expr is an instance of Add and so on.
5.23.4 Extensibility
You can define new queries or support new types by subclassing sympy.assumptions.AskHandler
and registering that handler for a particular key by calling register_handler:
sympy.assumptions.ask.register_handler(key, handler)
Register a handler in the ask system. key must be a string and handler a class inheriting
from AskHandler:
On queries that involve symbolic coefficients, logical inference is used. Work on improving
satisfiable function (sympy.logic.inference.satisfiable) should result in notable speed improve-
ments.
Logic inference used in one ask could be used to speed up further queries, and current
system does not take advantage of this. For example, a truth maintenance system (http:
//en.wikipedia.org/wiki/Truth_maintenance_system) could be implemented.
5.23.6 Misc
You can find more examples in the in the form of test under directory
sympy/assumptions/tests/
Term rewriting is a very general class of functionalities which are used to convert expres-
sions of one type in terms of expressions of different kind. For example expanding, com-
bining and converting expressions apply to term rewriting, and also simplification routines
can be included here. Currently SymPy has several functions and basic built-in methods for
performing various types of rewriting.
5.24.1 Expanding
The simplest rewrite rule is expanding expressions into a _sparse_ form. Expanding has sev-
eral flavors and include expanding complex valued expressions, arithmetic expand of prod-
ucts and powers but also expanding functions in terms of more general functions is possible.
Below are listed all currently available expand rules.
Expanding of arithmetic expressions involving products and powers:
>>> from sympy import *
>>> x, y, z = symbols('x,y,z')
>>> ((x + y)*(x - y)).expand(basic=True)
(continues on next page)
Arithmetic expand is done by default in expand() so the keyword basic can be omitted. How-
ever you can set basic=False to avoid this type of expand if you use rules described below.
This give complete control on what is done with the expression.
Another type of expand rule is expanding complex valued expressions and putting them into
a normal form. For this complex keyword is used. Note that it will always perform arithmetic
expand to obtain the desired normal form:
>>> (x + I*y).expand(complex=True)
re(x) + I*re(y) + I*im(x) - im(y)
Note also that the same behavior can be obtained by using as_real_imag() method. However
it will return a tuple containing the real part in the first place and the imaginary part in the
other. This can be also done in a two step process by using collect function:
>>> (x + I*y).as_real_imag()
(re(x) - im(y), re(y) + im(x))
There is also possibility for expanding expressions in terms of expressions of different kind.
This is very general type of expanding and usually you would use rewrite() to do specific
type of rewrite:
>>> GoldenRatio.expand(func=True)
1/2 + sqrt(5)/2
However, these optimizations can be very slow for large expressions. Moreover, if speed is a
concern, one can pass the option order='none'. Order of terms will then be dependent on
hashing algorithm implementation, but speed will be greatly improved.
More information:
sympy.simplify.cse_main.cse(exprs, symbols=None, optimizations=None, postpro-
cess=None, order=’canonical’, ignore=())
Perform common subexpression elimination on an expression.
Parameters exprs : list of sympy expressions, or a single sympy expression
The expressions to reduce.
symbols : infinite iterator yielding unique Symbols
The symbols used to label the common subexpressions which are
pulled out. The numbered_symbols generator is useful. The default
is a stream of symbols of the form “x0”, “x1”, etc. This must be an
infinite iterator.
optimizations : list of (callable, callable) pairs
The (preprocessor, postprocessor) pairs of external optimization
functions. Optionally ‘basic’ can be passed for a set of predefined
basic optimizations. Such ‘basic’ optimizations were used by default
in old implementation, however they can be really slow on larger ex-
pressions. Now, no pre or post optimizations are made by default.
postprocess : a function which accepts the two return values of cse and
returns the desired form of output from cse, e.g. if you want the
replacements reversed the function might be the following lambda:
lambda r, e: return reversed(r), e
order : string, ‘none’ or ‘canonical’
The order by which Mul and Add arguments are processed. If set to
‘canonical’, arguments will be canonically ordered. If set to ‘none’,
ordering will be faster but dependent on expressions hashes, thus
machine dependent and variable. For large expressions where speed
is a concern, use the setting order=’none’.
ignore : iterable of Symbols
Examples
The series module implements series expansions as a function and many related functions.
5.25.1 Contents
Series Expansions
Limits
Notes
First we try some heuristics for easy and frequent cases like “x”, “1/x”, “x**2” and sim-
ilar, so that it’s fast. For all other cases, we use the Gruntz algorithm (see the gruntz()
function).
Examples
class sympy.series.limits.Limit
Represents an unevaluated limit.
Examples
doit(**hints)
Evaluates limit
As is explained above, the workhorse for limit computations is the function gruntz() which
implements Gruntz’ algorithm for computing limits.
This section explains the basics of the algorithm used for computing limits. Most of the time
the limit() function should just work. However it is still useful to keep in mind how it is
implemented in case something does not work as expected.
First we define an ordering on functions. Suppose f (x) and g(x) are two real-valued functions
such that limx→∞ f (x) = ∞ and similarly limx→∞ g(x) = ∞. We shall say that f (x) dominates
(x)a
g(x), written f (x) ≻ g(x), if for all a, b ∈ R>0 we have limx→∞ fg(x) b = ∞. We also say that f (x) and
g(x) are of the same comparability class if neither f (x) ≻ g(x) nor g(x) ≻ f (x) and shall denote
it as f (x) ≍ g(x).
Note that whenever a, b ∈ R>0 then af (x)b ≍ f (x), and we shall use this to extend the definition
of ≻ to all functions which tend to 0 or ±∞ as x → ∞. Thus we declare that f (x) ≍ 1/f (x) and
f (x) ≍ −f (x).
It is easy to show the following examples:
• ex ≻ xm
2
• ex ≻ emx
x m
• ee ≻ ex
• xm ≍ xn
1
• ex+ x ≍ ex+log x ≍ ex .
From the above definition, it is possible to prove the following property:
Suppose ω, g1 , g2 , . . . are functions of x, limx→∞ ω = 0 and ω ≻ gi for all i. Let c1 , c2 , . . . ∈
R with c1 < c2 < · · ·.
∑
Then limx→∞ i gi ω ci = limx→∞ g1 ω c1 .
For g1 = g and ω as above we also have the following easy result:
• limx→∞ gω c = 0 for c > 0
• limx→∞ gω c = ±∞ for c < 0, where the sign is determined by the (eventual) sign of g
• limx→∞ gω 0 = limx→∞ g.
Using these results yields the following strategy for computing limx→∞ f (x):
1. Find the set of most rapidly varying subexpressions (MRV set) of f (x). That is, from the
set of all subexpressions of f (x), find the elements that are maximal under the relation
≻.
2. Choose a function ω that is in the same comparability class as the elements in the MRV
set, such that limx→∞ ω = 0.
3. Expand f (x) as a series in ω in such a way that the antecedents of the above theorem are
satisfied.
4. Apply the theorem and conclude the computation of limx→∞ f (x), possibly by recursively
working on g1 (x).
Notes
This exposition glossed over several details. Many are described in the file gruntz.py, and all
can be found in Gruntz’ very readable thesis. The most important points that have not been
explained are:
1. Given f(x) and g(x), how do we determine if f (x) ≻ g(x), g(x) ≻ f (x) or g(x) ≍ f (x)?
2. How do we find the MRV set of an expression?
3. How do we compute series expansions?
4. Why does the algorithm terminate?
If you are interested, be sure to take a look at Gruntz Thesis.
Reference
e1 ---> e2 ---> e3
\
-> e4
Here we can do e1, e2, e3, e4 or e1, e2, e4, e3. To do this we assemble the nodes into a
tree, and sort them by height.
This function builds the tree, rewrites then sorts the nodes.
sympy.series.gruntz.mrv_leadterm(e, x)
Returns (c0, e0) for e.
sympy.series.gruntz.calculate_series(e, x, logx=None)
Calculates at least one term of the series of “e” in “x”.
This is a place that fails most often, so it is in its own function.
sympy.series.gruntz.limitinf(e, x)
Limit e(x) for x-> oo
sympy.series.gruntz.sign(e, x)
Returns a sign of an expression e(x) for x->oo.
e > 0 for x sufficiently large ... 1
e == 0 for x sufficiently large ... 0
e < 0 for x sufficiently large ... -1
The result of this function is currently undefined if e changes sign arbitrarily often for
arbitrarily large x (e.g. sin(x)).
Note that this returns zero only if e is constantly zero for x sufficiently large. [If e is
constant, of course, this is just the same thing as the sign of e.]
sympy.series.gruntz.mrv(e, x)
Returns a SubsSet of most rapidly varying (mrv) subexpressions of ‘e’, and e rewritten
in terms of these
sympy.series.gruntz.mrv_max1(f, g, exps, x)
Computes the maximum of two sets of expressions f and g, which are in the same com-
parability class, i.e. mrv_max1() compares (two elements of) f and g and returns the set,
which is in the higher comparability class of the union of both, if they have the same
order of variation. Also returns exps, with the appropriate substitutions made.
sympy.series.gruntz.mrv_max3(f, expsf, g, expsg, union, expsboth, x)
Computes the maximum of two sets of expressions f and g, which are in the same compa-
rability class, i.e. max() compares (two elements of) f and g and returns either (f, expsf)
[if f is larger], (g, expsg) [if g is larger] or (union, expsboth) [if f, g are of the same class].
class sympy.series.gruntz.SubsSet
Stores (expr, dummy) pairs, and how to rewrite expr-s.
The gruntz algorithm needs to rewrite certain expressions in term of a new variable w.
We cannot use subs, because it is just too smart for us. For example:
> Omega=[exp(exp(_p - exp(-_p))/(1 - 1/_p)), exp(exp(_p))]
> O2=[exp(-exp(_p) + exp(-exp(-_p))*exp(_p)/(1 - 1/_p))/_w, 1/_w]
> e = exp(exp(_p - exp(-_p))/(1 - 1/_p)) - exp(exp(_p))
> e.subs(Omega[0],O2[0]).subs(Omega[1],O2[1])
-1/w + exp(exp(p)*exp(-exp(-p))/(1 - 1/p))
The mrv set is {exp(x), exp(-x), exp(x - exp(-x))}. We introduce corresponding dummy
variables d1, d2, d3 and rewrite:
d3 + d1 + x.
This class first of all keeps track of the mapping expr->variable, i.e. will at this stage be
a dictionary:
{exp(x): d1, exp(-x): d2, exp(x - exp(-x)): d3}.
[It turns out to be more convenient this way round.] But sometimes expressions in the
mrv set have other expressions from the mrv set as subexpressions, and we need to keep
track of that as well. In this case, d3 is really exp(x - d2), so rewrites at this stage is:
{d3: exp(x-d2)}.
The function rewrite uses all this information to correctly rewrite our expression in terms
of w. In this case w can be chosen to be exp(-x), i.e. d2. The correct rewriting then is:
exp(-w)/w + 1/w + x.
copy()
Create a shallow copy of SubsSet
do_subs(e)
Substitute the variables with expressions
meets(s2)
Tell whether or not self and s2 have non-empty intersection
union(s2, exps=None)
Compute the union of self and s2, adjusting exps
This is achieved by creating a wrapper around Basic.series(). This allows for the use of se-
ries(x*cos(x),x), which is possibly more intuitive than (x*cos(x)).series(x).
Examples
Reference
Order Terms
This module also implements automatic keeping track of the order of your expansion.
Examples
Reference
class sympy.series.order.Order
Represents the limiting behavior of some function
The order of a function characterizes the function based on the limiting behavior of the
function as it goes to some limit. Only taking the limit point to be a number is currently
supported. This is expressed in big O notation [R497].
The formal definition for the order of a function g(x) about a point a is such that g(x) =
O(f (x)) as x → a if and only if for any δ > 0 there exists a M > 0 such that |g(x)| ≤ M |f (x)|
for |x − a| < δ. This is equivalent to limx→a sup |g(x)/f (x)| < ∞.
Let’s illustrate it on the following example by taking the expansion of sin(x) about 0:
where in this case O(x5 ) = x5 /5! − x7 /7! + · · ·. By the definition of O, for any δ > 0 there is
an M such that:
Notes
References
[R497]
Examples
>>> O(x)*x
O(x**2)
>>> O(x) - O(x)
O(x)
>>> O(cos(x))
O(1)
>>> O(cos(x), (x, pi/2))
O(x - pi/2, (x, pi/2))
contains(expr)
Return True if expr belongs to Order(self.expr, *self.variables). Return False if self
belongs to expr. Return None if the inclusion relation cannot be determined (e.g.
when self and expr have different symbols).
Series Acceleration
TODO
Reference
sympy.series.acceleration.richardson(A, k, n, N)
Calculate an approximation for lim k->oo A(k) using Richardson extrapolation with the
terms A(n), A(n+1), …, A(n+N+1). Choosing N ~= 2*n often gives good results.
A simple example is to calculate exp(1) using the limit definition. This limit converges
slowly; n = 100 only produces two accurate digits:
Richardson extrapolation with 11 appropriately chosen terms gives a value that is accu-
rate to the indicated precision:
sympy.series.acceleration.shanks(A, k, n, m=1)
Calculate an approximation for lim k->oo A(k) using the n-term Shanks transfor-
mation S(A)(n). With m > 1, calculate the m-fold recursive Shanks transformation
S(S(…S(A)…))(n).
The Shanks transformation is useful for summing Taylor series that converge slowly near
a pole or singularity, e.g. for log(2):
Residues
TODO
Reference
sympy.series.residues.residue(expr, x, x0)
Finds the residue of expr at the point x=x0.
The residue is defined as the coefficient of 1/(x-x0) in the power series expansion about
x=x0.
References
1. http://en.wikipedia.org/wiki/Residue_theorem
Examples
Sequences
Examples
Sequences Base
class sympy.series.sequences.SeqBase
Base class for sequences
coeff(pt)
Returns the coefficient at point pt
coeff_mul(other)
Should be used when other is not a sequence. Should be defined to define custom
behaviour.
Notes
Examples
Examples
[1, 1]
>>> sequence(x+y*(-2)**(-n), (n, 0, oo)).find_linear_recurrence(30)
[1/2, 1/2]
>>> sequence(3*5**n + 12).find_linear_recurrence(20,gfvar=x)
([6, -5], 3*(-21*x + 5)/((x - 1)*(5*x - 1)))
>>> sequence(lucas(n)).find_linear_recurrence(15,gfvar=x)
([1, 1], (x - 2)/(x**2 + x - 1))
free_symbols
This method returns the symbols in the object, excluding those that take on a specific
value (i.e. the dummy symbols).
Examples
gen
Returns the generator for the sequence
interval
The interval on which the sequence is defined
length
Length of the sequence
start
The starting point of the sequence. This point is included
stop
The ending point of the sequence. This point is included
variables
Returns a tuple of variables that are bounded
Elementary Sequences
class sympy.series.sequences.SeqFormula
Represents sequence based on a formula.
Elements are generated using a formula.
See also:
sympy.series.sequences.SeqPer (page 1147)
Examples
>>> s.coeff(3)
9
supports slicing
>>> s[:]
[0, 1, 4, 9, 16, 25]
iterable
>>> list(s)
[0, 1, 4, 9, 16, 25]
coeff_mul(coeff)
See docstring of SeqBase.coeff_mul
class sympy.series.sequences.SeqPer
Represents a periodic sequence.
The elements are repeated after a given period.
See also:
sympy.series.sequences.SeqFormula (page 1146)
Examples
>>> s.coeff(3)
1
supports slicing
>>> s[:]
[1, 2, 3, 1, 2, 3]
iterable
>>> list(s)
[1, 2, 3, 1, 2, 3]
Periodic formulas
coeff_mul(coeff)
See docstring of SeqBase.coeff_mul
Singleton Sequences
class sympy.series.sequences.EmptySequence
Represents an empty sequence.
The empty sequence is available as a singleton as S.EmptySequence.
Examples
coeff_mul(coeff)
See docstring of SeqBase.coeff_mul
Compound Sequences
class sympy.series.sequences.SeqAdd
Represents term-wise addition of sequences.
Rules:
• The interval on which sequence is defined is the intersection of respective inter-
vals of sequences.
Examples
static reduce(args)
Simplify SeqAdd (page 1148) using known rules.
Iterates through all pairs and ask the constituent sequences if they can simplify
themselves with any other constituent.
Notes
Examples
static reduce(args)
Simplify a SeqMul (page 1149) using known rules.
Iterates through all pairs and ask the constituent sequences if they can simplify
themselves with any other constituent.
Notes
Fourier Series
Examples
scalex(s)
Scale x by a term independent of x.
f(x) -> f(s*x)
This is fast, if Fourier series of f(x) is already computed.
Examples
shift(s)
Shift the function by a term independent of x.
f(x) -> f(x) + s
This is fast, if Fourier series of f(x) is already computed.
Examples
shiftx(s)
Shift x by a term independent of x.
f(x) -> f(x + s)
This is fast, if Fourier series of f(x) is already computed.
Examples
sigma_approximation(n=3)
Return σ-approximation of Fourier series with respect to order n.
Sigma approximation adjusts a Fourier summation to eliminate the Gibbs phe-
nomenon which would otherwise occur at discontinuities. A sigma-approximated
summation for a Fourier series of a T-periodical function can be written as
1 ∑
m−1 (k) [ ( 2πk ) ( 2πk )]
s(θ) = a0 + sinc · ak cos θ + bk sin θ ,
2 m T T
k=1
( )
where a0 , ak , bk , k = 1, . . . , m − 1 are standard Fourier series coefficients and sinc k
m
is a Lanczos σ factor (expressed in terms of normalized sinc function).
Parameters n : int
Highest order of the terms taken into account in approximation.
Returns Expr
Sigma approximation of function expanded into Fourier series.
See also:
sympy.series.fourier.FourierSeries.truncate (page 1152)
Notes
References
[R492], [R493]
Examples
truncate(n=3)
Return the first n nonzero terms of the series.
If n is None return an iterator.
Parameters n : int or None
Amount of non-zero terms in approximation or None.
Returns Expr or iterator
Approximation of function expanded into Fourier series.
See also:
sympy.series.fourier.FourierSeries.sigma_approximation (page 1151)
Examples
sympy.series.fourier.fourier_series(f, limits=None)
Computes Fourier sine/cosine series expansion.
Returns a FourierSeries (page 1150) object.
See also:
sympy.series.fourier.FourierSeries (page 1150)
Notes
Computing Fourier series can be slow due to the integration required in computing an,
bn.
It is faster to compute Fourier series of a function by using shifting and scaling on an
already computed Fourier series rather than computing again.
e.g. If the Fourier series of x**2 is known the Fourier series of x**2 - 1 can be found
by shifting by -1.
References
[R494]
Examples
Shifting
>>> s.shift(1).truncate()
-4*cos(x) + cos(2*x) + 1 + pi**2/3
>>> s.shiftx(1).truncate()
-4*cos(x + 1) + cos(2*x + 2) + pi**2/3
Scaling
>>> s.scale(2).truncate()
-8*cos(x) + 2*cos(2*x) + 2*pi**2/3
>>> s.scalex(2).truncate()
-4*cos(2*x) + cos(4*x) + pi**2/3
See also:
sympy.series.formal.fps (page 1154)
infinite
Returns an infinite representation of the series
integrate(x=None, **kwargs)
Integrate Formal Power Series.
Examples
polynomial(n=6)
Truncated series as polynomial.
Returns series sexpansion of f upto order O(x**n) as a polynomial(without O term).
truncate(n=6)
Truncated series.
Returns truncated series expansion of f upto order O(x**n).
If n is None, returns an infinite iterator.
sympy.series.formal.fps(f, x=None, x0=0, dir=1, hyper=True, order=4, ratio-
nal=True, full=False)
Generates Formal Power Series of f.
Returns the formal series expansion of f around x = x0 with respect to x in the form of
a FormalPowerSeries object.
Formal Power Series is represented using an explicit formula computed using different
algorithms.
See compute_fps() (page 1155) for the more details regarding the computation of for-
mula.
Parameters x : Symbol, optional
If x is None and f is univariate, the univariate symbols will be sup-
plied, otherwise an error will be raised.
x0 : number, optional
Point to perform series expansion about. Default is 0.
dir : {1, -1, ‘+’, ‘-‘}, optional
If dir is 1 or ‘+’ the series is calculated from the right and for -1 or ‘-‘
the series is calculated from the left. For smooth functions this flag
will not alter the results. Default is 1.
hyper : {True, False}, optional
Examples
Rational Functions
Parameters x : Symbol
x0 : number, optional
Point to perform series expansion about. Default is 0.
dir : {1, -1, ‘+’, ‘-‘}, optional
If dir is 1 or ‘+’ the series is calculated from the right and for -1 or ‘-‘
the series is calculated from the left. For smooth functions this flag
will not alter the results. Default is 1.
hyper : {True, False}, optional
Set hyper to False to skip the hypergeometric algorithm. By default
it is set to False.
See also:
sympy.series.formal.rational_algorithm (page 1156), sympy.series.formal.
hyper_algorithm (page 1160)
Rational Algorithm
sympy.series.formal.rational_independent(terms, x)
Returns a list of all the rationally independent terms.
Examples
rational_algorithm() (page 1156) uses apart() function for partial fraction decom-
position. apart() by default uses ‘undetermined coefficients method’. By setting
full=True, ‘Bronstein’s algorithm’ can be used instead.
Looks for derivative of a function up to 4’th order (by default). This can be overridden
using order option.
Returns formula : Expr
ind : Expr
Independent terms.
order : int
See also:
sympy.polys.partfrac.apart (page 902)
Notes
References
[R488], [R489]
Examples
Hypergeometric Algorithm
sympy.series.formal.simpleDE(f, x, g, order=4)
Generates simple DE.
DE is of the form
∑
k−1
f k (x) + Aj f j (x) = 0
j=0
f j (x) → r(k + j)
Examples
sympy.series.formal.hyper_re(DE, r, k)
Converts a DE into a RE.
Performs the substitution:
Examples
sympy.series.formal.rsolve_hypergeometric(f, x, P, Q, k, m)
Solves RE of hypergeometric type.
Attempts to solve RE of the form
Q(k)*a(k + m) - P(k)*a(k)
Transformations that preserve Hypergeometric type:
a. x**n*f(x): b(k + m) = R(k - n)*b(k)
b. f(A*x): b(k + m) = A**m*R(k)*b(k)
c. f(x**n): b(k + n*m) = R(k/n)*b(k)
d. f(x**(1/m)): b(k + 1) = R(k*m)*b(k)
e. f’(x): b(k + m) = ((k + m + 1)/(k + 1))*R(k + 1)*b(k)
Some of these transformations have been used to solve the RE.
Returns formula : Expr
ind : Expr
Independent terms.
order : int
References
[R490], [R491]
Examples
Examples
sympy.series.formal.hyper_algorithm(f, x, k, order=4)
Hypergeometric algorithm for computing Formal Power Series.
Steps:
• Generates DE
• Convert the DE into RE
• Solves the RE
See also:
sympy.series.formal.simpleDE (page 1157), sympy.series.formal.solve_de
(page 1159)
Examples
>>> hyper_algorithm(exp(x), x, k)
(Piecewise((1/factorial(k), Eq(Mod(k, 1), 0)), (0, True)), 1, 1)
Limits of Sequences
References
[R495]
Examples
sympy.series.limitseq.dominant(expr, n)
Finds the dominant term in a sum, that is a term that dominates every other term.
If limit(a/b, n, oo) is oo then a dominates b. If limit(a/b, n, oo) is 0 then b dominates a.
Otherwise, a and b are comparable.
If there is no unique dominant term, then returns None.
See also:
sympy.series.limitseq.dominant (page 1161)
Examples
References
[R496]
Examples
Admissible Terms
The algorithm is designed for sequences built from rational functions, indefinite sums,
and indefinite products over an indeterminate n. Terms of alternating sign are also
allowed, but more complex oscillatory behavior is not supported.
5.26 Sets
5.26.1 Set
class sympy.sets.sets.Set
The base class for any kind of set.
This is not meant to be used directly as a container of items. It does not behave like the
builtin set; see FiniteSet (page 1171) for that.
Real intervals are represented by the Interval (page 1169) class and unions of sets by
the Union (page 1172) class. The empty set is represented by the EmptySet (page 1175)
class and available as a singleton as S.EmptySet.
Attributes
is_Complement
is_EmptySet
is_Intersection
is_UniversalSet
boundary
The boundary or frontier of a set
A point x is on the boundary of a set S if
1. x is in the closure of S. I.e. Every neighborhood of x contains a point in S.
2. x is not in the interior of S. I.e. There does not exist an open set centered on x
contained entirely within S.
There are the points on the outer rim of S. If S is open then these points need not
actually be contained within S.
For example, the boundary of an interval is its start and end points. This is true
regardless of whether or not the interval is open.
Examples
closure
Property method which returns the closure of a set. The closure is defined as the
union of the set itself and its boundary.
Examples
complement(universe)
The complement of ‘self’ w.r.t the given universe.
Examples
contains(other)
Returns True if ‘other’ is contained in ‘self’ as an element.
As a shortcut it is possible to use the ‘in’ operator:
Examples
inf
The infimum of ‘self’
Examples
interior
Property method which returns the interior of a set. The interior of a set S consists
all points of S that do not belong to the boundary of S.
Examples
intersect(other)
Returns the intersection of ‘self’ and ‘other’.
intersection(other)
Alias for intersect() (page 1164)
is_closed
A property method to check whether a set is closed. A set is closed if it’s complement
is an open set.
Examples
is_disjoint(other)
Returns True if ‘self’ and ‘other’ are disjoint
References
[R498]
Examples
is_open
Property method to check whether a set is open. A set is open if and only if it has
an empty intersection with its boundary.
Examples
is_proper_subset(other)
Returns True if ‘self’ is a proper subset of ‘other’.
Examples
is_proper_superset(other)
Returns True if ‘self’ is a proper superset of ‘other’.
Examples
is_subset(other)
Returns True if ‘self’ is a subset of ‘other’.
Examples
is_superset(other)
Returns True if ‘self’ is a superset of ‘other’.
Examples
isdisjoint(other)
Alias for is_disjoint() (page 1165)
issubset(other)
Alias for is_subset() (page 1166)
issuperset(other)
Alias for is_superset() (page 1166)
measure
The (Lebesgue) measure of ‘self’
Examples
powerset()
Find the Power set of ‘self’.
References
[R499]
Examples
sup
The supremum of ‘self’
Examples
symmetric_difference(other)
Returns symmetric difference of self and other.
References
[R500]
Examples
union(other)
Returns the union of ‘self’ and ‘other’.
Examples
sympy.sets.sets.imageset(*args)
Return an image of the set under transformation f.
If this function can’t compute the image, it returns an unevaluated ImageSet object.
f (x)|x ∈ self
See also:
sympy.sets.fancysets.ImageSet (page 1179)
Examples
Expressions applied to the set of Integers are simplified to show as few negatives as
possible and linear expressions are converted to a canonical form. If this is not desirable
then the unevaluated ImageSet should be used.
Elementary Sets
5.26.2 Interval
class sympy.sets.sets.Interval
Represents a real interval as a Set.
Usage: Returns an interval with end points “start” and “end”.
For left_open=True (default left_open is False) the interval will be open on the left.
Similarly, for right_open=True the interval will be open on the right.
Notes
References
[R501]
Examples
Attributes
is_Complement
is_EmptySet
is_Intersection
is_UniversalSet
classmethod Lopen(a, b)
Return an interval not including the left boundary.
classmethod Ropen(a, b)
Return an interval not including the right boundary.
as_relational(x)
Rewrite an interval in terms of inequalities and logic operators.
end
The right end point of ‘self’.
This property takes the same value as the ‘sup’ property.
Examples
is_left_unbounded
Return True if the left endpoint is negative infinity.
is_right_unbounded
Return True if the right endpoint is positive infinity.
left
The left end point of ‘self’.
This property takes the same value as the ‘inf’ property.
Examples
left_open
True if ‘self’ is left-open.
Examples
classmethod open(a, b)
Return an interval including neither boundary.
right
The right end point of ‘self’.
This property takes the same value as the ‘sup’ property.
Examples
right_open
True if ‘self’ is right-open.
Examples
start
The left end point of ‘self’.
This property takes the same value as the ‘inf’ property.
Examples
5.26.3 FiniteSet
class sympy.sets.sets.FiniteSet
Represents a finite set of discrete numbers
References
[R502]
Examples
Attributes
is_Complement
is_EmptySet
is_Intersection
is_UniversalSet
as_relational(symbol)
Rewrite a FiniteSet in terms of equalities and logic operators.
Compound Sets
5.26.4 Union
class sympy.sets.sets.Union
Represents a union of sets as a Set (page 1162).
See also:
Intersection (page 1173)
References
[R503]
Examples
The Union constructor will always try to merge overlapping intervals, if possible. For
example:
>>> Union(Interval(1, 2), Interval(2, 3))
Interval(1, 3)
Attributes
is_Complement
is_EmptySet
is_Intersection
is_UniversalSet
as_relational(symbol)
Rewrite a Union in terms of equalities and logic operators.
5.26.5 Intersection
class sympy.sets.sets.Intersection
Represents an intersection of sets as a Set (page 1162).
See also:
Union (page 1172)
References
[R504]
Examples
Attributes
is_Complement
is_EmptySet
is_UniversalSet
as_relational(symbol)
Rewrite an Intersection in terms of equalities and logic operators
5.26.6 ProductSet
class sympy.sets.sets.ProductSet
Represents a Cartesian Product of Sets.
Returns a Cartesian product given several sets as either an iterable or individual argu-
ments.
Can use ‘*’ operator on any sets for convenient shorthand.
Notes
References
[R505]
Examples
Attributes
is_Complement
is_EmptySet
is_Intersection
is_UniversalSet
is_iterable
A property method which tests whether a set is iterable or not. Returns True if set
is iterable, otherwise returns False.
Examples
5.26.7 Complement
class sympy.sets.sets.Complement
Represents the set difference or relative complement of a set with another set.
A − B = {x ∈ A|x
notinB}
See also:
Intersection (page 1173), Union (page 1172)
References
[R506]
Examples
Attributes
is_EmptySet
is_Intersection
is_UniversalSet
static reduce(A, B)
Simplify a Complement (page 1175).
Singleton Sets
5.26.8 EmptySet
class sympy.sets.sets.EmptySet
Represents the empty set. The empty set is available as a singleton as S.EmptySet.
See also:
UniversalSet (page 1176)
References
[R507]
Examples
Attributes
is_Complement
is_Intersection
is_UniversalSet
5.26.9 UniversalSet
class sympy.sets.sets.UniversalSet
Represents the set of all things. The universal set is available as a singleton as
S.UniversalSet
See also:
EmptySet (page 1175)
References
[R508]
Examples
Attributes
is_Complement
is_EmptySet
is_Intersection
Special Sets
5.26.10 Naturals
class sympy.sets.fancysets.Naturals
Represents the natural numbers (or counting numbers) which are all positive integers
starting from 1. This set is also available as the Singleton, S.Naturals.
See also:
Examples
Attributes
is_Complement
is_EmptySet
is_Intersection
is_UniversalSet
5.26.11 Naturals0
class sympy.sets.fancysets.Naturals0
Represents the whole numbers which are all the non-negative integers, inclusive of zero.
See also:
Attributes
is_Complement
is_EmptySet
is_Intersection
is_UniversalSet
5.26.12 Integers
class sympy.sets.fancysets.Integers
Represents all integers: positive, negative and zero. This set is also available as the
Singleton, S.Integers.
See also:
Examples
Attributes
is_Complement
is_EmptySet
is_Intersection
is_UniversalSet
5.26.13 ImageSet
class sympy.sets.fancysets.ImageSet
Image of a set under a mathematical function. The transformation must be given as a
Lambda function which has as many arguments as the elements of the set upon which
it operates, e.g. 1 argument when acting on the set of integers or 2 arguments when
acting on a complex region.
This function is not normally called directly, but is called from imageset.
See also:
sympy.sets.sets.imageset (page 1168)
Examples
>>> x = Symbol('x')
>>> N = S.Naturals
>>> squares = ImageSet(Lambda(x, x**2), N) # {x**2 for x in N}
>>> 4 in squares
True
>>> 5 in squares
False
If you want to get value for x = 2, 1/2 etc. (Please check whether the x value is in bases et
or not before passing it as args)
>>> squares.lamda(2)
4
>>> squares.lamda(S(1)/2)
1/4
>>> n = Dummy('n')
>>> solutions = ImageSet(Lambda(n, n*pi), S.Integers) # solutions of sin(x) = 0
>>> dom = Interval(-1, 1)
>>> dom.intersect(solutions)
{0}
Attributes
is_Complement
is_EmptySet
is_Intersection
is_UniversalSet
5.26.14 Range
class sympy.sets.fancysets.Range
Represents a range of integers. Can be called as Range(stop), Range(start, stop), or
Range(start, stop, step); when stop is not given it defaults to 1.
Range(stop) is the same as Range(0, stop, 1) and the stop value (juse as for Python ranges)
is not included in the Range values.
The stop value is made canonical so equivalent ranges always have the same args:
Infinite ranges are allowed. If the starting point is infinite, then the final value is stop
- step. To iterate such a range, it needs to be reversed:
Although Range is a set (and supports the normal set operations) it maintains the order
of the elements and can be used in contexts where range would be used.
Athough slicing of a Range will always return a Range – possibly empty – an empty set
will be returned from any intersection that is empty:
>>> Range(3)[:0]
Range(0, 0, 1)
>>> Range(3).intersect(Interval(4, oo))
EmptySet()
>>> Range(3).intersect(Range(4, oo))
EmptySet()
Attributes
is_Complement
is_EmptySet
is_Intersection
is_UniversalSet
reversed
Return an equivalent Range in the opposite order.
Examples
5.26.15 ComplexRegion
class sympy.sets.fancysets.ComplexRegion
Represents the Set of all Complex Numbers. It can represent a region of Complex Plane
in both the standard forms Polar and Rectangular coordinates.
• Polar Form Input is in the form of the ProductSet or Union of ProductSets of the
intervals of r and theta, & use the flag polar=True.
Z = {z in C | z = r*[cos(theta) + I*sin(theta)], r in [r], theta in [theta]}
• Rectangular Form Input is in the form of the ProductSet or Union of ProductSets of
interval of x and y the of the Complex numbers in a Plane. Default input type is in
rectangular form.
Z = {z in C | z = x + I*y, x in [Re(z)], y in [Im(z)]}
See also:
Reals
Examples
• c2 represents the Union of two rectangular regions in complex plane. One of them
surrounded by the coordinates of c1 and other surrounded by the coordinates (4, 1),
(6, 1), (6, 8) and (4, 8).
>>> r = Interval(0, 1)
>>> theta = Interval(0, 2*S.Pi)
>>> c2 = ComplexRegion(r*theta, polar=True) # Polar Form
>>> c2 # unit Disk
ComplexRegion(Interval(0, 1) x Interval.Ropen(0, 2*pi), True)
• c2 represents the region in complex plane inside the Unit Disk centered at the origin.
Attributes
is_Complement
is_EmptySet
is_Intersection
is_UniversalSet
a_interval
Return the union of intervals of x when, self is in rectangular form, or the union of
intervals of r when self is in polar form.
Examples
b_interval
Return the union of intervals of y when, self is in rectangular form, or the union of
intervals of theta when self is in polar form.
Examples
classmethod from_real(sets)
Converts given subset of real numbers to a complex region.
Examples
polar
Returns True if self is in polar form.
Examples
psets
Return a tuple of sets (ProductSets) input of the self.
Examples
sets
Return raw input sets to the self.
Examples
sympy.sets.fancysets.normalize_theta_set(theta)
Normalize a Real Set theta in the Interval [0, 2*pi). It returns a normalized value of theta
in the Set. For Interval, a maximum of one cycle [0, 2*pi], is returned i.e. for theta equal
to [0, 10*pi], returned normalized value would be [0, 2*pi). As of now intervals with end
points as non-multiples of pi is not supported.
Raises NotImplementedError
The algorithms for Normalizing theta Set are not yet implemented.
ValueError
The input is not valid, i.e. the input is not a real set.
RuntimeError
It is a bug, please report to the github issue tracker.
Examples
5.27 Simplify
5.27.1 simplify
Note that we could have obtained the same result by using specific simplification func-
tions:
In some cases, applying simplify() (page 1185) may actually result in some more com-
plicated expression. The default ratio=1.7 prevents more extreme cases: if (result
length)/(input length) > ratio, then input is returned unmodified. The measure parame-
ter lets you specify the function used to determine how complex an expression is. The
function should take a single argument as an expression and return a number such that
if expression a is more complex than expression b, then measure(a) > measure(b). The
default measure function is count_ops(), which returns the total number of operations
in the expression.
For example, if ratio=1, simplify output can’t be longer than input.
Since simplify(root) would result in a slightly longer expression, root is returned un-
changed instead:
Note that the shortest expression is not necessary the simplest, so setting ratio to 1 may
not be a good idea. Heuristically, the default value ratio=1.7 seems like a reasonable
choice.
You can easily define your own measure function based on what you feel should represent
the “size” or “complexity” of the input expression. Note that some choices, such as
lambda expr: len(str(expr)) may appear to be good metrics, but have other problems
(in this case, the measure function may slow down simplify too much for very large
expressions). If you don’t know what a good metric would be, the default, count_ops, is
a good one.
For example:
So you can see that h is simpler than g using the count_ops metric. However, we may not
like how simplify (in this case, using logcombine) has created the b**(log(1/a) + 1)
term. A simple way to reduce this would be to give more weight to powers as operations
in count_ops. We can do this by using the visual=True option:
Note that because simplify() internally tries many different simplification strategies
and then compares them using the measure function, we get a completely different result
that is still different from the input expression by doing this.
If rational=True, Floats will be recast as Rationals before simplification. If ratio-
nal=None, Floats will be recast as Rationals but the result will be recast as Floats. If
rational=False(default) then nothing will be done to the Floats.
If inverse=True, it will be assumed that a composition of inverse functions, such as sin
and asin, can be cancelled in any order. For example, asin(sin(x)) will yield x without
checking whether x belongs to the set where this relation is true. The default is False.
5.27.2 separatevars
Notes
The order of the factors is determined by Mul, so that the separated expressions may
not necessarily be grouped together.
Although factoring is necessary to separate variables in some expressions, it is not nec-
essary in all cases, so one should not count on the returned factors being factored.
Examples
>>> e = 2*x**2*z*sin(y)+2*z*x**2
>>> separatevars(e)
2*x**2*z*(sin(y) + 1)
>>> separatevars(e, symbols=(x, y), dict=True)
{'coeff': 2*z, x: x**2, y: sin(y) + 1}
>>> separatevars(e, [x, y, alpha], dict=True)
{'coeff': 2*z, alpha: 1, x: x**2, y: sin(y) + 1}
If the expression is not really separable, or is only partially separable, separatevars will
do the best it can to separate it by using factoring.
If the expression is not separable then expr is returned unchanged or (if dict=True) then
None is returned.
5.27.3 nthroot
Examples
Algorithm
First nsimplify is used to get a candidate root; if it is not a root the minimal polynomial
is computed; the answer is one of its roots.
5.27.4 besselsimp
sympy.simplify.simplify.besselsimp(expr)
Simplify bessel-type functions.
This routine tries to simplify bessel-type functions. Currently it only works on the Bessel J
and I functions, however. It works by looking at all such functions in turn, and eliminating
factors of “I” and “-1” (actually their polar equivalents) in front of the argument. Then,
functions of half-integer order are rewritten using strigonometric functions and functions
of integer order (> 1) are rewritten using functions of low order. Finally, if the expression
was changed, compute factorization of the result with factor().
5.27.5 hypersimp
sympy.simplify.simplify.hypersimp(f, k)
Given combinatorial term f(k) simplify its consecutive term ratio i.e. f(k+1)/f(k). The
input term can be composed of functions and integer sequences which have equivalent
representation in terms of gamma special function.
The algorithm performs three basic steps:
1. Rewrite all functions in terms of gamma, if possible.
2. Rewrite all occurrences of gamma in terms of products of gamma and rising factorial
with integer, absolute constant exponent.
3. Perform simplification of nested fractions, powers and if the resulting expression is
a quotient of polynomials, reduce their total degree.
5.27.6 hypersimilar
sympy.simplify.simplify.hypersimilar(f, g, k)
Returns True if ‘f’ and ‘g’ are hyper-similar.
Similarity in hypergeometric sense means that a quotient of f(k) and g(k) is a rational
function in k. This procedure is useful in solving recurrence relations.
For more information see hypersimp().
5.27.7 nsimplify
Examples
5.27.8 posify
sympy.simplify.simplify.posify(eq)
Return eq (with generic symbols made positive) and a dictionary containing the mapping
between the old and new symbols.
Any symbol that has positive=None will be replaced with a positive dummy symbol hav-
ing the same name. This replacement will allow more symbolic processing of expres-
sions, especially those involving powers and logarithms.
A dictionary that can be sent to subs to restore eq to its original symbols is also returned.
>>> eq = 1/x
>>> log(eq).expand()
log(1/x)
>>> log(posify(eq)[0]).expand()
-log(_x)
>>> p, rep = posify(eq)
>>> log(p).expand().subs(rep)
-log(x)
>>> eq = x**2 - 4
>>> solve(eq, x)
[-2, 2]
>>> eq_x, reps = posify([eq, x]); eq_x
[_x**2 - 4, _x]
>>> solve(*eq_x)
[2]
5.27.9 logcombine
sympy.simplify.simplify.logcombine(expr, force=False)
Takes logarithms and combines them using the following rules:
• log(x) + log(y) == log(x*y) if both are not negative
posify (page 1191) replace all symbols with symbols having positive assumptions
sympy.core.function.expand_log (page 199) expand the logarithms of products and
powers; the opposite of logcombine
Examples
The transformation is limited to factors and/or terms that contain logs, so the result
depends on the initial state of expansion:
>>> eq = (2 + 3*I)*log(x)
>>> logcombine(eq, force=True) == eq
True
>>> logcombine(eq.expand(), force=True)
log(x**2) + I*log(x**3)
5.27.10 Radsimp
radsimp
Examples
No simplification beyond removal of the gcd is done. One might want to polish the result
a little, however, by collecting square root terms:
>>> r2 = sqrt(2)
>>> r5 = sqrt(5)
>>> ans = radsimp(1/(y*r2 + x*r2 + a*r5 + b*r5)); pprint(ans)
___ ___ ___ ___
\/ 5 *a + \/ 5 *b - \/ 2 *x - \/ 2 *y
------------------------------------------
2 2 2 2
5*a + 10*a*b + 5*b - 2*x - 4*x*y - 2*y
>>> n, d = fraction(ans)
>>> pprint(factor_terms(signsimp(collect_sqrt(n))/d, radical=True))
___ ___
\/ 5 *(a + b) - \/ 2 *(x + y)
------------------------------------------
2 2 2 2
5*a + 10*a*b + 5*b - 2*x - 4*x*y - 2*y
Results with symbols will not always be valid for all substitutions:
rad_rationalize
sympy.simplify.radsimp.rad_rationalize(num, den)
Rationalize num/den by removing square roots in the denominator; num and den are
sum of terms whose squares are rationals
Examples
collect
Examples
You can also work with multivariate polynomials. However, remember that this function
is greedy so it will care only about a single symbol at time, in specification order:
>>> w = Wild('w1')
>>> collect(a*x**y - b*x**y, w**y)
x**y*(a - b)
It is also possible to work with symbolic powers, although it has more complicated be-
havior, because in this case power’s base and symbolic part of the exponent are treated
as a single symbol:
However if you incorporate rationals to the exponents, then you will get well known
behavior:
Note also that all previously stated facts about collect() (page 1194) function apply to
the exponential function, so you can get:
If you are interested only in collecting specific powers of some symbols then set exact
flag in arguments:
You can also apply this function to differential equations, where derivatives of arbitrary
order can be collected. Note that if you collect with respect to a function or a derivative
of a function, all derivatives of that function will also be collected. Use exact=True to
prevent this from happening:
Or you can even match both derivative order and exponent at the same time:
Finally, you can apply a function to each of the collected coefficients. For example you
can factorize symbolic coefficients of polynomial:
Note: Arguments are expected to be in expanded form, so you might have to call
expand() prior to calling this function.
sympy.simplify.radsimp.rcollect(expr, *vars)
Recursively collect sums in an expression.
See also:
collect (page 1194), collect_const (page 1197), collect_sqrt (page 1197)
Examples
>>> rcollect(expr, y)
(x + y*(x**2 + x + 1))/(x + y)
collect_sqrt
sympy.simplify.radsimp.collect_sqrt(expr, evaluate=None)
Return expr with terms having common square roots collected together. If evaluate
is False a count indicating the number of sqrt-containing terms will be returned and, if
non-zero, the terms of the Add will be returned, else the expression itself will be returned
as a single term. If evaluate is True, the expression with any collected terms will be
returned.
Note: since I = sqrt(-1), it is collected, too.
See also:
collect (page 1194), collect_const (page 1197), rcollect (page 1196)
Examples
If evaluate is False then the arguments will be sorted and returned as a list and a count
of the number of sqrt-containing terms will be returned:
collect_const
Examples
>>> collect_const(x - y - z)
x - (y + z)
>>> collect_const(-y - z)
-(y + z)
>>> collect_const(2*x - 2*y - 2*z, 2)
2*(x - y - z)
>>> collect_const(2*x - 2*y - 2*z, -2)
2*x - 2*(y + z)
fraction
sympy.simplify.radsimp.fraction(expr, exact=False)
Returns a pair with expression’s numerator and denominator. If the given expression is
not a fraction then this function will return the tuple (expr, 1).
This function will not make any attempt to simplify nested fractions or to do any term
rewriting at all.
If only one of the numerator/denominator pair is needed then use numer(expr) or de-
nom(expr) functions respectively.
>>> fraction(x/y)
(x, y)
>>> fraction(x)
(x, 1)
>>> fraction(1/y**2)
(1, y**2)
>>> fraction(x*y/2)
(x*y, 2)
>>> fraction(Rational(1, 2))
(1, 2)
If we know nothing about sign of some exponent and ‘exact’ flag is unset, then structure
this exponent’s structure will be analyzed and pretty fraction will be returned:
>>> fraction(exp(-x))
(1, exp(x))
The exact flag will also keep any unevaluated Muls from being evaluated:
5.27.11 Ratsimp
ratsimp
sympy.simplify.ratsimp.ratsimp(expr)
Put an expression over a common denominator, cancel and reduce.
Examples
trigsimp
sympy.simplify.trigsimp.trigsimp(expr, **opts)
reduces expression by using known trig identities
Notes
method: - Determine the method to use. Valid choices are ‘matching’ (default), ‘groeb-
ner’, ‘combined’, and ‘fu’. If ‘matching’, simplify the expression recursively by targeting
common patterns. If ‘groebner’, apply an experimental groebner basis algorithm. In this
case further options are forwarded to trigsimp_groebner, please refer to its docstring.
If ‘combined’, first run the groebner basis algorithm with small default parameters, then
run the ‘matching’ algorithm. ‘fu’ runs the collection of trigonometric transformations
described by Fu, et al. (see the f u docstring).
Examples
>>> trigsimp(log(e))
log(2)
powsimp
Notes
If deep is True then powsimp() will also simplify arguments of functions. By default deep
is set to False.
If force is True then bases will be combined without checking for assumptions, e.g.
sqrt(x)*sqrt(y) -> sqrt(x*y) which is not true if x and y are both negative.
You can make powsimp() only combine bases or only combine exponents by changing
combine=’base’ or combine=’exp’. By default, combine=’all’, which does both. com-
bine=’base’ will only combine:
a a a 2x x
x * y => (x*y) as well as things like 2 => 4
a b (a + b)
x * x => x
combine=’exp’ will strictly only combine exponents in the way that used to be automatic.
Also use deep=True if you need the old behavior.
When combine=’all’, ‘exp’ is evaluated first. Consider the first example below for when
there could be an ambiguity relating to this. This is done so things like the second
example can be completely combined. If you want ‘base’ combined first, do something
like powsimp(powsimp(expr, combine=’base’), combine=’exp’).
Examples
>>> a=sqrt(x*sqrt(y))
>>> a*a**3 == a**4
True
But if an integer power of that radical has been autoexpanded then Mul does not join
the resulting factors:
powdenest
When there are sums of logs in exp() then a product of powers may be obtained e.g.
exp(3*(log(a) + 2*log(b))) - > a**3*b**6.
Examples
>>> powdenest((x**(2*a/3))**(3*x))
(x**(2*a/3))**(3*x)
>>> powdenest(exp(3*x*log(2)))
2**(3*x)
>>> powdenest(sqrt(x**2))
sqrt(x**2)
But exp() will be denested by moving all non-log terms outside of the function; this may
result in the collapsing of the exp to a power with a different base:
>>> powdenest(exp(3*y*log(x)))
x**(3*y)
>>> powdenest(exp(y*(log(a) + log(b))))
(a*b)**y
>>> powdenest(exp(3*(log(a) + log(b))))
a**3*b**3
>>> powdenest(((x**(2*a/3))**(3*y/i))**x)
((x**(2*a/3))**(3*y/i))**x
>>> powdenest((x**(2*i)*y**(4*i))**z, force=True)
(x*y**2)**(2*i*z)
combsimp
sympy.simplify.combsimp.combsimp(expr)
Simplify combinatorial expressions.
This function takes as input an expression containing factorials, binomials, Pochhammer
symbol and other “combinatorial” functions, and tries to minimize the number of those
functions and reduce the size of their arguments.
The algorithm works by rewriting all combinatorial functions as gamma functions and
applying gammasimp() except simplification steps that may make an integer argument
non-integer. See docstring of gammasimp for more information.
Then it rewrites expression in terms of factorials and binomials by rewriting gammas as
factorials and converting (a+b)!/a!b! into binomials.
If expression has gamma functions or combinatorial functions with non-integer argu-
ment, it is automatically passed to gammasimp.
Examples
sqrtdenest
sympy.simplify.sqrtdenest.sqrtdenest(expr, max_iter=3)
Denests sqrts in an expression that contain other square roots if possible, otherwise
returns the expr unchanged. This is based on the algorithms of [1].
See also:
sympy.solvers.solvers.unrad
References
[1] http://researcher.watson.ibm.com/researcher/files/us-fagin/symb85.pdf
[2] D. J. Jeffrey and A. D. Rich, ‘Symplifying Square Roots of Square Roots by Denesting’
(available at http://www.cybertester.com/data/denest.pdf)
Examples
cse
Examples
opt_cse
sympy.simplify.cse_main.opt_cse(exprs, order=’canonical’)
Find optimization opportunities in Adds, Muls, Pows and negative coefficient Muls
Parameters exprs : list of sympy expressions
The expressions to optimize.
order : string, ‘none’ or ‘canonical’
The order by which Mul and Add arguments are processed. For large
expressions where speed is a concern, use the setting order=’none’.
Returns opt_subs : dictionary of expression substitutions
Examples
tree_cse
hyperexpand
Examples
use
Examples
>>> f = (x + y)**2*x + 1
EPath class
class sympy.simplify.epathtools.EPath
Manipulate expressions using paths.
EPath grammar in EBNF notation:
Examples
select(expr)
Retrieve parts of an expression selected by a path.
Examples
>>> path.select(expr)
[x, y]
>>> path.select(expr)
[x, x, y]
epath
Examples
This page describes how the function hyperexpand() and related code work. For usage, see
the documentation of the symplify module.
This section describes the algorithm used to expand hypergeometric functions. Most of it is
based on the papers [Roach1996] and [Roach1997].
Recall that the hypergeometric function is (initially) defined as
( ) ∑∞
a1 , · · · , ap (a1 )n · · · (ap )n z n
p Fq z = .
b1 , · · · , b q (b1 )n · · · (bq )n n!
n=0
It turns out that there are certain differential operators that can change the ap and pq param-
eters by integers. If a sequence of such operators is known that converts the set of indices
a0r and b0s into ap and bq , then we shall say the pair ap , bq is reachable from a0r , b0s . Our general
strategy is thus as follows: given a set ap , bq of parameters, try to look up an origin a0r , b0s for
which we know an expression, and then apply the sequence of differential operators to the
known expression to find an expression for the Hypergeometric function we are interested
in.
Notation
In the following, the symbol a will always denote a numerator parameter and the symbol b
will always denote a denominator parameter. The subscripts p, q, r, s denote vectors of that
length, so e.g. ap denotes a vector of p numerator parameters. The subscripts i and j denote
“running indices”, so they should usually be used in conjunction with a “for all i”. E.g. ai < 4
for all i. Uppercase subscripts I and J denote a chosen, fixed index. So for example aI > 0 is
true if the inequality holds for the one index I we are currently interested in.
( ) ( )
d ap a1 ···ap ap +1
It is also easy to show that dz p Fq bq
z = b1 ···bq p Fq bq +1
z , where ap +1 is the vector a1 +1, a2 +
1, . . . and similarly for bq + 1. Combining this[ with the shift operators, we arrive at] one(form) of
∏q a1 ···ap ∏p
j=1 B(bj ) − (b1 −1)···(bq −1)
d ap
the Hypergeometric differential equation: dz i=1 A(ai ) p Fq bq z =
0. This holds if all shift operators are defined, i.e. if no ai = 0 and no bj = 1.
Clearing
[ denominators ) and multiplying)]through by
) z we arrive at the following equation:
∏q ( d ∏p ( d (
d
z dz j=1 z dz + b j − 1 − z i=1 z dz + ai F ap
p q bq z = 0. Even though our derivation does not
show it, it can be checked that this equation holds whenever the p Fq is defined.
d
Notice that, under suitable conditions on aI , bJ , each of the operators A(ai ), B(bj ) and z dz
can be expressed in terms of A(aI ) or B(bJ ). Our ( next
) aim is to write the Hypergeometric
differential equation as follows: [XA(aI ) − r]p Fq ap
= 0, for some operator X and some
z
bq
( ) ( )
̸ 0, then we can write this as −1
constant r to be determined. If r = r X p Fq
ap +eI
bq
z = p Fq abqp z ,
−1
and so r X undoes the shifting of A(aI ), whence it will be called an inverse-shift operator.
Now A(aI ) exists if aI ̸= 0, and then z dz
d
= aI A(aI ) − a)I . Observe also that ))
all the operators
d
∏p ( d (∏
p
(
d
A(ai ), B(bj ) and z dz commute. We have i=1 z dz + ai = i=1,i̸=I z dz + ai aI A(aI ), so this
gives us (the first half
) of X. The other half does not have such a nice expression. We find
∏q ∏q
z dz j=1 z dz + bj − 1 = (aI A(aI ) − aI ) j=1 (aI A(aI ) − aI + bj − 1). Since the first half had no
d d
∏q
constant term, we infer r = −aI j=1 (bj − 1 − aI ).
This tells us under which conditions we can “un-shift” A(aI ), namely when aI ̸= 0 and r ̸= 0.
Substituting aI − 1 for aI then tells us under what conditions we can decrement the index aI .
Doing a similar analysis for B(aJ ), we arrive at the following rules:
• An index aI can be decremented if aI ̸= 1 and aI ̸= bj for all bj .
• An index bJ can be incremented if bJ ̸= −1 and bJ ̸= ai for all ai .
Combined with the conditions (stated above) for the existence of shift operators, we have thus
established the rules of the game!
Reduction of Order
( ) ( )
a∗
Notice that, quite trivially, if aI = bJ , we have p Fq ap
bq
z = p−1 Fq−1
p
b∗
z , where a∗p means ap
q
with aI omitted, and similarly for b∗q . We call this reduction of order.
(aI )n
In fact, we can do even better. If aI − bJ ∈ Z>0 , then it is easy to see that (bJ )n is actually a
d k n
polynomial in n. It is also easy to see that (z dz ) z = nk z n . Combining these two remarks we
find:
If aI − bJ ∈ Z>0 , then there exists a polynomial p(n) = p0 + p1 n + · · · )
(of degree aI − bJ )
( ) ( ( )2 ( ∗ )
(aI )n a
such that (bJ )n = p(n) and p Fq bq z = p0 + p1 z dz + p2 z dz + · · · p−1 Fq−1 b∗p z .
ap d d
q
It remains to investigate the following question: suppose ap , bq and a0p , b0q are both suitable,
and also ai − a0i ∈ Z, bj − b0j ∈ Z. When is ap , bq reachable from a0p , b0q ? It is clear that we can
treat all parameters independently that are incongruent mod 1. So assume that ai and bj are
congruent to r mod 1, for all i and j. The same then follows for a0i and b0j .
If r ̸= 0, then any such ap , bq is reachable from any a0p , b0q . To see this notice that there exist
constants c, c0 , congruent mod 1, such that ai < c < bj for all i and j, and similarly a0i < c0 < b0j .
If n = c − c0 > 0 then we first inverse-shift all the b0j n times up, and then similarly shift up
all the a0i n times. If n < 0 then we first inverse-shift down the a0i and then shift down the b0j .
This reduces to the case c = c0 . But evidently we can now shift or inverse-shift around the ai0
arbitrarily so long as we keep them less than c, and similarly for the b0j so long as we keep
them bigger than c. Thus ap , bq is reachable from a0p , b0q .
If r = 0 then the problem is slightly more involved. WLOG no parameter is zero. We now
have one additional complication: no parameter can ever move through zero. Hence ap , bq
is reachable from a0p , b0q if and only if the number of ai < 0 equals the number of a0i < 0, and
similarly for the bi and b0i . But in a suitable set of parameters, all bj > 0! This is because the
Hypergeometric function is undefined if one of the bj is a non-positive integer and all ai are
smaller than the bj . Hence the number of bj ≤ 0 is always zero.
We can thus associate to every suitable set of parameters ap , bq , where no ai = 0, the following
invariants:
• For every r ∈ [0, 1) the number αr of parameters ai ≡ r (mod 1), and similarly the number
βr of parameters bi ≡ r (mod 1).
• The number γ of integers ai with ai < 0.
The above reasoning shows that ap , bq is reachable from a0p , b0q if and only if the invariants
αr , βr , γ all agree. Thus in particular “being reachable from” is a symmetric relation on suitable
parameters without zeros.
If all goes well then for a given set of parameters we find an origin in our database for which
we have a nice formula. We now have to apply (potentially) many differential operators to it. If
we do this blindly then the result will be very messy. This is because with Hypergeometric type
functions, the derivative is usually expressed as a sum of two contiguous functions. Hence
if we compute N derivatives, then the answer will involve 2N contiguous functions! This is
clearly undesirable. In fact we know from the Hypergeometric differential equation that we
need at most max(p, q + 1) contiguous functions to express all derivatives.
Hence instead of differentiating blindly, we will work with a C(z)-module basis: for an origin
a0r , b0s we either store (for particularly pretty answers) or compute a set of N functions (typ-
ically N = max(r, s + 1)) with the property that the derivative of any of them is a C(z)-linear
combination of them. In formulae, we store a vector B of N functions, a matrix M and a vector
C (the latter two with entries in C(z)), with the following properties:
( 0 )
• r Fs ab0r z = CB
s
d
• z dz B = M B.
Then we can compute as many derivatives as we want and we will always end up with C(z)-
linear combination of at most N special functions.
As hinted above, B, M and C can either all be stored (for particularly pretty answers) or
computed from a single p Fq formula.
Loose Ends
This describes the bulk of the hypergeometric function algorithm. There a few further tricks,
described in the hyperexpand.py source file. The extension to Meijer G-functions is also de-
scribed there.
Slater’s theorem essentially evaluates a G-function as a sum of residues. If all poles are
simple, the resulting series can be recognised as hypergeometric series. Thus a G-function
can be evaluated as a sum of Hypergeometric functions.
If the poles are not simple, the resulting series are not hypergeometric. This is known as
the “confluent” or “logarithmic” case (the latter because the resulting series tend to contain
logarithms). The answer depends in a complicated way on the multiplicities of various poles,
and there is no accepted notation for representing it (as far as I know). However if there are
only finitely many multiple poles, we can evaluate the G function as a sum of hypergeometric
functions, plus finitely many extra terms. I could not find any good reference for this, which
is why I work it out here.
Recall the general setup. We define
∫ ∏m ∏n
1 j=1 Γ(bj − s) Γ(1 − aj + s)
j=1
G(z) = ∏q ∏p z s ds,
2πi L j=m+1 Γ(1 − bj + s) j=n+1 Γ(aj − s)
where L is a contour starting and ending at +∞, enclosing all of the poles of Γ(bj − s) for
j = 1, . . . , n once in the negative direction, and no other poles. Also the integral is assumed
absolutely convergent.
In what follows, for any complex numbers a, b, we write a ≡ b (mod 1) if and only if there
exists an integer k such that a − b = k. Thus there are double poles iff ai ≡ aj (mod 1) for some
i ̸= j ≤ n.
We now assume that whenever bj ≡ ai (mod 1) for i ≤ m, j > n then bj < ai . This means that
no quotient of the relevant gamma functions is a polynomial, and can always be achieved by
“reduction of order”. Fix a complex number c such that {bi |bi ≡ c (mod 1), i ≤ m} is not empty.
Enumerate this set as b, b + k1 , . . . , b + ku , with ki non-negative integers. Enumerate similarly
{aj |aj ≡ c (mod 1), j > n} as b + l1 , . . . , b + lv . Then li > kj for all i, j. For finite confluence, we
need to assume v ≥ u for all such c.
Let c1 , . . . , cw be distinct (mod 1) and exhaust the congruence classes of the bi . I claim
∑
w
G(z) = − (Fj (z) + Rj (z)),
j=1
where Fj (z) is a hypergeometric function and Rj (z) is a finite sum, both to be specified later.
Indeed corresponding to every cj there is a sequence of poles, at mostly finitely many of them
multiple poles. This is where the j-th term comes from.
Hence fix again c, enumerate the relevant bi as b, b + k1 , . . . , b + ku . We will look at the aj
corresponding to a + l1 , . . . , a + lu . The other ai are not treated specially. The corresponding
and
(−1)lu +t (−1)lu (−1)t
ress=b+lu +t Γ(b − s) = − =− .
(lu + t)! lu ! (lu + 1)t
Hence
∏n ∏m
(−1)lu ∏
∗
j=1 Γ(1 − aj + lu + b) j=1 Γ(bj − lu − b)
u
(−1)δi
ress=b+lu +t =−z b+lu
∏p ∏ q
lu ! i=1 (lu − ki + 1)δi j=n+1 Γ(aj − lu − b)∗ j=m+1 Γ(1 − bj + lu + b)
∏n ∏p
t ∏ ∗
(lu − li + 1)t j=1 (1 − aj + lu + b)t j=n+1 (−1) (lu + b + 1 − aj )t
u t
t (−1)
×z ∏m ∏ ,
(lu + 1)t i=1 (lu − ki + 1)t j=1 (−1)t (lu + b + 1 − bj )∗t qj=m+1 (1 − bj + lu + b)t
where C designates the factor in the residue independent of t. (This result can also be written
in slightly simpler form by converting all the lu etc back to a∗ − b∗ , but doing so is going to
require more notation still and is not helpful for computation.)
Adding new formulae to the tables is straightforward. At the top of the file sympy/simplify/
hyperexpand.py, there is a function called add_formulae(). Nested in it are defined two
helpers, add(ap, bq, res) and addb(ap, bq, B, C, M), as well as dummys a, b, c, and z.
The first step in adding a new formula is by using add(ap, bq, res). This declares hyper(ap,
bq, z) == res. Here ap and bq may use the dummys a, b, and c as free symbols. For example
∑∞ n
the well-known formula 0 (−a) nz
n! = (1 − z)a is declared by the following line: add((-a, ),
(), (1-z)**a).
From the information provided, the matrices B, C and M will be computed, and the formula is
now available when expanding hypergeometric functions. Next the test file sympy/simplify/
tests/test_hyperexpand.py should be run, in particular the test test_formulae(). This will
test the newly added formula numerically. If it fails, there is (presumably) a typo in what was
entered.
Since all newly-added formulae are probably relatively complicated, chances are that the
automatically computed basis is rather suboptimal (there is no good way of testing this, other
than observing very messy output). In this case the matrices B, C and M should be computed
by hand. Then the helper addb can be used to declare a hypergeometric formula with hand-
computed basis.
An example
Because this explanation so far might be very theoretical and difficult to understand, we walk
through an explicit example now. We take the Fresnel function C(z) which obeys the following
hypergeometric representation:
( 1 )
π2 z4
C(z) = z · 1 F2 1 5 −
4 .
2, 4 16
First we try to add this formula to the lookup table by using the (simpler) function add(ap,
bq, res). The first two arguments are simply the lists containing the parameter sets of 1 F2 .
The res argument is a little bit more complicated. We only know C(z) in terms of 1 F2 (. . . |f (z))
with f a function of z, in our case
π2 z4
f (z) = − .
16
What we need is a formula where the hypergeometric function has only z as argument
1 F2 (. . . |z). We introduce the new complex symbol w and search for a function g(w) such that
f (g(w)) = w
holds. Then we can replace every z in C(z) by g(w). In the case of our example the function g
could look like
( )
2 iπ 1
g(w) = √ exp w4 .
π 4
We get these functions mainly by guessing and testing the result. Hence we proceed by
computing f (g(w)) (and simplifying naively)
π 2 g(w)4
f (g(w)) = −
16
( ( ) 1 )4
π 2 g √2π exp iπ
4 w
4
=−
16
4 ( )4 1 4
π 2 √2π4 exp iπ4 w4
=−
16
= − exp (iπ) w
=w
and indeed get back w. (In case of branched functions we have to be aware of branch cuts. In
that case we take w to be a positive real number and check the formula. If what we have found
works for positive w, then just replace exp() inside any branched function by exp_polar()
and what we get is right for all w.) Hence we can write the formula as
( 1 )
C(g(w)) = g(w) · 1 F2 1 4 5 w .
2, 4
and trivially
( ( ) 1)
( ) C √2 exp iπ 4 w
1 4
C(g(w)) π
1 F2
4
1 5 w = = ( iπ ) 1
2, 4 g(w) √2
π
exp 4 w 4
which is exactly what is needed for the third parameter, res, in add. Finally, the whole function
call to add this rule to the table looks like:
add([S(1)/4],
[S(1)/2, S(5)/4],
fresnelc(exp(pi*I/4)*root(z,4)*2/sqrt(pi)) / (exp(pi*I/4)*root(z,4)*2/sqrt(pi))
)
Using this rule we will find that it works but the results are not really nice in terms of simplicity
and number of special function instances included. We can obtain much better results by
adding the formula to the lookup table in another way. For this we use the (more complicated)
function addb(ap, bq, B, C, M). The first two arguments are again the lists containing the
parameter sets of 1 F2 . The remaining three are the matrices mentioned earlier on this page.
We know that the n = max (p, q + 1)-th derivative can be expressed as a linear combination of
lower order derivatives. The matrix B contains the basis {B0 , B1 , . . .} and is of shape n × 1. The
best way to get Bi is to take the first n = max(p, q + 1) derivatives of the expression for p Fq
and take out useful pieces. In our case we find that n = max (1, 2 + 1) = 3. For computing the
d
derivatives, we have to use the operator z dz . The first basis element B0 is set to the expression
for 1 F2 from above:
√ ( ) ( 2 ( ) 1)
π exp − ıπ 4 C
√ exp ıπ z 4
π 4
B0 = 1
2z 4
d
Next we compute z dz B0 . For this we can directly use SymPy!
>>> from sympy import Symbol, sqrt, exp, I, pi, fresnelc, root, diff, expand
>>> z = Symbol(”z”)
>>> B0 = sqrt(pi)*exp(-I*pi/4)*fresnelc(2*root(z,4)*exp(I*pi/4)/sqrt(pi))/\
... (2*root(z,4))
>>> z * diff(B0, z)
z*(cosh(2*sqrt(z))/(4*z) - sqrt(pi)*exp(-I*pi/4)*fresnelc(2*z**(1/4)*exp(I*pi/4)/
,→sqrt(pi))/(8*z**(5/4)))
>>> expand(_)
cosh(2*sqrt(z))/4 - sqrt(pi)*exp(-I*pi/4)*fresnelc(2*z**(1/4)*exp(I*pi/4)/sqrt(pi))/
,→(8*z**(1/4))
>>> from sympy import (Symbol, cosh, sqrt, pi, exp, I, fresnelc, root,
... diff, expand)
>>> z = Symbol(”z”)
>>> B1prime = cosh(2*sqrt(z))/4 - sqrt(pi)*exp(-I*pi/4)*\
... fresnelc(2*root(z,4)*exp(I*pi/4)/sqrt(pi))/(8*root(z,4))
>>> z * diff(B1prime, z)
(continues on next page)
>>> expand(_)
sqrt(z)*sinh(2*sqrt(z))/4 - cosh(2*sqrt(z))/16 + sqrt(pi)*exp(-I*pi/
,→4)*fresnelc(2*z**(1/4)*exp(I*pi/4)/sqrt(pi))/(32*z**(1/4))
We see the common pattern and can collect the pieces. Hence it makes sense to choose B1
and B2 as follows
√ ( 1)
π exp(− ıπ
4 )C
√2 exp( ıπ )z 4
B0
π 4
B = B1 =
1
2z 4 √
cosh (2 z)
B2 √ √
sinh (2 z) z
(This is in contrast to the basis B = (B0 , B1′ , B2′ ) that would have been computed automatically
if we used just add(ap, bq, res).)
Because it must hold that p Fq (· · ·|z) = CB the entries of C are obviously
1
C = 0
0
Note that the entries of C and M should typically be rational functions in z, with rational
coefficients. This is all we need to do in order to add a new formula to the lookup table for
hyperexpand.
A vital part of the algorithm is a relatively large table of hypergeometric function represen-
tations. The following automatically generated list contains all the representations imple-
mented in SymPy (of course many more are derived from them). These formulae are mostly
taken from [Luke1969] and [Prudnikov1990]. They are all tested numerically.
( )
0 F0 z = ez
( )
1 F0 a z = (−z + 1)−a
( )
a, a − 1 (√ )−2a+1
2 F1
2 z = 22a−1 −z + 1 + 1
2a
( )
1, 1 log (−z + 1)
2 F1 z =−
2 z
(1 ) √
2, 1 z
atanh ( z)
F
2 1 3 = √
2 z
(1 1 ) √
,2 asin ( z)
2 F1
2
3 z = √
2 z
( ) √ −2a √ −2a
a, a + 12 ( z + 1) (− z + 1)
2 F1 1 z = +
2 2 2
( )
a, −a ( (√ ))
2 F1 1 z = cos 2a asin z
2
( √)
asin ( z)
1, 1
2 F1
3 = √ √z
2 z −z + 1
(1 1 )
2, 2 z
2K (z)
2 F1 =
1 π
( 1 1 )
−2, 2 2E (z)
2 F1 z =
1 π
( 1 ) √ √
− 2 , 1, 1 2 z atanh ( z) 2 log (−z + 1)
3 F2 1 z =− + −
2 , 2 3 3 3z
(√ )
( 1 ) ( ) 4 log −z+1
+ 21
− 2 , 1, 1 4 16 √ 2 16
3 F2 z = − −z + 1 + +
2, 2 9 9z 3z 9z
( )
1
1 F1 z = z −b+1 (b − 1) ez γ (b − 1, z)
b
( ) (z ) ( )
a a− 21 −a+ 12 z2 1
F
1 1 z = 4 z e I 1
a− 2 Γ a +
2a 2 2
( )
a ( )−a ( )
1 F1 z = a zeiπ γ a, zeiπ
a+1
( 1 )
−2 √ √ (√ )
F
1 1 1 z = zi π erf zi + ez
2
( ( ) ( ))
√ √ √ iπ
2 4√ √ √ iπ
2 4√
( ) e−
ze 4 ze 4 iπ
π i sinh (2 z)S π
+ cosh (2 z)C π
4
1
1 F2 3 5 z = √
4, 4 24z
( ) √ √ √ ( )−a ( )
1 ai π z1 erf ( zi) a zeiπ γ a, zeiπ
2, a =− −
2 F2 3 z
2, a +1 2a − 1 2a − 1
( )
1, 1 − log (z) + Ei (z) γ
2 F2 z = −
2, 2 z z
( )
( √ )
0 F1 1 z = cosh 2 z
2
( )
( √ )
= z − 2 + 2 Ib−1 2 z Γ (b)
b 1
0 F1 z
b
( )
1 ( ( √ ) ( √ ))
= 2−2a z − 2 + 4 I2a−1 4 4 z + J2a−1 4 4 z Γ (2a)
a
0 F3 1 1 z
2 , a, a
+ 2
( ) (
√ iπ )−2a+1 ( √ √ iπ ) ( √ √ iπ )
0 F3 z = 2 ze 2 I2a−1 2 2 4 ze 4 J2a−1 2 2 4 ze 4 Γ2 (2a)
a, a + 12 , 2a
( ) ( ) ( ) ( )
a a−1 −a+1
(√ ) (√ ) 1 1 a− 21 −a+ 12 2
(√ ) 2 1
1 F2 z =2·4 z Ia− 32 z Ia− 12 z Γ a− Γ a+ −4 z Ia− 1 z Γ a+
a − 12 , 2a 2 2 2 2
( 1
) √ √
2
π (−b + 1) I−b+1 ( z) Ib−1 ( z)
1 F2 z =
b, −b + 2 sin (bπ)
( 1 ) √
Shi (2 z)
F
1 2
2 z
3 3 = √
2, 2 2 z
( √ iπ )
√ − 3iπ 4
( 3 ) 3 πe 4 S 2 √zeπ 4
4
1 F2 3 7 z = 3
2, 4 4z 4
( √ iπ )
√ − iπ 4
( 1 ) πe 4 C 2 √zeπ
4
1 F2
4
1 5 z = √
2, 4 24z
( ) ( √ )−2a+1
a, a + 12 z (√ ) (√ )
2 F3 z = I2a−b z Ib−1 z Γ (b) Γ (2a − b + 1)
2a, b, 2a − b + 1 2
( ) √ √
1, 1 − log (2 z) + Chi (2 z) γ
2 F3 z = −
2, 2, 23 z z
( ) −a
1, 1, a a (−z) (Γ (a) − Γ (a, −z)) a (−a + 1) (log (−z) + E1 (−z) + γ) aez a
3 F3 z = 2 + 2 − 2a + 1)
− 2 − 2a + 1)
+ 2 − 2a + 1)
2, 2, a + 1 (a − 1) z (a z (a z (a
5.28.5 References
5.29 Stats
Expression Meaning
P(condition) Probability
E(expression) Expected value
variance(expression) Variance
density(expression) Probability Density Function
sample(expression) Produce a realization
where(condition) Where the condition is true
5.29.1 Examples
Finite Types
sympy.stats.DiscreteUniform(name, items)
Create a Finite Random Variable representing a uniform distribution over the input set.
Returns a RandomSymbol.
Examples
sympy.stats.Die(name, sides=6)
Create a Finite Random Variable representing a fair die.
Returns a RandomSymbol.
sympy.stats.Coin(name, p=1/2)
Create a Finite Random Variable representing a Coin toss.
Probability p is the chance of gettings “Heads.” Half by default
Returns a RandomSymbol.
Examples
sympy.stats.Hypergeometric(name, N, m, n)
Create a Finite Random Variable representing a hypergeometric distribution.
Returns a RandomSymbol.
Examples
sympy.stats.FiniteRV(name, density)
Create a Finite Random Variable given a dict representing the density.
Returns a RandomSymbol.
>>> E(X)
2.00000000000000
>>> P(X >= 2)
0.700000000000000
Discrete Types
sympy.stats.Geometric(name, p)
Create a discrete random variable with a Geometric distribution.
The density of the Geometric distribution is given by
References
Examples
>>> p = S.One / 5
>>> z = Symbol(”z”)
>>> X = Geometric(”x”, p)
>>> density(X)(z)
(4/5)**(z - 1)/5
>>> E(X)
5
>>> variance(X)
20
sympy.stats.Poisson(name, lamda)
Create a discrete random variable with a Poisson distribution.
The density of the Poisson distribution is given by
λk e−λ
f (k) :=
k!
Parameters lamda: Positive number, a rate
Returns A RandomSymbol.
References
Examples
>>> density(X)(z)
lambda**z*exp(-lambda)/factorial(z)
>>> E(X)
lambda
>>> simplify(variance(X))
lambda
Continuous Types
References
[R538]
Examples
>>> X = Arcsin(”x”, a, b)
>>> density(X)(z)
1/(pi*sqrt((-a + z)*(b - z)))
>>> cdf(X)(z)
Piecewise((0, a > z),
(2*asin(sqrt((-a + z)/(-a + b)))/pi, b >= z),
(1, True))
References
[R539], [R540]
Examples
>>> D = density(X)(z)
>>> pprint(D, use_unicode=False)
/ / z \\ / z \ 2/ z \
| 2*beta*log|-----|| - alpha*log|-----| - beta*log |-----|
|alpha \sigma/| \sigma/ \sigma/
|----- + -----------------|*e
\ z z /
xα−1 (1 − x)β−1
f (x) :=
B(α, β)
References
[R541], [R542]
Examples
>>> D = density(X)(z)
>>> pprint(D, use_unicode=False)
alpha - 1 beta - 1
z *(-z + 1)
---------------------------
B(alpha, beta)
xα−1 (1 + x)−α−β
f (x) :=
B(α, β)
with x > 0.
Parameters alpha : Real number, α > 0, a shape
beta : Real number, β > 0, a shape
Returns A RandomSymbol.
References
[R543], [R544]
Examples
>>> D = density(X)(z)
>>> pprint(D, use_unicode=False)
alpha - 1 -alpha - beta
z *(z + 1)
-------------------------------
B(alpha, beta)
References
[R545], [R546]
Examples
>>> x0 = Symbol(”x0”)
>>> gamma = Symbol(”gamma”, positive=True)
>>> z = Symbol(”z”)
>>> density(X)(z)
1/(pi*gamma*(1 + (-x0 + z)**2/gamma**2))
sympy.stats.Chi(name, k)
Create a continuous random variable with a Chi distribution.
The density of the Chi distribution is given by
with x ≥ 0.
Parameters k : A positive Integer, k > 0, the number of degrees of freedom
Returns A RandomSymbol.
References
[R547], [R548]
Examples
>>> X = Chi(”x”, k)
>>> density(X)(z)
2**(-k/2 + 1)*z**(k - 1)*exp(-z**2/2)/gamma(k/2)
sympy.stats.ChiNoncentral(name, k, l)
Create a continuous random variable with a non-central Chi distribution.
The density of the non-central Chi distribution is given by
e−(x +λ )/2 xk λ
2 2
with x ≥ 0. Here, Iν (x) is the modified Bessel function of the first kind (page 536).
Parameters k : A positive Integer, k > 0, the number of degrees of freedom
l : Shift parameter
Returns A RandomSymbol.
References
[R549]
Examples
>>> X = ChiNoncentral(”x”, k, l)
>>> density(X)(z)
l*z**k*(l*z)**(-k/2)*exp(-l**2/2 - z**2/2)*besseli(k/2 - 1, l*z)
sympy.stats.ChiSquared(name, k)
Create a continuous random variable with a Chi-squared distribution.
The density of the Chi-squared distribution is given by
1
( k ) x 2 −1 e− 2
k x
f (x) := k
2 Γ 2
2
with x ≥ 0.
Parameters k : A positive Integer, k > 0, the number of degrees of freedom
Returns A RandomSymbol.
References
[R550], [R551]
Examples
>>> X = ChiSquared(”x”, k)
>>> density(X)(z)
2**(-k/2)*z**(k/2 - 1)*exp(-z/2)/gamma(k/2)
>>> gammasimp(E(X))
k
>>> simplify(expand_func(variance(X)))
2*k
sympy.stats.Dagum(name, p, a, b)
Create a continuous random variable with a Dagum distribution.
The density of the Dagum distribution is given by
( ( x )ap )
ap b
x (( x )a + 1)p+1
f (x) :=
b
with x > 0.
Parameters p : Real number, p > 0, a shape
a : Real number, a > 0, a shape
b : Real number, b > 0, a scale
Returns A RandomSymbol.
References
[R552]
Examples
>>> X = Dagum(”x”, p, a, b)
>>> density(X)(z)
a*p*(z/b)**(a*p)*((z/b)**a + 1)**(-p - 1)/z
>>> cdf(X)(z)
Piecewise(((1 + (z/b)**(-a))**(-p), z >= 0), (0, True))
sympy.stats.Erlang(name, k, l)
Create a continuous random variable with an Erlang distribution.
The density of the Erlang distribution is given by
λk xk−1 e−λx
f (x) :=
(k − 1)!
References
[R553], [R554]
Examples
>>> X = Erlang(”x”, k, l)
>>> D = density(X)(z)
>>> pprint(D, use_unicode=False)
k k - 1 -l*z
l *z *e
---------------
Gamma(k)
>>> simplify(E(X))
k/l
>>> simplify(variance(X))
k/l**2
sympy.stats.Exponential(name, rate)
Create a continuous random variable with an Exponential distribution.
The density of the exponential distribution is given by
f (x) := λ exp(−λx)
References
[R555], [R556]
Examples
>>> X = Exponential(”x”, l)
>>> density(X)(z)
lambda*exp(-lambda*z)
>>> cdf(X)(z)
Piecewise((1 - exp(-lambda*z), z >= 0), (0, True))
>>> E(X)
1/lambda
>>> variance(X)
lambda**(-2)
>>> skewness(X)
2
>>> density(X)(z)
10*exp(-10*z)
>>> E(X)
1/10
>>> std(X)
1/10
with x > 0.
Parameters d1 : d1 > 0, where d_1 is the degrees of freedom (n_1 - 1)
d2 : d2 > 0, where d_2 is the degrees of freedom (n_2 - 1)
Returns A RandomSymbol.
References
[R557], [R558]
Examples
>>> D = density(X)(z)
>>> pprint(D, use_unicode=False)
d2
-- ______________________________
2 / d1 -d1 - d2
d2 *\/ (d1*z) *(d1*z + d2)
--------------------------------------
/d1 d2\
z*B|--, --|
\2 2 /
References
[R559], [R560]
Examples
>>> D = density(X)(z)
>>> pprint(D, use_unicode=False)
d1 d2
(continues on next page)
with x ≥ m.
Parameters a : Real number, a ∈ (0, ∞) the shape
s : Real number, s ∈ (0, ∞) the scale
m : Real number, m ∈ (−∞, ∞) the minimum
Returns A RandomSymbol.
References
[R561]
Examples
>>> X = Frechet(”x”, a, s, m)
>>> density(X)(z)
a*((-m + z)/s)**(-a - 1)*exp(-((-m + z)/s)**(-a))/s
>>> cdf(X)(z)
Piecewise((exp(-((-m + z)/s)**(-a)), m <= z), (0, True))
sympy.stats.Gamma(name, k, theta)
Create a continuous random variable with a Gamma distribution.
References
[R562], [R563]
Examples
>>> D = density(X)(z)
>>> pprint(D, use_unicode=False)
-z
-----
-k k - 1 theta
theta *z *e
---------------------
Gamma(k)
>>> E(X)
k*theta
>>> V = simplify(variance(X))
>>> pprint(V, use_unicode=False)
2
k*theta
sympy.stats.GammaInverse(name, a, b)
Create a continuous random variable with an inverse Gamma distribution.
The density of the inverse Gamma distribution is given by
( )
β α −α−1 −β
f (x) := x exp
Γ(α) x
with x > 0.
Parameters a : Real number, a > 0 a shape
b : Real number, b > 0 a scale
Returns A RandomSymbol.
References
[R564]
Examples
>>> X = GammaInverse(”x”, a, b)
>>> D = density(X)(z)
>>> pprint(D, use_unicode=False)
-b
---
a -a - 1 z
b *z *e
---------------
Gamma(a)
>>> cdf(X)(z)
Piecewise((uppergamma(a, b/z)/gamma(a), z > 0), (0, True))
sympy.stats.Kumaraswamy(name, a, b)
Create a Continuous Random Variable with a Kumaraswamy distribution.
The density of the Kumaraswamy distribution is given by
References
[R565]
Examples
>>> X = Kumaraswamy(”x”, a, b)
>>> D = density(X)(z)
>>> pprint(D, use_unicode=False)
b - 1
a - 1 / a \
a*b*z *\- z + 1/
>>> cdf(X)(z)
Piecewise((0, z < 0),
(-(-z**a + 1)**b + 1, z <= 1),
(1, True))
sympy.stats.Laplace(name, mu, b)
Create a continuous random variable with a Laplace distribution.
The density of the Laplace distribution is given by
( )
1 |x − µ|
f (x) := exp −
2b b
References
[R566], [R567]
Examples
>>> mu = Symbol(”mu”)
>>> b = Symbol(”b”, positive=True)
>>> z = Symbol(”z”)
>>> density(X)(z)
exp(-Abs(mu - z)/b)/(2*b)
>>> cdf(X)(z)
Piecewise((exp((-mu + z)/b)/2, mu > z),
(-exp((mu - z)/b)/2 + 1, True))
sympy.stats.Logistic(name, mu, s)
Create a continuous random variable with a logistic distribution.
The density of the logistic distribution is given by
e−(x−µ)/s
f (x) := ( )2
s 1 + e−(x−µ)/s
References
[R568], [R569]
Examples
>>> density(X)(z)
exp((mu - z)/s)/(s*(exp((mu - z)/s) + 1)**2)
>>> cdf(X)(z)
1/(exp((mu - z)/s) + 1)
References
[R570], [R571]
Examples
>>> D = density(X)(z)
>>> pprint(D, use_unicode=False)
2
-(-mu + log(z))
-----------------
2
___ 2*sigma
\/ 2 *e
------------------------
____
2*\/ pi *sigma*z
>>> density(X)(z)
sqrt(2)*exp(-log(z)**2/2)/(2*sqrt(pi)*z)
sympy.stats.Maxwell(name, a)
Create a continuous random variable with a Maxwell distribution.
f (x) :=
π a3
with x ≥ 0.
Parameters a : Real number, a > 0
Returns A RandomSymbol.
References
[R572], [R573]
Examples
>>> X = Maxwell(”x”, a)
>>> density(X)(z)
sqrt(2)*z**2*exp(-z**2/(2*a**2))/(sqrt(pi)*a**3)
>>> E(X)
2*sqrt(2)*a/sqrt(pi)
>>> simplify(variance(X))
a**2*(-8 + 3*pi)/pi
with x > 0.
Parameters mu : Real number, µ ≥ 1
2 a shape
omega : Real number, ω > 0, the spread
Returns A RandomSymbol.
References
[R574]
Examples
>>> D = density(X)(z)
>>> pprint(D, use_unicode=False)
2
-mu*z
-------
mu -mu 2*mu - 1 omega
2*mu *omega *z *e
----------------------------------
Gamma(mu)
>>> simplify(E(X))
sqrt(mu)*sqrt(omega)*gamma(mu + 1/2)/gamma(mu + 1)
>>> V = simplify(variance(X))
>>> pprint(V, use_unicode=False)
2
omega*Gamma (mu + 1/2)
omega - -----------------------
Gamma(mu)*Gamma(mu + 1)
>>> cdf(X)(z)
Piecewise((lowergamma(mu, mu*z**2/omega)/gamma(mu), z > 0),
(0, True))
References
[R575], [R576]
Examples
>>> mu = Symbol(”mu”)
>>> sigma = Symbol(”sigma”, positive=True)
>>> z = Symbol(”z”)
>>> y = Symbol(”y”)
>>> X = Normal(”x”, mu, sigma)
>>> density(X)(z)
sqrt(2)*exp(-(-mu + z)**2/(2*sigma**2))/(2*sqrt(pi)*sigma)
>>> simplify(skewness(X))
0
>>> E(2*X + 1)
1
References
[R577], [R578]
Examples
>>> density(X)(z)
beta*xm**beta*z**(-beta - 1)
sympy.stats.QuadraticU(name, a, b)
Create a Continuous Random Variable with a U-quadratic distribution.
The density of the U-quadratic distribution is given by
References
[R579]
Examples
>>> X = QuadraticU(”x”, a, b)
>>> D = density(X)(z)
>>> pprint(D, use_unicode=False)
/ 2
| / a b \
|12*|- - - - + z|
| \ 2 2 /
<----------------- for And(b >= z, a <= z)
| 3
| (-a + b)
|
\ 0 otherwise
sympy.stats.RaisedCosine(name, mu, s)
Create a Continuous Random Variable with a raised cosine distribution.
The density of the raised cosine distribution is given by
( ( ))
1 x−µ
f (x) := 1 + cos π
2s s
with x ∈ [µ − s, µ + s].
Parameters mu : Real number
s : Real number, s > 0
Returns A RandomSymbol.
References
[R580]
Examples
>>> D = density(X)(z)
>>> pprint(D, use_unicode=False)
/ /pi*(-mu + z)\
|cos|------------| + 1
| \ s /
<--------------------- for And(z >= mu - s, z <= mu + s)
| 2*s
|
\ 0 otherwise
sympy.stats.Rayleigh(name, sigma)
Create a continuous random variable with a Rayleigh distribution.
The density of the Rayleigh distribution is given by
x −x2 /2σ2
f (x) := e
σ2
with x > 0.
Parameters sigma : Real number, σ > 0
Returns A RandomSymbol.
References
[R581], [R582]
Examples
>>> density(X)(z)
z*exp(-z**2/(2*sigma**2))/sigma**2
>>> E(X)
sqrt(2)*sqrt(pi)*sigma/2
>>> variance(X)
-pi*sigma**2/2 + 2*sigma**2
sympy.stats.StudentT(name, nu)
Create a continuous random variable with a student’s t distribution.
The density of the student’s t distribution is given by
( ) ( )− ν+1
Γ ν+1
2( ) x2 2
f (x) := √ ν 1+
νπΓ 2 ν
References
[R583], [R584]
Examples
>>> D = density(X)(z)
>>> pprint(D, use_unicode=False)
nu 1
- -- - -
2 2
/ 2\
| z |
|1 + --|
\ nu/
-----------------
____ / nu\
\/ nu *B|1/2, --|
\ 2 /
>>> cdf(X)(z)
1/2 + z*gamma(nu/2 + 1/2)*hyper((1/2, nu/2 + 1/2), (3/2,),
-z**2/nu)/(sqrt(pi)*sqrt(nu)*gamma(nu/2))
sympy.stats.Triangular(name, a, b, c)
Create a continuous random variable with a triangular distribution.
The density of the triangular distribution is given by
0 for x < a,
2(x−a)
(b−a)(c−a) for a ≤ x < c,
2
f (x) := b−a for x = c,
2(b−x)
for c < x ≤ b,
(b−a)(b−c)
0 for b < x.
References
[R585], [R586]
Examples
>>> a = Symbol(”a”)
>>> b = Symbol(”b”)
>>> c = Symbol(”c”)
>>> z = Symbol(”z”)
References
[R587], [R588]
Examples
>>> X = Uniform(”x”, a, b)
>>> density(X)(z)
Piecewise((1/(-a + b), (b >= z) & (a <= z)), (0, True))
>>> simplify(E(X))
a/2 + b/2
>>> simplify(variance(X))
a**2/12 - a*b/6 + b**2/12
sympy.stats.UniformSum(name, n)
Create a continuous random variable with an Irwin-Hall distribution.
The probability distribution function depends on a single parameter n which is an integer.
The density of the Irwin-Hall distribution is given by
⌊x⌋
∑ ( )
1 k n
f (x) := (−1) (x − k)n−1
(n − 1)! k
k=0
References
[R589], [R590]
Examples
>>> X = UniformSum(”x”, n)
>>> D = density(X)(z)
>>> pprint(D, use_unicode=False)
floor(z)
___
\ `
(continues on next page)
>>> cdf(X)(z)
Piecewise((0, z < 0), (Sum((-1)**_k*(-_k + z)**n*binomial(n, _k),
(_k, 0, floor(z)))/factorial(n), n >= z), (1, True))
Compute cdf with specific ‘x’ and ‘n’ values as follows : >>> cdf(UniformSum(“x”, 5),
evaluate=False)(2).doit() 9/40
The argument evaluate=False prevents an attempt at evaluation of the sum for general
n, before the argument 2 is passed.
sympy.stats.VonMises(name, mu, k)
Create a Continuous Random Variable with a von Mises distribution.
The density of the von Mises distribution is given by
eκ cos(x−µ)
f (x) :=
2πI0 (κ)
with x ∈ [0, 2π].
Parameters mu : Real number, measure of location
k : Real number, measure of concentration
Returns A RandomSymbol.
References
[R591], [R592]
Examples
>>> mu = Symbol(”mu”)
>>> k = Symbol(”k”, positive=True)
>>> z = Symbol(”z”)
>>> D = density(X)(z)
>>> pprint(D, use_unicode=False)
k*cos(mu - z)
e
------------------
2*pi*besseli(0, k)
References
[R593], [R594]
Examples
>>> X = Weibull(”x”, l, k)
>>> density(X)(z)
k*(z/lambda)**(k - 1)*exp(-(z/lambda)**k)/lambda
>>> simplify(E(X))
lambda*gamma(1 + 1/k)
>>> simplify(variance(X))
lambda**2*(-gamma(1 + 1/k)**2 + gamma(1 + 2/k))
sympy.stats.WignerSemicircle(name, R)
Create a continuous random variable with a Wigner semicircle distribution.
The density of the Wigner semicircle distribution is given by
2 √ 2
f (x) := R − x2
πR2
with x ∈ [−R, R].
Parameters R : Real number, R > 0, the radius
Returns A RandomSymbol.
References
[R595], [R596]
Examples
>>> X = WignerSemicircle(”x”, R)
>>> density(X)(z)
2*sqrt(R**2 - z**2)/(pi*R**2)
>>> E(X)
0
Examples
>>> x = Symbol(”x”)
>>> E(X)
0
>>> P(X>0)
1/2
5.29.3 Interface
Examples
class sympy.stats.Probability
Symbolic expression for the probability.
Examples
Integral representation:
>>> prob.rewrite(Integral)
Integral(sqrt(2)*exp(-_z**2/2)/(2*sqrt(pi)), (_z, 1, oo))
>>> prob.evaluate_integral()
sqrt(2)*(-sqrt(2)*sqrt(pi)*erf(sqrt(2)/2) + sqrt(2)*sqrt(pi))/(4*sqrt(pi))
numsamples : int
Enables sampling and approximates the expectation with this many
samples
evalf : Bool (defaults to True)
If sampling return a number rather than a complex expression
evaluate : Bool (defaults to True)
In case of continuous systems return unevaluated integral
Examples
class sympy.stats.Expectation
Symbolic expression for the expectation.
Examples
>>> Expectation(X).rewrite(Integral)
Integral(sqrt(2)*X*exp(-(X - mu)**2/(2*sigma**2))/(2*sqrt(pi)*sigma), (X, -oo,␣
,→oo))
>>> Expectation(X).rewrite(Probability)
Integral(x*Probability(Eq(X, x)), (x, -oo, oo))
Examples
>>> x = Symbol('x')
>>> D = Die('D', 6)
>>> X = Normal(x, 0, 1)
>>> density(D).dict
{1: 1/6, 2: 1/6, 3: 1/6, 4: 1/6, 5: 1/6, 6: 1/6}
>>> density(2*D).dict
{2: 1/6, 4: 1/6, 6: 1/6, 8: 1/6, 10: 1/6, 12: 1/6}
>>> density(X)(x)
sqrt(2)*exp(-x**2/2)/(2*sqrt(pi))
Examples
Following convention, if the condition is a random symbol then that symbol is considered
fixed.
>>> X = Normal('X', 0, 1)
>>> Y = Normal('Y', 0, 1)
>>> pprint(density(X + Y, Y)(z), use_unicode=False)
2
-(-Y + z)
-----------
___ 2
\/ 2 *e
------------------
____
2*\/ pi
Examples
>>> where(X**2<1)
Domain: (-1 < x) & (x < 1)
>>> where(X**2<1).set
Interval.open(-1, 1)
Examples
>>> X = Die('X', 6)
>>> p = Symbol('p')
>>> B = Bernoulli('B', p, 1, 0)
>>> variance(2*X)
35/3
>>> simplify(variance(B))
p*(-p + 1)
class sympy.stats.Variance
Symbolic expression for the variance.
Examples
>>> Variance(X).rewrite(Integral)
Integral(sqrt(2)*(X - Integral(sqrt(2)*X*exp(-(X - mu)**2/(2*sigma**2))/
,→(2*sqrt(pi)*sigma), (X, -oo, oo)))**2*exp(-(X - mu)**2/(2*sigma**2))/
>>> Variance(X).rewrite(Probability)
-Integral(x*Probability(Eq(X, x)), (x, -oo, oo))**2 +␣
,→Integral(x**2*Probability(Eq(X, x)), (x, -oo, oo))
>>> Variance(X).rewrite(Expectation)
-Expectation(X)**2 + Expectation(X**2)
>>> Variance(a*X).doit()
a**2*Variance(X)
>>> Variance(X + Y)
Variance(X + Y)
>>> Variance(X + Y).doit()
2*Covariance(X, Y) + Variance(X) + Variance(Y)
Examples
>>> covariance(X, X)
lambda**(-2)
>>> covariance(X, Y)
0
>>> covariance(X, Y + rate*X)
1/lambda
class sympy.stats.Covariance
Symbolic expression for the covariance.
Examples
Evaluate the covariance, X and Y are independent, therefore zero is the result:
>>> cexpr.evaluate_integral()
0
Examples
>>> p = Symbol('p')
>>> B = Bernoulli('B', p, 1, 0)
>>> simplify(std(B))
sqrt(p*(-p + 1))
Examples
Examples
5.29.4 Mechanics
class sympy.stats.frv.FinitePSpace
class sympy.stats.crv.ContinuousDomain
class sympy.stats.crv.ContinuousPSpace
Additionally there are a few specialized classes that implement certain common random vari-
able types. There is for example a DiePSpace that implements SingleFinitePSpace and a
NormalPSpace that implements SingleContinuousPSpace.
class sympy.stats.frv_types.DiePSpace
class sympy.stats.crv_types.NormalPSpace
RandomVariables can be extracted from these objects using the PSpace.values method.
As previously mentioned SymPy Stats employs a relatively complex class structure. Inher-
itance is widely used in the implementation of end-level classes. This tactic was chosen to
balance between the need to allow SymPy to represent arbitrarily defined random variables
and optimizing for common cases. This complicates the code but is structured to only be
important to those working on extending SymPy Stats to other random variable types.
Users will not use this class structure. Instead these mechanics are exposed through variable
creation functions Die, Coin, FiniteRV, Normal, Exponential, etc…. These build the appropri-
ate SinglePSpaces and return the corresponding RandomVariable. Conditional and Product
spaces are formed in the natural construction of SymPy expressions and the use of interface
functions E, Given, Density, etc….
sympy.stats.Die()
sympy.stats.Normal()
There are some additional functions that may be useful. They are largely used internally.
sympy.stats.rv.random_symbols(expr)
Returns all RandomSymbols within a SymPy Expression.
sympy.stats.rv.pspace(expr)
Returns the underlying Probability Space of a random expression.
For internal use.
Examples
sympy.stats.rv.rs_swap(a, b)
Build a dictionary to swap RandomSymbols based on their underlying symbol.
i.e. if X = ('x', pspace1) and Y = ('x', pspace2) then X and Y match and the key,
value pair {X:Y} will appear in the result
Inputs: collections a and b of random variables which share common symbols Output:
dict mapping RVs in a to RVs in b
5.30 ODE
These are functions that are imported into the global namespace with from sympy import *.
These functions (unlike Hint Functions (page 1270), below) are intended for use by ordinary
users of SymPy.
dsolve()
Examples
>>> from sympy import Function, dsolve, Eq, Derivative, sin, cos, symbols
>>> from sympy.abc import x
>>> f = Function('f')
>>> dsolve(Derivative(f(x), x, x) + 9*f(x), f(x))
Eq(f(x), C1*sin(3*x) + C2*cos(3*x))
>>> dsolve(eq)
[Eq(x(t), C1*x0(t) + C2*x0(t)*Integral(8*exp(Integral(7*t, t))*exp(Integral(12*t,␣
,→t))/x0(t)**2, t)),
>>> dsolve(eq)
{Eq(x(t), -exp(C1)/(C2*exp(C1) - cos(t))), Eq(y(t), -1/(C1 - cos(t)))}
• See test_ode.py for many tests, which serves also as a set of examples for how to
use dsolve() (page 1262).
• dsolve() (page 1262) always returns an Equality (page 167) class (except for the
case when the hint is all or all_Integral). If possible, it solves the solution explic-
itly for the function being solved for. Otherwise, it returns an implicit solution.
• Arbitrary constants are symbols named C1, C2, and so on.
• Because all solutions should be mathematically equivalent, some hints may return
the exact same result for an ODE. Often, though, two different hints will return the
same solution formatted differently. The two should be equivalent. Also note that
sometimes the values of the arbitrary constants in two different solutions may not
be the same, because one constant may have “absorbed” other constants into it.
• Do help(ode.ode_<hintname>) to get help more information on a specific hint,
where <hintname> is the name of a hint without _Integral.
Usage
dsolve(eq, func) -> Solve a system of ordinary differential equations eq for
func being list of functions including x(t), y(t), z(t) where number of functions
in the list depends upon the number of equations provided in eq.
Details
classify_ode()
Notes
Examples
checkodesol()
Examples
homogeneous_order()
sympy.solvers.ode.homogeneous_order(eq, *symbols)
Returns the order n if g is homogeneous and None if it is not homogeneous.
Examples
infinitesimals()
Solving the above mentioned PDE is not trivial, and can be solved only by making intel-
ligent assumptions for ξ and η (heuristics). Once an infinitesimal is found, the attempt
to find more heuristics stops. This is done to optimise the speed of solving the differ-
ential equation. If a list of all the infinitesimals is needed, hint should be flagged as
all, which gives the complete list of infinitesimals. If the infinitesimals for a particular
heuristic needs to be found, it can be passed as a flag to hint.
References
Examples
checkinfsol()
The infinitesimals should be given in the form of a list of dicts [{xi(x, y): inf, eta(x,
y): inf}], corresponding to the output of the function infinitesimals. It returns a list of
values of the form [(True/False, sol)] where sol is the value obtained after substi-
tuting the infinitesimals in the PDE. If it is True, then sol would be 0.
These functions are intended for internal use by dsolve() (page 1262) and others. Unlike
User Functions (page 1262), above, these are not intended for every-day use by ordinary
SymPy users. Instead, functions such as dsolve() (page 1262) should be used. Nonetheless,
these functions contain useful information in their docstrings on the various ODE solving
methods. For this reason, they are documented here.
allhints
odesimp
Examples
constant_renumber
Examples
Only constants in the given range (inclusive) are renumbered; the renumbering always
starts from 1:
constantsimp
sympy.solvers.ode.constantsimp(expr, constants)
Simplifies an expression with arbitrary constants in it.
This function is written specifically to work with dsolve() (page 1262), and is not in-
tended for general use.
Simplification is done by “absorbing” the arbitrary constants into other arbitrary con-
stants, numbers, and symbols that they are not independent of.
The symbols must all have the same name with numbers after it, for example, C1, C2,
C3. The symbolname here would be ‘C’, the startnumber would be 1, and the endnumber
would be 3. If the arbitrary constants are independent of the variable x, then the inde-
pendent symbol would be x. There is no need to specify the dependent function, such as
f(x), because it already has the independent symbol, x, in it.
Because terms are “absorbed” into arbitrary constants and because constants are
renumbered after simplifying, the arbitrary constants in expr are not necessarily equal
to the ones of the same name in the returned result.
If two or more arbitrary constants are added, multiplied, or raised to the power of each
other, they are first absorbed together into a single arbitrary constant. Then the new
constant is combined into other terms if necessary.
Absorption of constants is done with limited assistance:
1. terms of Add (page 163)s are collected to try join constants so ex (C1 cos(x)+C2 cos(x))
will simplify to ex C1 cos(x);
2. powers with exponents that are Add (page 163)s are expanded so eC1 +x will be sim-
plified to C1 ex .
Use constant_renumber() (page 1271) to renumber constants after simplification or
else arbitrary numbers on constants may appear, e.g. C1 + C3 x.
In rare cases, a single constant can be “simplified” into two constants. Every differential
equation solution should have as many arbitrary constants as the order of the differential
equation. The result here will be technically correct, but it may, for example, have C1
and C2 in an expression, when C1 is actually equal to C2 . Use your discretion in such
situations, and also take advantage of the ability to use hints in dsolve() (page 1262).
Examples
sol_simplicity
Simplicity Return
sol solved for func -2
sol not solved for func but can be -1
sol is not solved nor solvable for func len(str(sol))
sol contains an Integral (page 714) oo
oo here means the SymPy infinity, which should compare greater than any integer.
If you already know solve() (page 1328) cannot solve sol, you can use trysolv-
ing=False to skip that step, which is the only potentially slow step. For example,
dsolve() (page 1262) with the simplify=False flag should do this.
If sol is a list of solutions, if the worst solution in the list returns oo it returns that,
otherwise it returns len(str(sol)), that is, the length of the string representation of
the whole list.
Examples
>>> from sympy import symbols, Function, Eq, tan, cos, sqrt, Integral
>>> from sympy.solvers.ode import ode_sol_simplicity
>>> x, C1, C2 = symbols('x, C1, C2')
>>> f = Function('f')
1st_exact
P (x, y) ∂x + Q(x, y) ∂y = 0
is exact if there is some function F (x, y) such that P (x, y) = ∂F /∂x and Q(x, y) = ∂F /∂y. It
can be shown that a necessary and sufficient condition for a first order ODE to be exact
is that ∂P /∂y = ∂Q/∂x. Then, the solution will be as given below:
Where the first partials of P and Q exist and are continuous in a simply connected region.
A note: SymPy currently has no way to represent inert substitution on an expression,
so the hint 1st_exact_Integral will return an integral with dy. This is supposed to
represent the function that you are solving for.
References
• http://en.wikipedia.org/wiki/Exact_differential_equation
• M. Tenenbaum & H. Pollard, “Ordinary Differential Equations”, Dover 1963, pp. 73
# indirect doctest
Examples
1st_homogeneous_coeff_best
References
• http://en.wikipedia.org/wiki/Homogeneous_differential_equation
• M. Tenenbaum & H. Pollard, “Ordinary Differential Equations”, Dover 1963, pp. 59
# indirect doctest
Examples
1st_homogeneous_coeff_subs_dep_div_indep
such that P and Q are homogeneous and of the same order. A function F (x, y) is homo-
geneous of order n if F (xt, yt) = tn F (x, y). Equivalently, F (x, y) can be rewritten as G(y/x)
or H(x/y). See also the docstring of homogeneous_order() (page 1267).
If the coefficients P and Q in the differential equation above are homogeneous functions
of the same order, then it can be shown that the substitution y = u1 x (i.e. u1 = y/x) will
turn the differential equation into an equation separable in the variables x and u. If h(u1 )
is the function that results from making the substitution u1 = f (x)/x on P (x, f (x)) and g(u2 )
is the function that results from the substitution on Q(x, f (x)) in the differential equation
P (x, f (x)) + Q(x, f (x))f ′ (x) = 0, then the general solution is:
References
• http://en.wikipedia.org/wiki/Homogeneous_differential_equation
• M. Tenenbaum & H. Pollard, “Ordinary Differential Equations”, Dover 1963, pp. 59
# indirect doctest
Examples
1st_homogeneous_coeff_subs_indep_div_dep
such that P and Q are homogeneous and of the same order. A function F (x, y) is homo-
geneous of order n if F (xt, yt) = tn F (x, y). Equivalently, F (x, y) can be rewritten as G(y/x)
or H(x/y). See also the docstring of homogeneous_order() (page 1267).
If the coefficients P and Q in the differential equation above are homogeneous functions
of the same order, then it can be shown that the substitution x = u2 y (i.e. u2 = x/y) will
turn the differential equation into an equation separable in the variables y and u2 . If
h(u2 ) is the function that results from making the substitution u2 = x/f (x) on P (x, f (x))
and g(u2 ) is the function that results from the substitution on Q(x, f (x)) in the differential
equation P (x, f (x)) + Q(x, f (x))f ′ (x) = 0, then the general solution is:
References
• http://en.wikipedia.org/wiki/Homogeneous_differential_equation
• M. Tenenbaum & H. Pollard, “Ordinary Differential Equations”, Dover 1963, pp. 59
# indirect doctest
Examples
1st_linear
These kinds
∫ of differential equations can be solved in a general way. The integrating
factor e P (x) dx will turn the equation into a separable equation. The general solution is:
>>> from sympy import Function, dsolve, Eq, pprint, diff, sin
>>> from sympy.abc import x
>>> f, P, Q = map(Function, ['f', 'P', 'Q'])
>>> genform = Eq(f(x).diff(x) + P(x)*f(x), Q(x))
>>> pprint(genform)
d
(continues on next page)
References
• http://en.wikipedia.org/wiki/Linear_differential_equation#First_order_equation
• M. Tenenbaum & H. Pollard, “Ordinary Differential Equations”, Dover 1963, pp. 92
# indirect doctest
Examples
>>> f = Function('f')
>>> pprint(dsolve(Eq(x*diff(f(x), x) - f(x), x**2*sin(x)),
... f(x), '1st_linear'))
f(x) = x*(C1 - cos(x))
Bernoulli
The substitution w = 1/y 1−n will transform an equation of this form into one that is linear
(see the docstring of ode_1st_linear() (page 1278)). The general solution is:
Note that the equation is separable when n = 1 (see the docstring of ode_separable()
(page 1286)).
References
• http://en.wikipedia.org/wiki/Bernoulli_differential_equation
• M. Tenenbaum & H. Pollard, “Ordinary Differential Equations”, Dover 1963, pp. 95
# indirect doctest
Examples
Liouville
References
• Goldstein and Braun, “Advanced Methods for the Solution of Differential Equations”,
pp. 98
• http://www.maplesoft.com/support/help/Maple/view.aspx?path=odeadvisor/
Liouville
# indirect doctest
Examples
Riccati_special_minus2
While it does not have a general solution [1], the “special” form, dy/dx = ay 2 − bxc , does
have solutions in many cases [2]. This routine returns a solution for a(dy/dx) = by 2 +cy/x+
d/x2 that is obtained by using a suitable change of variables to reduce it to the special
form and is valid when neither a nor b are zero and either c or d is zero.
References
1. http://www.maplesoft.com/support/help/Maple/view.aspx?path=odeadvisor/Riccati
2. http://eqworld.ipmnet.ru/en/solutions/ode/ode0106.pdf - http://eqworld.ipmnet.ru/
en/solutions/ode/ode0123.pdf
nth_linear_constant_coeff_homogeneous
These equations can be solved in a general manner, by taking the roots of the charac-
teristic equation an mn + an−1 mn−1 + · · · + a1 m + a0 = 0. The solution will then be the sum of
Cn xi erx terms, for each where Cn is an arbitrary constant, r is a root of the characteristic
equation and i is one of each from 0 to the multiplicity of the root - 1 (for example, a
root 3 of multiplicity 2 would create the terms C1 e3x + C2 xe3x ). The exponential is usually
expanded for complex roots using Euler’s equation eIx = cos(x) + I sin(x). Complex roots
always come in conjugate pairs in polynomials with real coefficients, so the two roots
will be represented (after simplifying the constants) as eax (C1 cos(bx) + C2 sin(bx)).
If SymPy cannot find exact roots to the characteristic equation, a CRootOf instance will
be return instead.
Note that because this method does not involve integration, there is no
nth_linear_constant_coeff_homogeneous_Integral hint.
The following is for internal use:
• returns = 'sol' returns the solution to the ODE.
• returns = 'list' returns a list of linearly independent solutions, for use with
non homogeneous solution methods like variation of parameters and undeter-
mined coefficients. Note that, though the solutions should be linearly inde-
pendent, this function does not explicitly check that. You can do assert sim-
plify(wronskian(sollist)) != 0 to check for linear independence. Also, assert
len(sollist) == order will need to pass.
• returns = 'both', return a dictionary {'sol': <solution to ODE>, 'list':
<list of linearly independent solutions>}.
References
# indirect doctest
Examples
nth_linear_constant_coeff_undetermined_coefficients
sympy.solvers.ode.ode_nth_linear_constant_coeff_undetermined_coefficients(eq,
func,
or-
der,
match)
Solves an nth order linear differential equation with constant coefficients using the
method of undetermined coefficients.
This method works on differential equations of the form
where P (x) is a function that has a finite number of linearly independent derivatives.
Functions that fit this requirement are finite sums functions of the form axi ebx sin(cx + d)
or axi ebx cos(cx + d), where i is a non-negative integer and a, b, c, and d are constants.
For example any polynomial in x, functions like x2 e2x , x sin(x), and ex cos(x) can all be
used. Products of sin’s and cos’s have a finite number of derivatives, because they can
be expanded into sin(ax) and cos(bx) terms. However, SymPy currently cannot do that
expansion, so you will need to manually rewrite the expression in terms of the above
2
to use this method. So, for example, you will need to manually convert sin (x) into (1 +
cos(2x))/2 to properly apply the method of undetermined coefficients on it.
This method works by creating a trial function from the expression and all of its linear
independent derivatives and substituting them into the original ODE. The coefficients for
each term will be a system of linear equations, which are be solved for and substituted,
giving the solution. If any of the trial functions are linearly dependent on the solution
to the homogeneous equation, they are multiplied by sufficient x to make them linearly
independent.
References
• http://en.wikipedia.org/wiki/Method_of_undetermined_coefficients
• M. Tenenbaum & H. Pollard, “Ordinary Differential Equations”, Dover 1963, pp. 221
# indirect doctest
Examples
nth_linear_constant_coeff_variation_of_parameters
sympy.solvers.ode.ode_nth_linear_constant_coeff_variation_of_parameters(eq,
func,
or-
der,
match)
Solves an nth order linear differential equation with constant coefficients using the
method of variation of parameters.
This method works on any differential equations of the form
This method works by assuming that the particular solution takes the form
∑
n
ci (x)yi (x),
x=1
where yi is the ith solution to the homogeneous equation. The solution is then solved
using Wronskian’s and Cramer’s Rule. The particular solution is given by
∑n (∫ )
Wi (x)
dx yi (x),
x=1
W (x)
where W (x) is the Wronskian of the fundamental system (the system of n linearly in-
dependent solutions to the homogeneous equation), and Wi (x) is the Wronskian of the
fundamental system with the ith column replaced with [0, 0, · · · , 0, P (x)].
This method is general enough to solve any nth order inhomogeneous linear
differential equation with constant coefficients, but sometimes SymPy cannot
simplify the Wronskian well enough to integrate it. If this method hangs, try
using the nth_linear_constant_coeff_variation_of_parameters_Integral
hint and simplifying the integrals manually. Also, prefer using
nth_linear_constant_coeff_undetermined_coefficients when it applies, because it
doesn’t use integration, making it faster and more reliable.
Warning, using simplify=False with ‘nth_linear_constant_coeff_variation_of_parameters’
in dsolve() (page 1262) may cause it to hang, because it will not attempt to simplify the
Wronskian before integrating. It is recommended that you only use simplify=False with
‘nth_linear_constant_coeff_variation_of_parameters_Integral’ for this method, especially
if the solution to the homogeneous equation has trigonometric functions in it.
References
• http://en.wikipedia.org/wiki/Variation_of_parameters
• http://planetmath.org/VariationOfParameters
• M. Tenenbaum & H. Pollard, “Ordinary Differential Equations”, Dover 1963, pp. 233
# indirect doctest
Examples
separable
References
# indirect doctest
Examples
almost_linear
This can be solved by substituting l(y) = u(y). Making the given substitution reduces it
to a linear differential equation of the form u′ + P (x)u + Q(x) = 0.
The general solution is
See also:
sympy.solvers.ode.ode_1st_linear() (page 1278)
References
Examples
linear_coefficients
References
Examples
separable_reduced
y ′ + (y/x)H(xn y) = 0.
This can be solved by substituting u(y) = xn y. The equation then reduces to the separable
u′
form u(power−H(u)) − x1 = 0.
The general solution is:
See also:
sympy.solvers.ode.ode_separable() (page 1286)
References
Examples
lie_group
After finding the solution by integration, it is then converted back to the original coordi-
nate system by substituting r and s in terms of x and y again.
References
Examples
1st_power_series
where y(x0 ) = b is the value of y at the initial value of x0 . To compute the values of
the Fn (x0 , b) the following algorithm is followed, until the required number of terms are
generated.
1. F1 = h(x0 , b)
∂Fn ∂Fn
2. Fn+1 = ∂x + ∂y F1
References
• Travis W. Walker, Analytic power series technique for solving first-order differential
equations, p.p 17, 18
Examples
2nd_power_series_ordinary
References
• http://tutorial.math.lamar.edu/Classes/DE/SeriesSolutions.aspx
• George E. Simmons, “Differential Equations with Applications and Historical Notes”,
p.p 176 - 184
Examples
2nd_power_series_regular
∑∞
The power series solution is of the form xm n
n=0 an x . The coefficients are determined
∑ n−1
q +(m+k)pn−k
by the following recurrence relation. an = − k=0 n−kf (m+n) . For the case in which
m1 − m2 is an integer, it can be seen from the recurrence relation that for the lower root
m, when n equals the difference of both the roots, the denominator becomes zero. So if
the numerator is not equal to zero, a second series solution exists.
References
Examples
These functions are intended for internal use of the Lie Group Solver. Nonetheless, they
contain useful information in their docstrings on the algorithms implemented for the various
heuristics.
abaco1_simple
sympy.solvers.ode.lie_heuristic_abaco1_simple(match, comp=False)
The first heuristic uses the following four sets of assumptions on ξ and η
ξ = 0, η = f (x)
ξ = 0, η = f (y)
ξ = f (x), η = 0
ξ = f (y), η = 0
The success of this heuristic is determined by algebraic factorisation. For the first as-
sumption ξ = 0 and η to be a function of x, the PDE
∂η ∂η ∂ξ ∂ξ ∂h ∂h
+( − )∗h− ∗ h2 − ξ ∗ −η∗ =0
∂x ∂y ∂x ∂y ∂x ∂y
reduces to f ′ (x)−f ∂h ∂h
∂y = 0 If ∂y is a function of x, then this can usually be integrated easily.
A similar idea is applied to the other 3 assumptions as well.
References
• E.S Cheb-Terrab, L.G.S Duarte and L.A,C.P da Mota, Computer Algebra Solving of
First Order ODEs Using Symmetry Methods, pp. 8
abaco1_product
sympy.solvers.ode.lie_heuristic_abaco1_product(match, comp=False)
The second heuristic uses the following two assumptions on ξ and η
η = 0, ξ = f (x) ∗ g(y)
η = f (x) ∗ g(y), ξ = 0
2
∂
The first assumption of this heuristic holds good if h12 ∂x∂y log(h) is separable in x and y,
then the separated factors containing x is f (x), and g(y) is obtained by
∫
e
∂
f ∂x ( f ∗h
1
) dy
( )
∂ 1
provided f ∂x f ∗h is a function of y only.
dy dy 1
The second assumption holds good if dx = h(x, y) is rewritten as dx = h(y,x) and the same
properties of the first assumption satisfies. After obtaining f (x) and g(y), the coordinates
are again interchanged, to get η as f (x) ∗ g(y)
References
• E.S. Cheb-Terrab, A.D. Roche, Symmetries and First Order ODE Patterns, pp. 7 -
pp. 8
bivariate
sympy.solvers.ode.lie_heuristic_bivariate(match, comp=False)
The third heuristic assumes the infinitesimals ξ and η to be bi-variate polynomials in x
and y. The assumption made here for the logic below is that h is a rational function in x
and y though that may not be necessary for the infinitesimals to be bivariate polynomials.
The coefficients of the infinitesimals are found out by substituting them in the PDE and
grouping similar terms that are polynomials and since they form a linear system, solve
and check for non trivial solutions. The degree of the assumed bivariates are increased
till a certain maximum value.
References
chi
sympy.solvers.ode.lie_heuristic_chi(match, comp=False)
dχ
The aim of the fourth heuristic is to find the function χ(x, y) that satisfies the PDE dx +
dx − ∂y χ = 0.
h dχ ∂h
References
• E.S Cheb-Terrab, L.G.S Duarte and L.A,C.P da Mota, Computer Algebra Solving of
First Order ODEs Using Symmetry Methods, pp. 8
abaco2_similar
sympy.solvers.ode.lie_heuristic_abaco2_similar(match, comp=False)
This heuristic uses the following two assumptions on ξ and η
η = g(x), ξ = f (x)
η = f (y), ξ = g(y)
For the first assumption,
∂h
1. First ∂y
∂2 h
is calculated. Let us say this value is A
∂yy
∫ A(x)
y dx
e C
2. If this is constant, then h is matched to the form A(x) + B(x)e C then, B(x) gives
f (x) and A(x) ∗ f (x) gives g(x)
∂A
3. Otherwise ∂X
∂A = γ is calculated. If
∂Y
a] γ is a function of x alone
γ ∂h −γ ′ (x)− ∂h ∫
b] ∂y h+γ
∂x
= G is a function of x alone. then, e G dx
gives f (x) and −γ ∗ f (x) gives
g(x)
dy dy 1
The second assumption holds good if dx = h(x, y) is rewritten as dx = h(y,x) and the same
properties of the first assumption satisfies. After obtaining f (x) and g(x), the coordinates
are again interchanged, to get ξ as f (x∗ ) and η as g(y ∗ )
References
• E.S. Cheb-Terrab, A.D. Roche, Symmetries and First Order ODE Patterns, pp. 10 -
pp. 12
function_sum
sympy.solvers.ode.lie_heuristic_function_sum(match, comp=False)
This heuristic uses the following two assumptions on ξ and η
η = 0, ξ = f (x) + g(y)
η = f (x) + g(y), ξ = 0
The first assumption of this heuristic holds good if
∂ ∂2
[(h 2 (h−1 ))−1 ]
∂y ∂x
is separable in x and y,
∂g
1. The separated factors containing y is ∂y . From this g(y) can be determined.
2. The separated factors containing x is f ′′ (x).
∂2
−1 f ′′ (x)
3. h ∂x 2 (h ) equals f (x)+g(y) . From this f (x) can be determined.
dy dy 1
The second assumption holds good if dx = h(x, y) is rewritten as dx = h(y,x) and the same
properties of the first assumption satisfies. After obtaining f (x) and g(y), the coordinates
are again interchanged, to get η as f (x) + g(y).
For both assumptions, the constant factors are separated among g(y) and f ′′ (x), such
that f ′′ (x) obtained from 3] is the same as that obtained from 2]. If not possible, then
this heuristic fails.
References
• E.S. Cheb-Terrab, A.D. Roche, Symmetries and First Order ODE Patterns, pp. 7 -
pp. 8
abaco2_unique_unknown
sympy.solvers.ode.lie_heuristic_abaco2_unique_unknown(match, comp=False)
This heuristic assumes the presence of unknown functions or known functions with non-
integer powers.
1. A list of all functions and non-integer powers containing x and y
∂f
2. Loop over each element f in the list, find ∂x
∂f =R
∂x
References
• E.S. Cheb-Terrab, A.D. Roche, Symmetries and First Order ODE Patterns, pp. 10 -
pp. 12
abaco2_unique_general
sympy.solvers.ode.lie_heuristic_abaco2_unique_general(match, comp=False)
This heuristic finds if infinitesimals of the form η = f (x), ξ = g(y) without making any
assumptions on h.
The complete sequence of steps is given in the paper mentioned below.
References
• E.S. Cheb-Terrab, A.D. Roche, Symmetries and First Order ODE Patterns, pp. 10 -
pp. 12
linear
sympy.solvers.ode.lie_heuristic_linear(match, comp=False)
This heuristic assumes
1. ξ = ax + by + c and
2. η = f x + gy + h
After substituting the following assumptions in the determining PDE, it reduces to
∂h ∂h
f + (g − a)h − bh2 − (ax + by + c) − (f x + gy + c)
∂x ∂y
Solving the reduced PDE obtained, using the method of characteristics, becomes imprac-
tical. The method followed is grouping similar terms and solving the system of linear
equations obtained. The difference between the bivariate heuristic is that h need not be
a rational function in this case.
References
• E.S. Cheb-Terrab, A.D. Roche, Symmetries and First Order ODE Patterns, pp. 10 -
pp. 12
These functions are intended for internal use by dsolve() (page 1262) for system of differ-
ential equations.
system_of_odes_linear_2eq_order1_type1
sympy.solvers.ode._linear_2eq_order1_type1(x, y, t, r, eq)
It is classified under system of two linear homogeneous first-order constant-coefficient
ordinary differential equations.
The equations which come under this type are
x′ = ax + by,
y ′ = cx + dy
The characteristics equation is written as
λ2 + (a + d)λ + ad − bc = 0
and its discriminant is D = (a − d)2 + 4bc. There are several cases
1. Case when ad − bc ̸= 0. The origin of coordinates, x = y = 0, is the only stationary point;
it is - a node if D = 0 - a node if D > 0 and ad − bc > 0 - a saddle if D > 0 and ad − bc < 0 - a
focus if D < 0 and a + d ̸= 0 - a centre if D < 0 and a + d ̸= 0.
1.1. If D > 0. The characteristic equation has two distinct real roots λ1 and λ2 . The
general solution of the system in question is expressed as
x = C1 beλ1 t + C2 beλ2 t
system_of_odes_linear_2eq_order1_type2
sympy.solvers.ode._linear_2eq_order1_type2(x, y, t, r, eq)
The equations of this type are
x′ = ax + by + k1, y ′ = cx + dy + k2
The general solution of this system is given by sum of its particular solution and the
general solution of the corresponding homogeneous system is obtained from type1.
1. When ad − bc ̸= 0. The particular solution will be x = x0 and y = y0 where x0 and y0 are
determined by solving linear system of equations
y = kx + (c2 − c1 k)t
2.2 If σ = a + bk = 0, particular solution is given by
1
x= b(c2 − c1 k)t2 + c1 t
2
y = kx + (c2 − c1 k)t
system_of_odes_linear_2eq_order1_type3
sympy.solvers.ode._linear_2eq_order1_type3(x, y, t, r, eq)
The equations of this type of ode are
x′ = f (t)x + g(t)y
y ′ = g(t)x + f (t)y
The solution of such equations is given by
system_of_odes_linear_2eq_order1_type4
sympy.solvers.ode._linear_2eq_order1_type4(x, y, t, r, eq)
The equations of this type of ode are .
x′ = f (t)x + g(t)y
y ′ = −g(t)x + f (t)y
The solution is given by
system_of_odes_linear_2eq_order1_type5
sympy.solvers.ode._linear_2eq_order1_type5(x, y, t, r, eq)
The equations of this type of ode are .
x′ = f (t)x + g(t)y
u′ (T ) = v, v ′ (T ) = au + bv
system_of_odes_linear_2eq_order1_type6
sympy.solvers.ode._linear_2eq_order1_type6(x, y, t, r, eq)
The equations of this type of ode are .
x′ = f (t)x + g(t)y
and on substituting the value of y in first equation give rise to first order ODEs. After
solving for x, we can obtain y by substituting the value of x in second equation.
system_of_odes_linear_2eq_order1_type7
sympy.solvers.ode._linear_2eq_order1_type7(x, y, t, r, eq)
The equations of this type of ode are .
x′ = f (t)x + g(t)y
y ′ = h(t)x + p(t)y
Differentiating the first equation and substituting the value of y from second equation
will give a second-order linear equation
This above equation can be easily integrated if following conditions are satisfied.
1. f gp − g 2 h + f g ′ − f ′ g = 0
2. f gp − g 2 h + f g ′ − f ′ g = ag, f g + gp + g ′ = bg
If first condition is satisfied then it is solved by current dsolve solver and in second case
it becomes a constant coefficient differential equation which is also solved by current
solver.
Otherwise if the above condition fails then, a particular solution is assumed as x = x0 (t)
and y = y0 (t) Then the general solution is expressed as
∫
g(t)F (t)P (t)
x = C1 x0 (t) + C2 x0 (t) dt
x20 (t)
∫
F (t)P (t) g(t)F (t)P (t)
y = C1 y0 (t) + C2 [ + y0 (t) dt]
x0 (t) x20 (t)
where C1 and C2 are arbitrary constants and
∫ ∫
f (t) dt p(t) dt
F (t) = e , P (t) = e
system_of_odes_linear_2eq_order2_type1
sympy.solvers.ode._linear_2eq_order2_type1(x, y, t, r, eq)
System of two constant-coefficient second-order linear homogeneous differential equa-
tions
x′′ = ax + by
y ′′ = cx + dy
The characteristic equation for above equations
λ4 − (a + d)λ2 + ad − bc = 0
1.1. If D ̸= 0. The characteristic equation has four distinct roots, λ1 , λ2 , λ3 , λ4 . The general
solution of the system is
√
where C1 , ..., C4 are arbitrary constants and k = 2(a + d)
1.3. If D = 0 and a = d ̸= 0 and b = 0:
√ √ √ √
x = 2 aC1 e at + 2 aC2 e− at
√ √ √ √
y = cC1 te at
− cC2 te− at
+ C3 e at
+ C4 e− at
√ √ √ √
y = 2 aC1 e at + 2 aC2 e− at
2. When ad − bc = 0 and a2 + b2 > 0. Then the original system becomes
x′′ = ax + by
y ′′ = k(ax + by)
2.1. If a + bk ̸= 0:
√ √
x = C1 et a+bk
+ C2 e−t a+bk
+ C3 bt + C4 b
√ √
y = C1 ket a+bk
+ C2 ke−t a+bk
− C3 at − C4 a
2.2. If a + bk = 0:
x = C1 bt3 + C2 bt2 + C3 t + C4
y = kx + 6C1 t + 2C2
system_of_odes_linear_2eq_order2_type2
sympy.solvers.ode._linear_2eq_order2_type2(x, y, t, r, eq)
The equations in this type are
x′′ = a1 x + b1 y + c1
y ′′ = a2 x + b2 y + c2
The general solution of this system is given by the sum of its particular solution and the
general solution of the homogeneous system. The general solution is given by the linear
system of 2 equation of order 2 and type 1
1. If a1 b2 − a2 b1 ̸= 0. A particular solution will be x = x0 and y = y0 where the constants x0
and y0 are determined by solving the linear algebraic system
a1 x0 + b1 y0 + c1 = 0, a2 x0 + b2 y0 + c2 = 0
2. If a1 b2 − a2 b1 = 0 and a21 + b21 > 0. In this case, the system in question becomes
x′′ = ax + by + c1 , y ′′ = k(ax + by) + c2
2.1. If σ = a + bk ̸= 0, the particular solution will be
1 −1
x= bσ (c1 k − c2 )t2 − σ −2 (ac1 + bc2 )
2
1
y = kx + (c2 − c1 k)t2
2
2.2. If σ = a + bk = 0, the particular solution will be
1 1
x= b(c2 − c1 k)t4 + c1 t2
24 2
1
y = kx + (c2 − c1 k)t2
2
system_of_odes_linear_2eq_order2_type3
sympy.solvers.ode._linear_2eq_order2_type3(x, y, t, r, eq)
These type of equation is used for describing the horizontal motion of a pendulum taking
into account the Earth rotation. The solution is given with a2 + 4b > 0:
system_of_odes_linear_2eq_order2_type4
sympy.solvers.ode._linear_2eq_order2_type4(x, y, t, r, eq)
These equations are found in the theory of oscillations
x′′ + a1 x′ + b1 y ′ + c1 x + d1 y = k1 eiωt
y ′′ + a2 x′ + b2 y ′ + c2 x + d2 y = k2 eiωt
The general solution of this linear nonhomogeneous system of constant-coefficient dif-
ferential equations is given by the sum of its particular solution and the general solution
of the corresponding homogeneous system (with k1 = k2 = 0)
1. A particular solution is obtained by the method of undetermined coefficients:
x = A∗ eiωt , y = B∗ eiωt
On substituting these expressions into the original system of differential equations, one
arrive at a linear nonhomogeneous system of algebraic equations for the coefficients A
and B.
2. The general solution of the homogeneous system of differential equations is deter-
mined by a linear combination of linearly independent particular solutions determined
by the method of undetermined coefficients in the form of exponentials:
x = Aeλt , y = Beλt
On substituting these expressions into the original system and collecting the coefficients
of the unknown A and B, one obtains
(λ2 + a1 λ + c1 )A + (b1 λ + d1 )B = 0
(a2 λ + c2 )A + (λ2 + b2 λ + d2 )B = 0
The determinant of this system must vanish for nontrivial solutions A, B to exist. This
requirement results in the following characteristic equation for λ
If all roots k1 , ..., k4 of this equation are distinct, the general solution of the original system
of the differential equations has the form
system_of_odes_linear_2eq_order2_type5
sympy.solvers.ode._linear_2eq_order2_type5(x, y, t, r, eq)
The equation which come under this category are
x′′ = a(ty ′ − y)
y ′′ = b(tx′ − x)
The transformation
u = tx′ − x, b = ty ′ − y
u′ = atv, v ′ = btu
If ab < 0:
1√ 1√
u = C1 a cos( |ab|t2 ) + C2 a sin(− |ab|t2 )
2 2
√ 1√ √ 1√
v = C1 |ab| sin( |ab|t2 ) + C2 |ab| cos(− |ab|t2 )
2 2
where C1 and C2 are arbitrary constants. On substituting the value of u and v in above
equations and integrating the resulting expressions, the general solution will become
∫ ∫
u u
x = C3 t + t dt, y = C 4 t + t dt
t2 t2
where C3 and C4 are arbitrary constants.
system_of_odes_linear_2eq_order2_type6
sympy.solvers.ode._linear_2eq_order2_type6(x, y, t, r, eq)
The equations are
x′′ = f (t)(a1 x + b1 y)
y ′′ = f (t)(a2 x + b2 y)
If k1 and k2 are roots of the quadratic equation
k 2 − (a1 + b2 )k + a1 b2 − a2 b1 = 0
system_of_odes_linear_2eq_order2_type7
sympy.solvers.ode._linear_2eq_order2_type7(x, y, t, r, eq)
The equations are given as
x′′ = f (t)(a1 x′ + b1 y ′ )
y ′′ = f (t)(a2 x′ + b2 y ′ )
If k1 and ‘k_2‘ are roots of the quadratic equation
k 2 − (a1 + b2 )k + a1 b2 − a2 b1 = 0
Then the system can be reduced by adding together the two equations multiplied by
appropriate constants give following two independent equations:
system_of_odes_linear_2eq_order2_type8
sympy.solvers.ode._linear_2eq_order2_type8(x, y, t, r, eq)
The equation of this category are
x′′ = af (t)(ty ′ − y)
y ′′ = bf (t)(tx′ − x)
The transformation
u = tx′ − x, v = ty ′ − y
√ √ ∫ √ √ ∫
v = C1 abe ab tf (t) dt − C2 abe− ab tf (t) dt
If ab < 0:
√ ∫ √ ∫
u = C1 a cos( |ab| tf (t) dt) + C2 a sin(− |ab| tf (t) dt)
√ √ ∫ √ √ ∫
v = C1 |ab| sin( |ab| tf (t) dt) + C2 |ab| cos(− |ab| tf (t) dt)
where C1 and C2 are arbitrary constants. On substituting the value of u and v in above
equations and integrating the resulting expressions, the general solution will become
∫ ∫
u u
x = C3 t + t 2
dt, y = C4 t + t dt
t t2
system_of_odes_linear_2eq_order2_type9
sympy.solvers.ode._linear_2eq_order2_type9(x, y, t, r, eq)
t2 x′′ + a1 tx′ + b1 ty ′ + c1 x + d1 y = 0
t2 y ′′ + a2 tx′ + b2 ty ′ + c2 x + d2 y = 0
These system of equations are euler type.
The substitution of t = σeτ (σ ̸= 0) leads to the system of constant coefficient linear differ-
ential equations
y ′′ + a2 x′ + (b2 − 1)y ′ + c2 x + d2 y = 0
The general solution of the homogeneous system of differential equations is determined
by a linear combination of linearly independent particular solutions determined by the
method of undetermined coefficients in the form of exponentials
x = Aeλt , y = Beλt
On substituting these expressions into the original system and collecting the coefficients
of the unknown A and B, one obtains
If all roots k1 , ..., k4 of this equation are distinct, the general solution of the original system
of the differential equations has the form
y = C1 (λ21 + (a1 − 1)λ1 + c1 )eλ1 t + C2 (λ22 + (a1 − 1)λ2 + c1 )eλ2 t + C3 (λ23 + (a1 − 1)λ3 + c1 )eλ3 t + C4 (λ24 + (a1 − 1)λ4 +
system_of_odes_linear_2eq_order2_type10
sympy.solvers.ode._linear_2eq_order2_type10(x, y, t, r, eq)
The equation of this category are
(αt2 + βt + γ)2 y ′′ = cx + dy
The transformation
∫
1 x y
τ= dt, u = √ ,v = √
αt2 + βt + γ |αt + βt + γ|
2 |αt + βt + γ|
2
system_of_odes_linear_2eq_order2_type11
sympy.solvers.ode._linear_2eq_order2_type11(x, y, t, r, eq)
The equations which comes under this type are
y ′′ = h(t)(tx′ − x) + p(t)(ty ′ − y)
The transformation
u = tx′ − x, v = ty ′ − y
system_of_odes_linear_3eq_order1_type1
sympy.solvers.ode._linear_3eq_order1_type1(x, y, z, t, r, eq)
x′ = ax
y ′ = bx + cy
z ′ = dx + ky + pz
Solution of such equations are forward substitution. Solving first equations gives the
value of x, substituting it in second and third equation and solving second equation gives
y and similarly substituting y in third equation give z.
x = C1 eat
bC1 at
y= e + C2 ect
a−c
C1 bk kC2 ct
z= (d + )eat + e + C3 ept
a−p a−c c−p
where C1 , C2 and C3 are arbitrary constants.
system_of_odes_linear_3eq_order1_type2
sympy.solvers.ode._linear_3eq_order1_type2(x, y, z, t, r, eq)
The equations of this type are
x′ = cy − bz
y ′ = az − cx
z ′ = bx − ay
1. First integral:
ax + by + cz = A − (1)
x2 + y 2 + z 2 = B 2 − (2)
where A and B are arbitrary constants. It follows from these integrals that the integral
lines are circles formed by the intersection of the planes (1) and sphere (2)
2. Solution:
x = aC0 + kC1 cos(kt) + (cC2 − bC3 ) sin(kt)
y = bC0 + kC2 cos(kt) + (aC2 − cC3 ) sin(kt)
z = cC0 + kC3 cos(kt) + (bC2 − aC3 ) sin(kt)
√
where k = a2 + b2 + c2 and the four constants of integration, C1 , ..., C4 are constrained by
a single relation,
system_of_odes_linear_3eq_order1_type3
sympy.solvers.ode._linear_3eq_order1_type3(x, y, z, t, r, eq)
Equations of this system of ODEs
ax′ = bc(y − z)
by ′ = ac(z − x)
cz ′ = ab(x − y)
1. First integral:
a2 x + b2 y + c2 z = A
where A is an arbitrary constant. It follows that the integral lines are plane curves.
2. Solution:
x = C0 + kC1 cos(kt) + a−1 bc(C2 − C3 ) sin(kt)
y = C0 + kC2 cos(kt) + ab−1 c(C3 − C1 ) sin(kt)
z = C0 + kC3 cos(kt) + abc−1 (C1 − C2 ) sin(kt)
√
where k = a2 + b2 + c2 and the four constants of integration, C1 , ..., C4 are constrained by
a single relation
a2 C1 + b2 C2 + c2 C3 = 0
system_of_odes_linear_3eq_order1_type4
sympy.solvers.ode._linear_3eq_order1_type4(x, y, z, t, r, eq)
Equations:
u′ = a1 u + a2 v + a3 w
v ′ = b1 u + b2 v + b3 w
w′ = c1 u + c2 v + c3 w
These system of equations are solved by homogeneous linear system of constant coef-
ficients of n equations of first order. Then substituting the value of u, v and w in trans-
formed equation gives value of x, y and z.
system_of_odes_linear_neq_order1_type1
sympy.solvers.ode._linear_neq_order1_type1(match_)
System of n first-order constant-coefficient linear nonhomogeneous differential equation
or that can be written as y⃗′ = A.⃗y where ⃗y is matrix of yk for k = 1, 2, ...n and A is a n × n
matrix.
Since these equations are equivalent to a first order homogeneous linear differential
equation. So the general solution will contain n linearly independent parts and solution
will consist some type of exponential functions. Assuming y = ⃗v ert is a solution of the
system where ⃗v is a vector of coefficients of y1 , ..., yn . Substituting y and y ′ = rvert into the
equation y⃗′ = A.⃗y , we get
r⃗v = A⃗v
where r comes out to be eigenvalue of A and vector ⃗v is the eigenvector of A correspond-
ing to r. There are three possibilities of eigenvalues of A
• n distinct real eigenvalues
• complex conjugate eigenvalues
• eigenvalues with multiplicity k
1. When all eigenvalues r1 , .., rn are distinct with n different eigenvectors v1 , ...vn then the
solution is given by
(A − rI).w⃗2 = w
⃗
(A − rI).w⃗3 = w⃗2
..
.
(A − rI).w⃗k = wk−1
⃗
2
Then the solutions to the system for the eigenspace are ert [w], ⃗ + w⃗2 ], ert [ t2 w
⃗ ert [tw ⃗ + tw⃗2 +
k−1 k−2
rt t t
w⃗3 ], ..., e [ (k−1)! w
⃗ + (k−2)! w⃗2 + ... + twk−1
⃗ + w⃗k ]
So, If y⃗1 , ..., y⃗n are n solution of obtained from three categories of A, then general solution
to the system y⃗′ = A.⃗y
system_of_odes_nonlinear_2eq_order1_type1
sympy.solvers.ode._nonlinear_2eq_order1_type1(x, y, t, eq)
Equations:
x′ = xn F (x, y)
y ′ = g(y)F (x, y)
Solution:
∫
1
x = φ(y), dy = t + C2
g(y)F (φ(y), y)
where
if n ̸= 1
∫
1 1
φ = [C1 + (1 − n) dy] 1−n
g(y)
if n = 1
∫ 1
dy
φ = C1 e g(y)
system_of_odes_nonlinear_2eq_order1_type2
sympy.solvers.ode._nonlinear_2eq_order1_type2(x, y, t, eq)
Equations:
x′ = eλx F (x, y)
y ′ = g(y)F (x, y)
Solution:
∫
1
x = φ(y), dy = t + C2
g(y)F (φ(y), y)
where
if λ ̸= 0
∫
1 1
φ = − log(C1 − λ dy)
λ g(y)
if λ = 0
∫
1
φ = C1 + dy
g(y)
system_of_odes_nonlinear_2eq_order1_type3
sympy.solvers.ode._nonlinear_2eq_order1_type3(x, y, t, eq)
Autonomous system of general form
x′ = F (x, y)
y ′ = G(x, y)
Assuming y = y(x, C1 ) where C1 is an arbitrary constant is the general solution of the
first-order equation
Then the general solution of the original system of equations has the form
∫
1
dx = t + C1
F (x, y(x, C1 ))
system_of_odes_nonlinear_2eq_order1_type4
sympy.solvers.ode._nonlinear_2eq_order1_type4(x, y, t, eq)
Equation:
x′ = f1 (x)g1 (y)ϕ(x, y, t)
y ′ = f2 (x)g2 (y)ϕ(x, y, t)
First integral:
∫ ∫
f2 (x) g1 (y)
dx − dy = C
f1 (x) g2 (y)
system_of_odes_nonlinear_2eq_order1_type5
sympy.solvers.ode._nonlinear_2eq_order1_type5(func, t, eq)
Clairaut system of ODEs
x = tx′ + F (x′ , y ′ )
y = ty ′ + G(x′ , y ′ )
The following are solutions of the system
(i) straight lines:
x = C1 t + F (C1 , C2 ), y = C2 t + G(C1 , C2 )
system_of_odes_nonlinear_3eq_order1_type1
sympy.solvers.ode._nonlinear_3eq_order1_type1(x, y, z, t, eq)
Equations:
First Integrals:
ax2 + by 2 + cz 2 = C1
a2 x2 + b2 y 2 + c2 z 2 = C2
where C1 and C2 are arbitrary constants. On solving the integrals for y and z and on
substituting the resulting expressions into the first equation of the system, we arrives at
a separable first-order equation on x. Similarly doing that for other two equations, we
will arrive at first order equation on y and z too.
References
-http://eqworld.ipmnet.ru/en/solutions/sysode/sode0401.pdf
system_of_odes_nonlinear_3eq_order1_type2
sympy.solvers.ode._nonlinear_3eq_order1_type2(x, y, z, t, eq)
Equations:
by ′ = (c − a)zxf (x, y, z, t)
cz ′ = (a − b)xyf (x, y, z, t)
First Integrals:
ax2 + by 2 + cz 2 = C1
a2 x2 + b2 y 2 + c2 z 2 = C2
where C1 and C2 are arbitrary constants. On solving the integrals for y and z and on
substituting the resulting expressions into the first equation of the system, we arrives at
a first-order differential equations on x. Similarly doing that for other two equations we
will arrive at first order equation on y and z.
References
-http://eqworld.ipmnet.ru/en/solutions/sysode/sode0402.pdf
system_of_odes_nonlinear_3eq_order1_type3
sympy.solvers.ode._nonlinear_3eq_order1_type3(x, y, z, t, eq)
Equations:
References
-http://eqworld.ipmnet.ru/en/solutions/sysode/sode0404.pdf
system_of_odes_nonlinear_3eq_order1_type4
sympy.solvers.ode._nonlinear_3eq_order1_type4(x, y, z, t, eq)
Equations:
where Fn = Fn (x, y, z, t)
1. First integral:
ax2 + by 2 + cz 2 = C1
where C is an arbitrary constant.
2. Assuming the function Fn is independent of t: Fn = Fn (x, y, z). Then on eliminating t
and z from the first two equations of the system, one arrives at the first-order equation
References
-http://eqworld.ipmnet.ru/en/solutions/sysode/sode0405.pdf
system_of_odes_nonlinear_3eq_order1_type5
sympy.solvers.ode._nonlinear_3eq_order1_type5(x, y, t, eq)
References
-http://eqworld.ipmnet.ru/en/solutions/sysode/sode0406.pdf
This module contains dsolve() (page 1262) and different helper functions that it uses.
dsolve() (page 1262) solves ordinary differential equations. See the docstring on the various
functions for their uses. Note that partial differential equations support is in pde.py. Note
that hint functions have docstrings describing their various methods, but they are intended
for internal use. Use dsolve(ode, func, hint=hint) to solve an ODE using a specific hint.
See also the docstring on dsolve() (page 1262).
Functions in this module
These are the user functions in this module:
• dsolve() (page 1262) - Solves ODEs.
• classify_ode() (page 1265) - Classifies ODEs into possible hints for dsolve()
(page 1262).
• checkodesol() (page 1267) - Checks if an equation is the solution to an ODE.
• homogeneous_order() (page 1267) - Returns the homogeneous order of an ex-
pression.
• infinitesimals() (page 1268) - Returns the infinitesimals of the Lie group of
point transformations of an ODE, such that it is invariant.
• ode_checkinfsol() - Checks if the given infinitesimals are the actual infinites-
imals of a first order ODE.
These are the non-solver helper functions that are for internal use. The user should
use the various options to dsolve() (page 1262) to obtain the functionality provided
by these functions:
• odesimp() (page 1270) - Does all forms of ODE simplification.
• ode_sol_simplicity() (page 1273) - A key function for comparing solutions
by simplicity.
• constantsimp() (page 1272) - Simplifies arbitrary constants.
• constant_renumber() (page 1271) - Renumber arbitrary constants.
• _handle_Integral() - Evaluate unevaluated Integrals.
See also the docstrings of these functions.
Currently implemented solver methods
The following methods are implemented for solving ordinary differential equations. See the
docstrings of the various hint functions for more information on each (run help(ode)):
• 1st order separable differential equations.
• 1st order differential equations whose coefficients or dx and dy are functions homoge-
neous of the same order.
• 1st order exact differential equations.
• 1st order linear differential equations.
• 1st order Bernoulli differential equations.
• Power series solutions for first order differential equations.
• Lie Group method of solving first order differential equations.
Finally, as with every contribution to SymPy, your method will need to be tested. Add a
test for each method in test_ode.py. Follow the conventions there, i.e., test the solver us-
ing dsolve(eq, f(x), hint=your_hint), and also test the solution using checkodesol()
(page 1267) (you can put these in a separate tests and skip/XFAIL if it runs too slow/doesn’t
work). Be sure to call your hint specifically in dsolve() (page 1262), that way the test won’t be
broken simply by the introduction of another matching hint. If your method works for higher
order (>1) ODEs, you will need to run sol = constant_renumber(sol, 'C', 1, order) for
each solution, where order is the order of the ODE. This is because constant_renumber
renumbers the arbitrary constants by printing order, which is platform dependent. Try to
test every corner case of your solver, including a range of orders if it is a nth order solver,
but if your solver is slow, such as if it involves hard integration, try to keep the test run time
down.
Feel free to refactor existing hints to avoid duplicating code or creating inconsistencies. If you
can show that your method exactly duplicates an existing method, including in the simplicity
and speed of obtaining the solutions, then you can remove the old, less general method. The
existing code is tested extensively in test_ode.py, so if anything is broken, one of those tests
will surely fail.
5.31 PDE
These are functions that are imported into the global namespace with from sympy import *.
They are intended for user use.
pde_separate()
Examples
pde_separate_add()
Examples
pde_separate_mul()
Examples
pdsolve()
Tips
• You can declare the derivative of an unknown function this way:
• See test_pde.py for many tests, which serves also as a set of examples for how
to use pdsolve().
• pdsolve always returns an Equality class (except for the case when the hint is
“all” or “all_Integral”). Note that it is not possible to get an explicit solution for
f(x, y) as in the case of ODE’s
• Do help(pde.pde_hintname) to get help more information on a specific hint
Examples
classify_pde()
Examples
checkpdesol()
Examples
These functions are meant for internal use. However they contain useful information on the
various solving methods.
pde_1st_linear_constant_coeff_homogeneous
df (x, y) df (x, y)
a +b + cf (x, y) = 0
dx dy
>>> pprint(pdsolve(genform))
-c*(a*x + b*y)
---------------
2 2
a + b
f(x, y) = F(-a*y + b*x)*e
References
• Viktor Grigoryan, “Partial Differential Equations” Math 124A - Fall 2010, pp.7
Examples
pde_1st_linear_constant_coeff
\|
(continues on next page)
References
• Viktor Grigoryan, “Partial Differential Equations” Math 124A - Fall 2010, pp.7
Examples
pde_1st_linear_variable_coeff
References
• Viktor Grigoryan, “Partial Differential Equations” Math 124A - Fall 2010, pp.7
Examples
This module contains pdsolve() and different helper functions that it uses. It is heavily inspired
by the ode module and hence the basic infrastructure remains the same.
Functions in this module
These are the user functions in this module:
• pdsolve() - Solves PDE’s
• classify_pde() - Classifies PDEs into possible hints for dsolve().
• pde_separate() - Separate variables in partial differential equation either by
additive or multiplicative separation approach.
These are the helper functions in this module:
• pde_separate_add() - Helper function for searching additive separable solu-
tions.
• pde_separate_mul() - Helper function for searching multiplicative
separable solutions.
Currently implemented solver methods
The following methods are implemented for solving partial differential equations. See the
docstrings of the various pde_hint() functions for more information on each (run help(pde)):
• 1st order linear homogeneous partial differential equations with constant coefficients.
• 1st order linear general partial differential equations with constant coefficients.
• 1st order linear partial differential equations with variable coefficients.
5.32 Solvers
Use solve() to solve algebraic equations. We suppose all equations are equaled to 0, so
solving x**2 == 1 translates into the following code:
The first argument for solve() is an equation (equaled to zero) and the second argument is
the symbol that we want to solve the equation for.
sympy.solvers.solvers.solve(f, *symbols, **flags)
Algebraically solves equations and systems of equations.
Currently supported are:
• polynomial,
• transcendental
• piecewise combinations of the above
• systems of linear and polynomial equations
• systems containing relational expressions.
Input is formed as:
• f
– a single Expr or Poly that must be zero,
– an Equality
– a Relational expression or boolean
– iterable of one or more of the above
• symbols (object(s) to solve for) specified as
Notes
solve() with check=True (default) will run through the symbol tags to elimate unwanted
solutions. If no assumptions are included all possible solutions will be returned.
>>> from sympy import Symbol, solve
>>> x = Symbol(”x”)
>>> solve(x**2 - 1)
[-1, 1]
If check=False then a solution to the numerator being zero is found: x = 0. In this case,
this is a spurious solution since sin(x)/x has the well known limit (without dicontinuity)
of 1 at x = 0:
>>> solve(sin(x)/x, check=False)
[0, pi]
In the following case, however, the limit exists and is equal to the value of x = 0 that is
excluded when check=True:
>>> eq = x**2*(1/x - z**2/x)
>>> solve(eq, x)
[]
>>> solve(eq, x, check=False)
[0]
>>> limit(eq, x, 0, '-')
0
>>> limit(eq, x, 0, '+')
0
Examples
The output varies according to the input and can be seen by example:
>>> from sympy import solve, Poly, Eq, Function, exp
>>> from sympy.abc import x, y, z, a, b
>>> f = Function('f')
>>> solve(x - y, x)
[y]
>>> solve(x - 3, x)
[3]
>>> solve(Eq(x, 3), x)
[3]
>>> solve(Poly(x - 3), x)
[3]
>>> solve(x**2 - y**2, x, set=True)
([x], {(-y,), (y,)})
>>> solve(x**4 - 1, x, set=True)
([x], {(-1,), (1,), (-I,), (I,)})
>>> solve(3, x)
[]
>>> solve(x - 3, y)
[]
>>> solve(x - 3)
[3]
>>> solve(x**2 - y**2)
[{x: -y}, {x: y}]
>>> solve(z**2*x**2 - z**2*y**2)
[{x: -y}, {x: y}, {z: 0}]
(continues on next page)
– Nothing heroic is done in this implicit solving so you may end up with a
symbol still in the solution:
>>> eqs = (x*y + 3*y + sqrt(3), x + 4 + y)
>>> solve(eqs, y, x + 2)
{y: -sqrt(3)/(x + 3), x + 2: (-2*x - 6 + sqrt(3))/(x + 3)}
>>> solve(eqs, y*x, x)
{x: -y - 4, x*y: -3*y - sqrt(3)}
– if you attempt to solve for a number remember that the number you have
obtained does not necessarily mean that the value is equivalent to the
expression obtained:
>>> solve(sqrt(2) - 1, 1)
[sqrt(2)]
>>> solve(x - y + 1, 1) # /!\ -1 is targeted, too
[x/(y - 1)]
(continues on next page)
– if there is no linear solution then the first successful attempt for a nonlinear
solution will be returned
∗ without a solution
– if no symbols are given, all free symbols will be selected and a list of mappings
returned
– if any equation doesn’t depend on the symbol(s) given it will be eliminated from
the equation set and an answer may be given implicitly in terms of variables that
were not of interest
When solving polynomial expressions, one might not want explicit solutions (which can
be quite long). If the expression is univariate, CRootOf instances will be returned in-
stead:
>>> solve(x**3 - x + 1)
[-1/((-1/2 - sqrt(3)*I/2)*(3*sqrt(69)/2 + 27/2)**(1/3)) - (-1/2 -
sqrt(3)*I/2)*(3*sqrt(69)/2 + 27/2)**(1/3)/3, -(-1/2 +
sqrt(3)*I/2)*(3*sqrt(69)/2 + 27/2)**(1/3)/3 - 1/((-1/2 +
sqrt(3)*I/2)*(3*sqrt(69)/2 + 27/2)**(1/3)), -(3*sqrt(69)/2 +
27/2)**(1/3)/3 - 1/(3*sqrt(69)/2 + 27/2)**(1/3)]
>>> solve(x**3 - x + 1, cubics=False)
[CRootOf(x**3 - x + 1, 0),
CRootOf(x**3 - x + 1, 1),
CRootOf(x**3 - x + 1, 2)]
Sometimes solutions will be obtained even when a flag is False because the expression
could be factored. In the following example, the equation can be factored as the product
of a linear and a quadratic factor so explicit solutions (which did not require solving a
cubic expression) are obtained:
Because of SymPy’s use of the principle root (issue #8789), some solutions to radical
equations will be missed unless check=False:
In the above example there is only a single solution to the equation. Other expressions
will yield spurious roots which must be checked manually; roots which give a negative
argument to odd-powered radicals will also need special checking:
The first solution is negative so real_root must be used to see that it satisfies
the expression:
If the roots of the equation are not real then more care will be necessary to find the roots,
especially for higher order equations. Consider the following expression:
We will construct a known value for this expression at x = 3 by selecting the 1-th root
for each radical:
The solve function is unable to find any exact roots to this equation:
The function unrad, however, can be used to get a form of the equation for which nu-
merical roots can be found:
Although eq or eq1 could have been used to find xvals, the solution can only be verified
with expr1:
>>> z = expr - v
>>> [xi.n(chop=1e-9) for xi in xvals if abs(z.subs(x, xi).n()) < 1e-9]
[]
>>> z1 = expr1 - v
>>> [xi.n(chop=1e-9) for xi in xvals if abs(z1.subs(x, xi).n()) < 1e-9]
[-3.0]
(0, 0) meaning that there is no solution to the equation amongst the symbols
given.
(If the first element of the tuple is not zero then the function is guaran-
teed to be dependent on a symbol in symbols.)
(symbol, solution) where symbol appears linearly in the numerator of f, is in
symbols (if given) and is not in exclude (if given). No simplification is done to
f other than a mul=True expansion, so the solution will correspond strictly to a
unique solution.
(n, d) where n and d are the numerator and denominator of f when the nu-
merator was not linear in any symbol of interest; n will never be a symbol unless
a solution for that symbol was found (in which case the second element is the
solution, not the denominator).
Examples
When not linear in x or y then the numerator and denominator are returned.
>>> solve_linear(x**2/y**2 - 3)
(x**2 - 3*y**2, y**2)
If the numerator of the expression is a symbol then (0, 0) is returned if the solution for
that symbol would have set any denominator to 0:
>>> eq = 1/(1/x - 2)
>>> eq.as_numer_denom()
(x, -2*x + 1)
>>> solve_linear(eq)
(0, 0)
But automatic rewriting may cause a symbol in the denominator to appear in the numer-
ator so a solution will be returned:
>>> (1/x)**-1
x
>>> solve_linear((1/x)**-1)
(x, 0)
(A solution for y is obtained because it is the first variable from the canonically sorted
list of symbols that had a linear solution.)
sympy.solvers.solvers.solve_linear_system(system, *symbols, **flags)
Solve system of N linear equations with M variables, which means both under- and
overdetermined systems are supported. The possible number of solutions is zero, one
or infinite. Respectively, this procedure will return None or a dictionary with solutions.
In the case of underdetermined systems, all arbitrary parameters are skipped. This may
cause a situation in which an empty dictionary is returned. In that case, all symbols can
be assigned arbitrary values.
Input to this functions is a Nx(M+1) matrix, which means it has to be in augmented
form. If you prefer to enter N equations and M unknowns then use solve(N eqs, ∗M symbols)
instead. Note: a local copy of the matrix is made by this routine so the matrix that is
passed will not be modified.
The algorithm used here is fraction-free Gaussian elimination, which results, after elimi-
nation, in an upper-triangular matrix. Then solutions are found using back-substitution.
This approach is more efficient and compact than the Gauss-Jordan method.
x + 4 y == 2
-2 x + y == 14
sympy.solvers.solvers.solve_linear_system_LU(matrix, syms)
Solves the augmented matrix system using LUsolve and returns a dictionary in which
solutions are keyed to the symbols of syms as ordered.
The matrix must be invertible.
See also:
sympy.matrices.LUsolve
Examples
>>> solve_linear_system_LU(Matrix([
... [1, 2, 0, 1],
... [3, 2, 2, 1],
... [2, 0, 0, 1]]), [x, y, z])
{x: 1/2, y: 1/4, z: -1/2}
sympy.solvers.solvers.nsolve(*args, **kwargs)
Solve a nonlinear equation system numerically:
f is a vector function of symbolic expressions representing the system. args are the
variables. If there is only one variable, this argument can be omitted. x0 is a starting
vector close to a solution.
Use the modules keyword to specify which modules should be used to evaluate the func-
tion and the Jacobian matrix. Make sure to use a module that supports matrices. For
more information on the syntax, please see the docstring of lambdify.
If the keyword arguments contain ‘dict’=True (default is False) nsolve will return a list
(perhaps empty) of solution mappings. This might be especially useful if you want to use
nsolve as a fallback to solve since using the dict argument for both methods produces
return values of consistent type structure. Please note: to keep this consistency with
solve, the solution will be returned in a list even though nsolve (currently at least) only
finds one solution at a time.
Overdetermined systems are supported.
To solve with higher precision than the default, use the prec argument.
>>> from sympy import cos
>>> nsolve(cos(x) - x, 1)
0.739085133215161
>>> nsolve(cos(x) - x, 1, prec=50)
0.73908513321516064165531208767387340401341175890076
>>> cos(_)
0.73908513321516064165531208767387340401341175890076
To solve for complex roots of real functions, a nonreal initial point must be specified:
>>> from sympy import I
>>> nsolve(x**2 + 2, I)
1.4142135623731*I
mpmath.findroot is used and you can find there more extensive documentation, espe-
cially concerning keyword parameters and available solvers. Note, however, that func-
tions which are very steep near the root the verification of the solution may fail. In this
case you should use the flag verif y = F alse and independently verify the solution.
>>> from sympy import cos, cosh
>>> from sympy.abc import i
>>> f = cos(x)*cosh(x) - 1
>>> nsolve(f, 3.14*100)
Traceback (most recent call last):
...
ValueError: Could not find root within given tolerance. (1.39267e+230 > 2.1684e-
,→19)
One might safely skip the verification if bounds of the root are known and a bisection
method is used:
>>> bounds = lambda i: (3.14*i, 3.14*(i + 1))
>>> nsolve(f, bounds(100), solver='bisect', verify=False)
315.730061685774
Alternatively, a function may be better behaved when the denominator is ignored. Since
this is not always the case, however, the decision of what function to use is left to the
discretion of the user.
>>> eq = x**2/(1 - x)/(1 - 2*x)**2 - 100
>>> nsolve(eq, 0.46)
Traceback (most recent call last):
...
ValueError: Could not find root within given tolerance. (10000 > 2.1684e-19)
(continues on next page)
Examples
To check assumptions of expr against another variable or expression, pass the expres-
sion or variable as against.
>>> check_assumptions(2*x + 1, x)
True
Examples
To check if an expression is zero using checksol, pass it as f and send an empty dictionary
for symbol:
sympy.solvers.deutils.ode_order(expr, func)
Returns the order of a given differential equation with respect to func.
This function is implemented recursively.
Examples
sympy.solvers.recurr.rsolve(f, y, init=None)
Solve univariate recurrence with rational coefficients.
Given k-th order linear recurrence L y = f , or equivalently:
Examples
sympy.solvers.recurr.rsolve_poly(coeffs, f, n, **hints)
Given linear recurrence operator L of order k with polynomial coefficients and inhomo-
geneous equation L y = f , where f is a polynomial, we seek for all polynomial solutions
over field K of characteristic zero.
The algorithm performs two basic steps:
(1) Compute degree N of the general polynomial solution.
(2) Find all polynomials of degree N or less of L y = f .
There are two methods for computing the polynomial solutions. If the degree bound
is relatively small, i.e. it’s smaller than or equal to the order of the recurrence, then
naive method of undetermined coefficients is being used. This gives system of algebraic
equations with N + 1 unknowns.
In the other case, the algorithm performs transformation of the initial equation to an
equivalent one, for which the system of algebraic equations has only r indeterminates.
This method is quite sophisticated (in comparison with the naive one) and was invented
together by Abramov, Bronstein and Petkovsek.
It is possible to generalize the algorithm implemented here to the case of linear q-
difference and differential equations.
Lets say that we would like to compute m-th Bernoulli polynomial up to a constant. For
this we can use b(n + 1) − b(n) = mnm−1 recurrence, which has solution b(n) = Bm + C. For
example:
References
References
[R535]
Examples
sympy.solvers.recurr.rsolve_hyper(coeffs, f, n, **hints)
Given linear recurrence operator L of order k with polynomial coefficients and inho-
mogeneous equation L y = f we seek for all hypergeometric solutions over field K of
characteristic zero.
The inhomogeneous part can be either hypergeometric or a sum of a fixed number of
pairwise dissimilar hypergeometric terms.
The algorithm performs three basic steps:
(1) Group together similar hypergeometric terms in the inhomogeneous part of L y = f ,
and find particular solution using Abramov’s algorithm.
(2) Compute generating set of L and find basis in it, so that all solutions are linearly
independent.
(3) Form final solution with the number of arbitrary constants equal to dimension of
basis of L.
Term a(n) is hypergeometric if it is annihilated by first order linear difference equations
with polynomial coefficients or, in simpler words, if consecutive term ratio is a rational
function.
The output of this procedure is a linear combination of fixed number of hypergeomet-
ric terms. However the underlying method can generate larger class of solutions -
D’Alembertian terms.
Note also that this method not only computes the kernel of the inhomogeneous equation,
but also reduces in to a basis so that solutions generated by this procedure are linearly
independent
References
[R536], [R537]
Examples
Examples
References
Examples
>>> solve_triangulated(F, x, y, z)
[(0, 0, 1), (0, 1, 0), (1, 0, 0)]
5.32.8 Inequalities
5.33 Diophantine
The word “Diophantine” comes with the name Diophantus, a mathematician lived in the great
city of Alexandria sometime around 250 AD. Often referred to as the “father of Algebra”, Dio-
phantus in his famous work “Arithmetica” presented 150 problems that marked the early
beginnings of number theory, the field of study about integers and their properties. Diophan-
tine equations play a central and an important part in number theory.
We call a “Diophantine equation” to an equation of the form, f (x1 , x2 , . . . xn ) = 0 where n ≥ 2 and
x1 , x2 , . . . xn are integer variables. If we can find n integers a1 , a2 , . . . an such that x1 = a1 , x2 =
a2 , . . . xn = an satisfies the above equation, we say that the equation is solvable. You can read
more about Diophantine equations in1 and2 .
Currently, following five types of Diophantine equations can be solved using diophantine()
(page 1352) and other helper functions of the Diophantine module.
• Linear Diophantine equations: a1 x1 + a2 x2 + . . . + an xn = b.
• General binary quadratic equation: ax2 + bxy + cy 2 + dx + ey + f = 0
• Homogeneous ternary quadratic equation: ax2 + by 2 + cz 2 + dxy + eyz + f zx = 0
• Extended Pythagorean equation: a1 x21 + a2 x22 + . . . + an x2n = an+1 x2n+1
• General sum of squares: x21 + x22 + . . . + x2n = k
This module contains diophantine() (page 1352) and helper functions that are needed to
solve certain Diophantine equations. It’s structured in the following manner.
• diophantine() (page 1352)
– diop_solve() (page 1354)
∗ classify_diop() (page 1353)
∗ diop_linear() (page 1354)
∗ diop_quadratic() (page 1356)
∗ diop_ternary_quadratic() (page 1360)
1 Andreescu, Titu. Andrica, Dorin. Cucurezeanu, Ion. An Introduction to Diophantine Equations
2 Diophantine Equation, Wolfram Mathworld, [online]. Available: http://mathworld.wolfram.com/
DiophantineEquation.html
5.33.3 Tutorial
Let’s start by solving the easiest type of Diophantine equations, i.e. linear Diophantine equa-
tions. Let’s solve 2x + 3y = 5. Note that although we write the equation in the above form,
when we input the equation to any of the functions in Diophantine module, it needs to be in
the form eq = 0.
Note that stepping one more level below the highest API, we can solve the very same equation
by calling diop_solve() (page 1354).
Note that it returns a tuple rather than a set. diophantine() (page 1352) always return a
set of tuples. But diop_solve() (page 1354) may return a single tuple or a set of tuples
depending on the type of the equation given.
We can also solve this equation by calling diop_linear() (page 1354), which is what
diop_solve() (page 1354) calls internally.
>>> from sympy.solvers.diophantine import diop_linear
>>> diop_linear(2*x + 3*y - 5)
(3*t_0 - 5, -2*t_0 + 5)
If the given equation has no solutions then the outputs will look like below.
>>> diophantine(2*x + 4*y - 3)
set()
>>> diop_solve(2*x + 4*y - 3)
(None, None)
>>> diop_linear(2*x + 4*y - 3)
(None, None)
Note that except for the highest level API, in case of no solutions, a tuple of N one are re-
turned. Size of the tuple is the same as the number of variables. Also, one can specifically
set the parameter to be used in the solutions by passing a customized parameter. Consider
the following example:
>>> m = symbols(”m”, integer=True)
>>> diop_solve(2*x + 3*y - 5, m)
(3*m_0 - 5, -2*m_0 + 5)
For linear Diophantine equations, the customized parameter is the prefix used for each free
variable in the solution. Consider the following example:
>>> diop_solve(2*x + 3*y - 5*z + 7, m)
(m_0, m_0 + 5*m_1 - 14, m_0 + 3*m_1 - 7)
In the solution above, m_0 and m_1 are independent free variables.
Please note that for the moment, users can set the parameter only for linear Diophantine
equations and binary quadratic equations.
Let’s try solving a binary quadratic equation which is an equation with two variables and
has a degree of two. Before trying to solve these equations, an idea about various cases
associated with the equation would help a lot. Please refer3 and4 for detailed analysis of
different cases and the nature of the solutions. Let us define ∆ = b2 − 4ac w.r.t. the binary
quadratic ax2 + bxy + cy 2 + dx + ey + f = 0.
When ∆ < 0, there are either no solutions or only a finite number of solutions.
>>> diophantine(x**2 - 4*x*y + 8*y**2 - 3*x + 7*y - 5)
{(2, 1), (5, 1)}
In the above equation ∆ = (−4)2 − 4 ∗ 1 ∗ 8 = −16 and hence only a finite number of solutions
exist.
When ∆ = 0 we might have either no solutions or parameterized solutions.
>>> diophantine(3*x**2 - 6*x*y + 3*y**2 - 3*x + 7*y - 5)
set()
>>> diophantine(x**2 - 4*x*y + 4*y**2 - 3*x + 7*y - 5)
(continues on next page)
3 Methods to solve Ax^2 + Bxy + Cy^2 + Dx + Ey + F = 0,[online], Available: http://www.alpertron.com.ar/
METHODS.HTM
4
Solving the equation ax^2+ bxy + cy^2 + dx + ey + f= 0, [online], Available: http://www.jpr2718.org/ax2p.pdf
The most interesting case is when ∆ > 0 and it is not a perfect square. In this case, the
equation has either no solutions or an infinite number of solutions. Consider the below cases
where ∆ = 8.
>>> diophantine(x**2 - 4*x*y + 2*y**2 - 3*x + 7*y - 5)
set()
>>> from sympy import sqrt
>>> n = symbols(”n”, integer=True)
>>> s = diophantine(x**2 - 2*y**2 - 2*x - 4*y, n)
>>> x_1, y_1 = s.pop()
>>> x_2, y_2 = s.pop()
>>> x_n = -(-2*sqrt(2) + 3)**n/2 + sqrt(2)*(-2*sqrt(2) + 3)**n/2 - sqrt(2)*(2*sqrt(2)␣
,→+ 3)**n/2 - (2*sqrt(2) + 3)**n/2 + 1
Here n is an integer. Although x_n and y_n may not look like integers, substituting in specific
values for n (and simplifying) shows that they are. For example consider the following example
where we set n equal to 9.
>>> from sympy import simplify
>>> simplify(x_n.subs({n: 9}))
-9369318
So, the above equation is equivalent to the equation X 2 − 5Y 2 = 920 after a linear transfor-
mation. If we want to find the linear transformation, we can use transformation_to_DN()
(page 1358)
>>> A, B = transformation_to_DN(x**2 - 3*x*y + y**2 - 7*x + 5*y - 3)
If you are only interested in a base solution rather than the parameterized general solution
(to be more precise, one of the general solutions), you can use diop_ternary_quadratic()
(page 1360).
function diop_general_pythagorean() (page 1362) can also be called directly to solve the
same equation. Either you can call diop_general_pythagorean() (page 1362) or use the
high level API. For the general sum of squares, this is also true, but one advantage of calling
diop_general_sum_of_squares() (page 1362) is that you can control how many solutions
are returned.
The sum_of_squares() (page 1366) routine will providean iterator that returns solutions and
one may control whether the solutions contain zeros or not (and the solutions not containing
zeros are returned first):
Simple Eqyptian fractions can be found with the Diophantine module, too. For example, here
are the ways that one might represent 1/2 as a sum of two unit fractions:
To get a more thorough understanding of the Diophantine module, please refer to the follow-
ing blog.
http://thilinaatsympy.wordpress.com/
5.33.4 References
This functions is imported into the global namespace with from sympy import *:
diophantine()
Examples
Usage
diophantine(eq, t, syms): Solve the diophantine equation eq. t is the optional param-
eter to be used by diop_solve(). syms is an optional list of symbols which determines
the order of the elements in the returned tuple.
By default, only the base solution is returned. If permute is set to True then permutations
of the base solution and/or permutations of the signs of the values will be returned when
applicable.
Details
classify_diop()
sympy.solvers.diophantine.classify_diop(eq, _dict=True)
These functions are intended for internal use in the Diophantine module.
diop_solve()
sympy.solvers.diophantine.diop_solve(eq, param=t)
Solves the diophantine equation eq.
Unlike diophantine(), factoring of eq is not attempted. Uses classify_diop() to de-
termine the type of the equation and calls the appropriate solver function.
See also:
diophantine (page 1352)
Examples
Usage
Details
diop_linear()
sympy.solvers.diophantine.diop_linear(eq, param=t)
Solves linear diophantine equations.
A linear diophantine equation is an equation of the form a1 x1 + a2 x2 + .. + an xn = 0 where
a1 , a2 , ..an are integer constants and x1 , x2 , ..xn are integer variables.
See also:
diop_quadratic (page 1356), diop_ternary_quadratic (page 1360),
diop_general_pythagorean (page 1362), diop_general_sum_of_squares (page 1362)
Examples
Usage
Details
base_solution_linear()
sympy.solvers.diophantine.base_solution_linear(c, a, b, t=None)
Return the base solution for the linear equation, ax + by = c.
Used by diop_linear() to find the base solution of a linear Diophantine equation. If t
is given then the parametrized solution is returned.
Examples
Usage
diop_quadratic()
sympy.solvers.diophantine.diop_quadratic(eq, param=t)
Solves quadratic diophantine equations.
i.e. equations of the form Ax2 + Bxy + Cy 2 + Dx + Ey + F = 0. Returns a set containing
the tuples (x, y) which contains the solutions. If there are no solutions then (N one, N one)
is returned.
See also:
diop_linear (page 1354), diop_ternary_quadratic (page 1360),
diop_general_sum_of_squares (page 1362), diop_general_pythagorean (page 1362)
References
[R509], [R510]
Examples
Usage
Details
diop_DN()
sympy.solvers.diophantine.diop_DN(D, N, t=t)
Solves the equation x2 − Dy 2 = N .
Mainly concerned with the case D > 0, D is not a perfect square, which is the same as
the generalized Pell equation. The LMM algorithm [R511] is used to solve this equation.
Returns one solution tuple, (x, y) for each class of the solutions. Other solutions of the
class can be constructed according to the values of D and N.
See also:
find_DN (page 1360), diop_bf_DN (page 1358)
References
[R511]
Examples
The output can be interpreted as follows: There are three fundamental solutions to the
equation x2 − 13y 2 = −4 given by (3, 1), (393, 109) and (36, 10). Each tuple is in the form
(x, y), i.e. solution (3, 1) means that x = 3 and y = 1.
Usage
Details
cornacchia()
sympy.solvers.diophantine.cornacchia(a, b, m)
Solves ax2 + by 2 = m where gcd(a, b) = 1 = gcd(a, m) and a, b > 0.
Uses the algorithm due to Cornacchia. The method only finds primitive solutions, i.e.
ones with gcd(x, y) = 1. So this method can’t be used to find the solutions of x2 + y 2 = 20
since the only solution to former is (x, y) = (4, 2) and it is not primitive. When a = b, only
the solutions with x ≤ y are found. For more details, see the References.
See also:
sympy.utilities.iterables.signed_permutations (page 1475)
References
[R512], [R513]
Examples
diop_bf_DN()
sympy.solvers.diophantine.diop_bf_DN(D, N, t=t)
Uses brute force to solve the equation, x2 − Dy 2 = N .
Mainly concerned with the generalized Pell equation which is the case when D > 0, D
is not a perfect square. For more information on the case refer [R514]. Let (t, u) be
the minimal positive solution of the equation x2 − Dy 2 = 1. Then this method requires
√
f rac| N | (t ± 1)2D to be small.
See also:
diop_DN (page 1356)
References
[R514]
Examples
Usage
Details
transformation_to_DN()
sympy.solvers.diophantine.transformation_to_DN(eq)
This function transforms general quadratic, ax2 + bxy + cy 2 + dx + ey + f = 0 to more easy
to deal with X 2 − DY 2 = N form.
This is used to solve the general quadratic equation by transforming it to the latter form.
Refer [R515] for more detailed information on the transformation. This function returns
a tuple (A, B) where A is a 2 X 2 matrix and B is a 2 X 1 matrix such that,
Transpose([x y]) = A * Transpose([X Y]) + B
See also:
find_DN (page 1360)
References
[R515]
Examples
By multiplying the denominator appropriately, we can get a Pell equation in the standard
form.
>>> eq * 676
X**2 - 13*Y**2 + 884
Usage
find_DN()
sympy.solvers.diophantine.find_DN(eq)
This function returns a tuple, (D, N ) of the simplified form, x2 − Dy 2 = N , corresponding
to the general quadratic, ax2 + bxy + cy 2 + dx + ey + f = 0.
Solving the general quadratic is then equivalent to solving the equation X 2 − DY 2 = N
and transforming the solutions by using the transformation matrices returned by trans-
formation_to_DN().
See also:
transformation_to_DN (page 1358)
References
[R516]
Examples
Interpretation of the output is that we get X 2 − 13Y 2 = −884 after transforming x2 − 3xy −
y 2 − 2y + 1 using the transformation returned by transformation_to_DN().
Usage
diop_ternary_quadratic()
sympy.solvers.diophantine.diop_ternary_quadratic(eq)
Solves the general quadratic ternary form, ax2 + by 2 + cz 2 + f xy + gyz + hxz = 0.
Returns a tuple (x, y, z) which is a base solution for the above equation. If there are no
solutions, (N one, N one, N one) is returned.
Examples
Usage
Details
square_factor()
sympy.solvers.diophantine.square_factor(a)
Returns an integer c s.t. a = c2 k, c, k ∈ Z. Here k is square free. a can be given as an
integer or a dictionary of factors.
See also:
sympy.ntheory.factor_.core (page 327)
Examples
descent()
sympy.solvers.diophantine.descent(A, B)
Returns a non-trivial solution, (x, y, z), to x2 = Ay 2 +Bz 2 using Lagrange’s descent method
with lattice-reduction. A and B are assumed to be valid for such a solution to exist.
This is faster than the normal Lagrange’s descent algorithm because the Gaussian re-
duction is used.
References
[R517]
Examples
diop_general_pythagorean()
sympy.solvers.diophantine.diop_general_pythagorean(eq, param=m)
Solves the general pythagorean equation, a21 x21 + a22 x22 + ... + a2n x2n − a2n+1 x2n+1 = 0.
Returns a tuple which contains a parametrized solution to the equation, sorted in the
same order as the input variables.
Examples
Usage
diop_general_sum_of_squares()
sympy.solvers.diophantine.diop_general_sum_of_squares(eq, limit=1)
Solves the equation x21 + x22 + ... + x2n − k = 0.
Returns at most limit number of solutions.
Examples
Usage
Details
Reference
diop_general_sum_of_even_powers()
sympy.solvers.diophantine.diop_general_sum_of_even_powers(eq, limit=1)
Solves the equation xe1 + xe2 + ... + xen − k = 0 where e is an even, integer power.
Returns at most limit number of solutions.
See also:
power_representation
Examples
Usage
partition()
possible partitions are returned irrespective of their size, otherwise only the partitions
of size k are returned. If the zero parameter is set to True then a suitable number of
zeros are added at the end of every partition of size less than k.
zero parameter is considered only if k is not None. When the partitions are over, the
last next() call throws the StopIteration exception, so this function should always be
used inside a try - except block.
Examples
Details
partition(n, k): Here n is a positive integer and k is the size of the partition which is
also positive integer.
sum_of_three_squares()
sympy.solvers.diophantine.sum_of_three_squares(n)
Returns a 3-tuple (a, b, c) such that a2 + b2 + c2 = n and a, b, c ≥ 0.
Returns None if n = 4a (8m + 7) for some a, m ∈ Z. See [R519] for more details.
See also:
sum_of_squares (page 1366)
References
[R519]
Examples
Usage
sum_of_four_squares()
sympy.solvers.diophantine.sum_of_four_squares(n)
Returns a 4-tuple (a, b, c, d) such that a2 + b2 + c2 + d2 = n.
Here a, b, c, d ≥ 0.
See also:
sum_of_squares (page 1366)
References
[R520]
Examples
Usage
sum_of_powers()
sympy.solvers.diophantine.sum_of_powers(n, p, k, zeros=False)
Returns a generator for finding k-tuples of integers, (n1 , n2 , ...nk ), such that n = np1 + np2 +
...npk .
Examples
>>> f = power_representation(1729, 3, 2)
>>> next(f)
(9, 10)
>>> next(f)
(1, 12)
If the flag zeros is True, the solution may contain tuples with zeros; any such solutions
will be generated after the solutions without zeros:
For even p the permutes ign function can be used to get all signed values:
Usage
sum_of_squares()
sympy.solvers.diophantine.sum_of_squares(n, k, zeros=False)
Return a generator that yields the k-tuples of nonnegative values, the squares of which
sum to n. If zeros is False (default) then the solution will not contain zeros. The nonneg-
ative elements of a tuple are sorted.
• If k == 1 and n is square, (n,) is returned.
• If k == 2 then n can only be written as a sum of squares if every prime in the
factorization of n that has the form 4*k + 3 has an even multiplicity. If n is prime
then it can only be written as a sum of two squares if it is in the form 4*k + 1.
• if k == 3 then n can be written as a sum of squares if it does not have the form
4**m*(8*k + 7).
• all integers can be written as the sum of 4 squares.
• if k > 4 then n can be partitioned and each partition can be written as a sum of 4
squares; if n is not evenly divisible by 4 then n can be written as a sum of squares
only if the an additional partition can be written as sum of squares. For example,
if k = 6 then n is partitioned into two parts, the first being written as a sum of 4
squares and the second being written as a sum of 2 squares – which can only be
done if the condition above for k = 2 can be met, so this will automatically reject
certain partitions of n.
See also:
sympy.utilities.iterables.signed_permutations (page 1475)
Examples
merge_solution
divisible
sympy.solvers.diophantine.divisible(a, b)
Returns T rue if a is divisible by b and F alse otherwise.
PQa
sympy.solvers.diophantine.PQa(P_0, Q_0, D)
Returns useful information needed to solve the Pell equation.
There are six sequences of integers defined related to the continued fraction represen-
tation of√
f racP + DQ, namely {Pi }, {Qi }, {ai },{Ai }, {Bi }, {Gi }. PQa() Returns these values as
a 6-tuple in the same order as mentioned above. Refer [R521] for more detailed infor-
mation.
References
[R521]
Examples
Usage
PQa(P_0, Q_0, D): P_0, Q_0 and D are integers corresponding to P0 , Q0 and D in the
continued√ fraction
f racP0 + DQ0 . Also it’s assumed that P02 == Dmod(|Q0 |) and D is square free.
equivalent
sympy.solvers.diophantine.equivalent(u, v, r, s, D, N)
Returns True if two solutions (u, v) and (r, s) of x2 − Dy 2 = N belongs to the same equiva-
lence class and False otherwise.
Two solutions (u, v) and (r, s) to the above equation fall to the same equivalence class iff
both (ur−Dvs) and (us−vr) are divisible by N . See reference [R522]. No test is performed
to test whether (u, v) and (r, s) are actually solutions to the equation. User should take
care of this.
References
[R522]
Examples
Usage
equivalent(u, v, r, s, D, N): (u, v) and (r, s) are two solutions of the equation x2 −
Dy 2 = N and all parameters involved are integers.
parametrize_ternary_quadratic
sympy.solvers.diophantine.parametrize_ternary_quadratic(eq)
Returns the parametrized general solution for the ternary quadratic equation eq which
has the form ax2 + by 2 + cz 2 + f xy + gyz + hxz = 0.
References
[R523]
Examples
diop_ternary_quadratic_normal
sympy.solvers.diophantine.diop_ternary_quadratic_normal(eq)
Solves the quadratic ternary diophantine equation, ax2 + by 2 + cz 2 = 0.
Here the coefficients a, b, and c should be non zero. Otherwise the equation will be a
quadratic binary or univariate equation. If solvable, returns a tuple (x, y, z) that satisfies
the given equation. If the equation does not have integer solutions, (N one, N one, N one) is
returned.
Examples
Usage
ldescent
sympy.solvers.diophantine.ldescent(A, B)
Return a non-trivial solution to w2 = Ax2 + By 2 using Lagrange’s method; return None if
there is no such solution. .
Here, A ̸= 0 and B ̸= 0 and A and B are square free. Output a tuple (w0 , x0 , y0 ) which is a
solution to the above equation.
References
[R524], [R525]
Examples
gaussian_reduce
sympy.solvers.diophantine.gaussian_reduce(w, a, b)
Returns a reduced solution (x, z) to the congruence X 2 − aZ 2 ≡ 0 (mod b) so that x2 + |a|z 2
is minimal.
References
[R526], [R527]
Details
holzer
sympy.solvers.diophantine.holzer(x, y, z, a, b, c)
Simplify the solution (x, y, z) of the equation ax2 + by 2 = cz 2 with a, b, c > 0 and z 2 ≥| ab | to
a new reduced solution (x′ , y ′ , z ′ ) such that z ′2 ≤| ab |.
The algorithm is an interpretation of Mordell’s reduction as described on page 8 of Cre-
mona and Rusin’s paper [R528] and the work of Mordell in reference [R529].
References
[R528], [R529]
prime_as_sum_of_two_squares
sympy.solvers.diophantine.prime_as_sum_of_two_squares(p)
Represent a prime p as a unique sum of two squares; this can only be done if the prime
is congruent to 1 mod 4.
See also:
sum_of_squares (page 1366)
Examples
Reference
sqf_normal
sympy.solvers.diophantine.sqf_normal(a, b, c, steps=False)
Return a′ , b′ , c′ , the coefficients of the square-free normal form of ax2 + by 2 + cz 2 = 0, where
a′ , b′ , c′ are pairwise prime. If steps is True then also return three tuples: sq, sqf , and
(a′ , b′ , c′ ) where sq contains the square factors of a, b and c after removing the gcd(a, b, c);
sqf contains the values of a, b and c after removing both the gcd(a, b, c) and the square
factors.
The solutions for ax2 +by 2 +cz 2 = 0 can be recovered from the solutions of a′ x2 +b′ y 2 +c′ z 2 =
0.
See also:
reconstruct (page 1372)
References
[R531]
Examples
reconstruct
sympy.solvers.diophantine.reconstruct(A, B, z)
Reconstruct the z value of an equivalent solution of ax2 + by 2 + cz 2 from the z value of a
solution of the square-free normal form of the equation, a′ ∗ x2 + b′ ∗ y 2 + c′ ∗ z 2 , where a′ ,
b′ and c′ are square free and gcd(a′ , b′ , c′ ) == 1.
sympy.solvers.inequalities.solve_rational_inequalities(eqs)
Solve a system of rational inequalities with rational coefficients.
See also:
solve_poly_inequality (page 1372)
Examples
>>> solve_rational_inequalities([[
... ((Poly(-x + 1), Poly(1, x)), '>='),
... ((Poly(-x + 1), Poly(1, x)), '<=')]])
{1}
>>> solve_rational_inequalities([[
... ((Poly(x), Poly(1, x)), '!='),
... ((Poly(-x + 1), Poly(1, x)), '>=')]])
Union(Interval.open(-oo, 0), Interval.Lopen(0, 1))
sympy.solvers.inequalities.solve_poly_inequality(poly, rel)
Solve a polynomial inequality with rational coefficients.
See also:
solve_poly_inequalities
Examples
Examples
Examples
sympy.solvers.inequalities.reduce_abs_inequalities(exprs, gen)
Reduce a system of inequalities with nested absolute values.
See also:
reduce_abs_inequality (page 1373)
Examples
sympy.solvers.inequalities.reduce_inequalities(inequalities, symbols=[])
Reduce a system of inequalities with rational coefficients.
Examples
Notes
Currently, we cannot solve all the inequalities due to limitations in solvif y. Also, the
solution returned for trigonometric inequalities are restricted in its periodic interval.
Examples
5.35 Solveset
This is the official documentation of the solveset module in solvers. It contains the frequently
asked questions about our new module to solve equations.
SymPy already has a pretty powerful solve function. But it has a lot of major issues
1. It doesn’t have a consistent output for various types of solutions It needs to return a lot
of types of solutions consistently:
• Single solution : x = 1
• Multiple solutions: x2 = 1
• No Solution: x2 + 1 = 0; x ∈ R
• Interval of solution: ⌊x⌋ = 0
• solveset has a cleaner input and output interface: solveset returns a set object and
a set object takes care of all types of output. For cases where it doesn’t “know” all the
solutions a ConditionSet with a partial solution is returned. For input it only takes the
equation, the variables to solve for and the optional argument domain over which the
equation is to be solved.
• solveset can return infinitely many solutions. For example solving for sin (x) = 0 returns
{2nπ|n ∈ Z} ∪ {2nπ + π|n ∈ Z}, whereas solve only returns [0, π].
• There is a clear code level and interface level separation between solvers for equations
in the complex domain and the real domain. For example solving ex = 1 when x is to
be solved in the complex domain, returns the set of all solutions, that is {2niπ|n ∈ Z},
whereas if x is to be solved in the real domain then only {0} is returned.
SymPy has a well developed sets module, which can represent most of the set containers in
Mathematics such as:
• FiniteSet
Represents a finite set of discrete numbers.
• Interval
Represents a real interval as a set.
• ProductSet
Represents a Cartesian product of sets.
• ImageSet
Represents the image of a set under a mathematical function
• ComplexRegion
Represents the set of all complex numbers in a region in the Argand plane.
• ConditionSet
Represents the set of elements, which satisfies a given condition.
Also, the predefined set classes such as:
• Naturals N
Represents the natural numbers (or counting numbers), which are all positive
integers starting from 1.
• Naturals0 N⊬
Represents the whole numbers, which are all the non-negative integers, inclu-
sive of 0.
• Integers Z
Represents all integers: positive, negative and zero.
• Reals R
Represents the set of all real numbers.
• Complexes C
Represents the set of all complex numbers.
• EmptySet ϕ
Represents the empty set.
The above six sets are available as Singletons, like S.Integers.
It is capable of most of the set operations in mathematics:
• Union
• Intersection
• Complement
• SymmetricDifference
The main reason for using sets as output to solvers is that it can consistently represent many
types of solutions. For the single variable case it can represent:
• No solution (by the empty set).
• Finitely many solutions (by FiniteSet).
• Infinitely many solutions, both countably and uncountably infinite solutions (using the
ImageSet module).
• Interval
• There can also be bizarre solutions to equations like the set of rational numbers.
No other Python object (list, dictionary, generator, Python sets) provides the flexibility of
mathematical sets which our sets module tries to emulate. The second reason to use sets is
that they are close to the entities which mathematicians deal with and it makes it easier to
reason about them. Set objects conform to Pythonic conventions when possible, i.e., x in A
and for i in A both work when they can be computed. Another advantage of using objects
closer to mathematical entities is that the user won’t have to “learn” our representation and
she can have her expectations transferred from her mathematical experience.
For the multivariate case we represent solutions as a set of points in a n-dimensional space
and a point is represented by a FiniteSet of ordered tuples, which is a point in Rn or Cn .
Please note that, the general FiniteSet is unordered, but a FiniteSet with a tuple as its only
argument becomes ordered, since a tuple is ordered. So the order in the tuple is mapped to
a pre-defined order of variables while returning solutions.
For example:
solveset has a cleaner input API, unlike solve. It takes a maximum of three arguments:
solveset(equation, variable=None, domain=S.Complexes)
• Equation(s)
The equation(s) to solve.
• Variable(s)
The variable(s) for which the equation is to be solved.
• Domain
The domain in which the equation is to be solved.
solveset removes the flags argument of solve, which had made the input API
messy and output API inconsistent.
Solveset uses various methods to solve an equation, here is a brief overview of the
methodology:
• The domain argument is first considered to know the domain in which the user
is interested to get the solution.
• If the given function is a relational (>=, <=, >, <), and the domain is real, then
solve_univariate_inequality and solutions are returned. Solving for com-
plex solutions of inequalities, like x2 < 0 is not yet supported.
• Based on the domain, the equation is dispatched to one of the two functions
solveset_real or solveset_complex, which solves the given equation in the
complex or real domain, respectively.
• If the given expression is a product of two or more functions, like say gh =
0, then the solution to the given equation is the Union of the solution of the
equations g = 0 and h = 0, if and only if both g and h are finite for a finite input.
So, the solution is built up recursively.
• If the function is trigonometric or hyperbolic, the function _solve_real_trig
is called, which solves it by converting it to complex exponential form.
• The function is now checked if there is any instance of a Piecewise expression,
if it is, then it’s converted to explicit expression and set pairs and then solved
recursively.
• The respective solver now tries to invert the equation using the routines in-
vert_real and invert_complex. These routines are based on the concept of
mathematical inverse (though not exactly). It reduces the real/complex valued
equation f (x) = y to a set of equations: {g(x) = h1 (y), g(x) = h2 (y), ..., g(x) = hn (y)}
where g(x) is a simpler function than f (x). There is some work needed to be
done in this to find invert of more complex expressions.
• After the invert, the equations are checked for radical or Abs (Modulus), then
the method _solve_radical tries to simplify the radical, by removing it using
techniques like squaring, cubing etc, and _solve_abs solves nested Modulus
by considering the positive and negative variants, iteratively.
• If none of the above method is successful, then methods of polynomial is used
as follows:
– The method to solve the rational function, _solve_as_rational, is called.
Based on the domain, the respective poly solver _solve_as_poly_real or
_solve_as_poly_complex is called to solve f as a polynomial.
– The underlying method _solve_as_poly solves the equation using polyno-
mial techniques if it’s already a polynomial equation or, with a change of
variables, can be made so.
• The final solution set returned by solveset is the intersection of the set of
solutions found above and the input domain.
• In the real domain, we use our ImageSet class in the sets module to return infinite so-
lutions. ImageSet is an image of a set under a mathematical function. For example, to
represent the solution of the equation sin (x) = 0, we can use the ImageSet as:
Where n is a dummy variable. It is basically the image of the set of integers under the
function 2πn.
• In the complex domain, we use complex sets, which are implemented as the ComplexRe-
gion class in the sets module, to represent infinite solution in the Argand plane. For
example to represent the solution of the equation |z| = 1, which is a unit circle, we can
use the ComplexRegion as:
Where the FiniteSet in the ProductSet is the range of the value of r, which is the radius
of the circle and the Interval is the range of θ, the angle from the x axis representing a
unit circle in the Argand plane.
Note: We also have non-polar form notation for representing solution in rectangular
form. For example, to represent first two quadrants in the Argand plane, we can write
the ComplexRegion as:
where the Intervals are the range of x and y for the set of complex numbers x + iy.
5.35.8 How does solveset ensure that it is not returning any wrong
solution?
Since the output type is same as the input type any composition of these transfor-
mations is also a valid transformation. And our aim is to find the right sequence of
compositions (given the atoms) which transforms the given condition set to a set
which is not a condition set i.e., FiniteSet, Interval, Set of Integers and their Union,
Intersection, Complement or ImageSet. We can assign a cost function to each set,
such that, the more desirable that form of set is to us, the less the value of the cost
function. This way our problem is now reduced to finding the path from the initial
ConditionSet to the lowest valued set on a graph where the atomic transformations
forms the edges.
5.35.10 How do we deal with cases where only some of the solutions
are known?
Creating a universal equation solver, which can solve each and every equation we
encounter in mathematics is an ideal case for solvers in a Computer Algebra System.
When cases which are not solved or can only be solved incompletely, a Condition-
Set is used and acts as an unevaluated solveset object.
Note that, mathematically, finding a complete set of solutions for an equation is
undecidable. See Richardson’s theorem.
ConditionSet is basically a Set of elements which satisfy a given condition. For
example, to represent the solutions of the equation in the real domain:
(x2 − 4)(sin(x) + x)
There are still a few things solveset can’t do, which the old solve can, such as
solving non linear multivariate & LambertW type equations. Hence, it’s not yet a
perfect replacement for old solve. The ultimate goal is to:
• Replace solve with solveset once solveset is at least as powerful as solve,
i.e., solveset does everything that solve can do currently, and
• eventually rename solveset to solve.
This simply means n is the solution only when it belongs to the Interval [0, ∞) and
−n is the solution only when −n belongs to the Interval (−∞, 0].
There are other cases to address too, like solving 2x + (a − 2) for x where a is a
symbolic parameter. As of now, It returns the solution as an intersection with R,
which is trivial, as it doesn’t reveal the domain of a in the solution.
Recently, we have also implemented a function to find the domain of the expression
in a FiniteSet (Intersection with the interval) in which it is not-empty. It is a useful
addition for dealing with symbolic parameters. For example:
5.35.13 References
Use solveset() (page 1383) to solve equations or expressions (assumed to be equal to 0) for
a single variable. Solving an equation like x2 == 1 can be done as follows:
>>> solveset(x**2 - 1, x)
{-1, 1}
The first argument for solveset() (page 1383) is an expression (equal to zero) or an equation
and the second argument is the symbol that we want to solve the equation for.
sympy.solvers.solveset.solveset(f, symbol=None, domain=S.Complexes)
Solves a given inequality or equation with set as output
Parameters f : Expr or a relational.
The target equation or inequality
symbol : Symbol
The variable for which the equation is solved
domain : Set
The domain over which the equation is solved
Returns Set
Notes
Python interprets 0 and 1 as False and True, respectively, but in this function they refer
to solutions of an expression. So 0 and 1 return the Domain and EmptySet, respectively,
while True and False return the opposite (as they are assumed to be solutions of relational
expressions).
Examples
• The default domain is complex. Not specifying a domain will lead to the solving of
the equation in the complex domain (and this is not affected by the assumptions on
the symbol):
>>> x = Symbol('x')
>>> pprint(solveset(exp(x) - 1, x), use_unicode=False)
{2*n*I*pi | n in Integers}
• If you want to use solveset to solve the equation in the real domain, provide a real
domain. (Using solveset_real does this automatically.)
>>> R = S.Reals
>>> x = Symbol('x')
>>> solveset(exp(x) - 1, x, R)
{0}
>>> solveset_real(exp(x) - 1, x)
{0}
The solution is mostly unaffected by assumptions on the symbol, but there may be some
slight difference:
• Inequalities can be solved over the real domain only. Use of a complex domain leads
to a NotImplementedError.
sympy.solvers.solveset.solveset_real(f, symbol)
sympy.solvers.solveset.solveset_complex(f, symbol)
sympy.solvers.solveset.invert_real(f_x, y, x, domain=Reals)
Inverts a real-valued function. Same as _invert, but sets the domain to S.Reals before
inverting.
sympy.solvers.solveset.invert_complex(f_x, y, x, domain=S.Complexes)
Reduce the complex valued equation f(x) = y to a set of equations {g(x) = h_1(y),
g(x) = h_2(y), ..., g(x) = h_n(y) } where g(x) is a simpler function than f(x).
The return value is a tuple (g(x), set_h), where g(x) is a function of x and set_h
is the set of function {h_1(y), h_2(y), ..., h_n(y)}. Here, y is not necessarily a
symbol.
The set_h contains the functions, along with the information about the domain in which
they are valid, through set operations. For instance, if y = Abs(x) - n is inverted in
the real domain, then set_h is not simply −n, n as the nature of n is unknown; rather, it
is: Intersection([0, oo)n)U Intersection((−oo, 0], −n)
By default, the complex domain is used which means that inverting even seemingly sim-
ple functions like exp(x) will give very different results from those obtained in the real
domain. (In the case of exp(x), the inversion via log is multi-valued in the complex
domain, having infinitely many branches.)
If you are working with real values only (or you are not sure which function to use) you
should probably set the domain to S.Reals (or use invert_real which does that automati-
cally).
See also:
invert_real (page 1385), invert_complex (page 1385)
Examples
sympy.solvers.solveset.domain_check(f, symbol, p)
Returns False if point p is infinite or any subexpression of f is infinite or becomes so after
replacing symbol with p. If none of these conditions is met then True will be returned.
Examples
• The function relies on the assumption that the original form of the equation has not
been changed by automatic simplification.
5.35.15 linear_eq_to_matrix
sympy.solvers.solveset.linear_eq_to_matrix(equations, *symbols)
Converts a given System of Equations into Matrix form. Here equations must be a linear
system of equations in symbols. The order of symbols in input symbols will determine the
order of coefficients in the returned Matrix.
The Matrix form corresponds to the augmented matrix form. For example:
4x + 2y + 3z = 1
3x + y + z = −6
2x + 4y + 9z = 2
This system would return A & b as given below:
[ 4 2 3 ] [ 1 ]
A = [ 3 1 1 ] b = [-6 ]
[ 2 4 9 ] [ 2 ]
Examples
5.35.16 linsolve
sympy.solvers.solveset.linsolve(system, *symbols)
Solve system of N linear equations with M variables, which means both under - and
overdetermined systems are supported. The possible number of solutions is zero, one or
infinite. Zero solutions throws a ValueError, where as infinite solutions are represented
parametrically in terms of given symbols. For unique solution a FiniteSet of ordered
tuple is returned.
All Standard input formats are supported: For the given set of Equations, the respective
input types are given below:
3x + 2y − z = 1
2x − 2y + 4z = −2
2x − y + 2z = 0
• Augmented Matrix Form, system given below:
[3 2 -1 1]
system = [2 -2 4 -2]
[2 -1 2 0]
[3 2 -1 ] [ 1 ]
A = [2 -2 4 ] b = [ -2 ]
[2 -1 2 ] [ 0 ]
system = (A, b)
Symbols can always be passed but are actually only needed when 1) a system of equations
is being passed and 2) the system is passed as an underdetermined matrix and one wants
to control the name of the free variables in the result. An error is raised if no symbols are
used for case 1, but if no symbols are provided for case 2, internally generated symbols
will be provided. When providing symbols for case 2, there should be at least as many
symbols are there are columns in matrix A.
The algorithm used here is Gauss-Jordan elimination, which results, after elimination, in
a row echelon form matrix.
Returns A FiniteSet containing an ordered tuple of values for the
unknowns for which the system has a solution. (Wrapping
the tuple in FiniteSet is used to maintain a consistent
Examples
• Parametric Solution: In case the system is underdetermined, the function will return
a parametric solution in terms of the given symbols. Those that are free will be
returned unchanged. e.g. in the system below, z is returned as the solution for
variable z; it can take on any value.
If no symbols are given, internally generated symbols will be used. The tau0 in the 3rd
position indicates (as before) that the 3rd variable – whatever it’s named – can take on
any value:
>>> linsolve([], x)
EmptySet()
5.35.17 nonlinsolve
sympy.solvers.solveset.nonlinsolve(system, *symbols)
Solve system of N non linear equations with M variables, which means both under and
overdetermined systems are supported. Positive dimensional system is also supported
(A system with infinitely many solutions is said to be positive-dimensional). In Positive
dimensional system solution will be dependent on at least one symbol. Returns both real
solution and complex solution(If system have). The possible number of solutions is zero,
one or infinite.
Parameters system : list of equations
The target system of equations
symbols : list of Symbols
symbols should be given as a sequence eg. list
Returns A FiniteSet of ordered tuple of values of symbols for which the system
has solution. Order of values in the tuple is same as symbols present in
the parameter symbols.
Please note that general FiniteSet is unordered, the solution returned
here is not simply a FiniteSet of solutions, rather it is a FiniteSet of
ordered tuple, i.e. the first & only argument to FiniteSet is a tuple of
x∗y−1=0
4 ∗ x ∗ ∗2 + y ∗ ∗2 − 5 = 0
system = [x ∗ y − 1, 4 ∗ x ∗ ∗2 + y ∗ ∗2 − 5]
symbols = [x, y]
Raises ValueError
The input is not valid. The symbols are not given.
AttributeError
The input symbols are not Symbol type.
Examples
2. If some of the equations are non polynomial equation then nonlinsolve will call
substitution function and returns real and complex solutions, if present.
3. If system is Non linear polynomial zero dimensional then it returns both solution (real
and complex solutions, if present using solvep olys ystem):
5. System having polynomial equations and only real solution is present (will be solved
using solvep olys ystem):
1. A positive dimensional system solver : nonlinsolve can return solution for positive di-
mensional system. It finds the Groebner Basis of the positive dimensional system(calling
it as basis) then we can start solving equation(having least number of variable first in the
basis) using solveset and substituting that solved solutions into other equation(of basis)
to get solution in terms of minimum variables. Here the important thing is how we are
substituting the known values and in which equations.
2. Real and Complex both solutions : nonlinsolve returns both real and complex solu-
tion. If all the equations in the system are polynomial then using solvep olys ystem both
real and complex solution is returned. If all the equations in the system are not poly-
nomial equation then goes to substitution method with this polynomial and non polyno-
mial equation(s), to solve for unsolved variables. Here to solve for particular variable
solveset_real and solveset_complex is used. For both real and complex solution function
s olveu singk nowv alues is used inside substitution function.(substitution function will be called
when there is any non polynomial equation(s) is present). When solution is valid then
add its general solution in the final result.
3. Complement and Intersection will be added if any : nonlinsolve maintains dict for
complements and Intersections. If solveset find complements or/and Intersection with
any Interval or set during the execution of substitution function ,then complement or/and
Intersection for that variable is added before returning final solution.
5.35.18 transolve
Examples
_transolve should not be used as an independent function, because it assumes that the
equation (f ) and the symbol comes from solveset and might have undergone a few modi-
fication(s). To use _transolve as an independent function the equation (f ) and the symbol
should be passed as they would have been by solveset.
_transolve uses two types of helper functions to solve equations of a particular class:
1) Better output
2) Extensible
The API of _transolve is designed such that it is easily extensible, i.e. the code that solves
a given class of equations is encapsulated in a helper and not mixed in with the code of
_transolve itself.
3) Modular
_transolve is designed to be modular i.e, for every class of equation a separate helper for
identification and solving is implemented. This makes it easy to change or modify any
of the method implemented directly in the helpers without interfering with the actual
structure of the API.
4) Faster Computation
Solving equation via _transolve is much faster as compared to _tsolve. In solve, attempts
are made computing every possibility to get the solutions. This series of attempts makes
solving a bit slow. In _transolve, computation begins only after a particular type of equa-
tion is identified.
Examples
The function extracts each term of the equation and checks if it is of exponential form
w.r.t symbol.
sympy.solvers.solveset._solve_exponential(lhs, rhs, symbol, domain)
Helper function for solving (supported) exponential equations.
Exponential equations are the sum of (currently) at most two terms with one or both of
them having a power with a symbol-dependent exponent.
For example
52x+3 − 53x−1
45−9x − e2−x
Parameters lhs, rhs : Expr
The exponential equation to be solved, lhs = rhs
symbol : Symbol
The variable in which the equation is solved
domain : Set
A set over which the equation is solved.
Returns A set of solutions satisfying the given equation.
A ConditionSet if the equation is unsolvable or
if the assumptions are not properly defined, in that case
a different style of ConditionSet is returned having the
solution(s) of the equation with the desired assumptions.
Examples
The logarithm function is the inverse of the exponential function. The defining relation
between exponentiation and logarithm is:
logb x = y if by = x
Therefore if we are given an equation with exponent terms, we can convert every term
to its corresponding log form. This is achieved by taking logarithms and expanding the
equation using log identities so that it can easily be handled by solveset.
For example:
32x = 2x+3
Taking log both sides will reduce the equation to
(2x) ∗ log(3) = (x + 3) ∗ log(2)
Examples
• Proof of correctness
logb x = y if by = x
When one side of the equation contains a single logarithm, the equation can be solved
by rewriting the equation as an equivalent exponential equation as defined above. But if
one side contains more than one logarithm, we need to use the properties of logarithm
to condense it into a single logarithm.
Take for example
log(2 ∗ x) − 15 = 0
x = exp(15)/2
log(x − 3) + log(x + 3) = 0
log((x − 3) ∗ (x + 3))
This equation contains one logarithm and can be solved by rewriting to exponents.
sympy.solvers.solveset._is_logarithmic(f, symbol)
Return True if the equation is in the form a ∗ log(f (x)) + b ∗ log(g(x)) + ... + c else False.
Parameters f : Expr
The equation to be checked
symbol : Symbol
The variable in which the equation is checked
Returns T rue if the equation is logarithmic otherwise F alse.
Examples
The function extracts each term and checks whether it is logarithmic w.r.t symbol.
5.35.20 Inequalities
5.36.1 Contents
N-dim array
Four classes are provided to handle N-dim arrays, given by the combinations dense/sparse
(i.e. whether to store all elements or only the non-zero ones in memory) and muta-
ble/immutable (immutable classes are SymPy objects, but cannot change after they have been
created).
Examples
The following examples show the usage of Array. This is an abbreviation for Immutable-
DenseNDimArray, that is an immutable and dense N-dim array, the other classes are analo-
gous. For mutable classes it is also possible to change element values after the object has
been constructed.
Array construction can detect the shape of nested lists and tuples:
>>> from sympy import Array
>>> a1 = Array([[1, 2], [3, 4], [5, 6]])
>>> a1
[[1, 2], [3, 4], [5, 6]]
>>> a1.shape
(3, 2)
>>> a1.rank()
2
>>> from sympy.abc import x, y, z
>>> a2 = Array([[[x, y], [z, x*z]], [[1, x*y], [1/x, x/y]]])
>>> a2
[[[x, y], [z, x*z]], [[1, x*y], [1/x, x/y]]]
>>> a2.shape
(2, 2, 2)
>>> a2.rank()
3
Slice support:
>>> m2[:, 1, 1]
[3, 7, 11]
Elementwise derivative:
>>> from sympy.abc import x, y, z
>>> m3 = Array([x**3, x*y, z])
>>> m3.diff(x)
[3*x**2, y, 0]
>>> m3.diff(z)
[0, 0, 1]
>>> (1+x)*m3
[x**3*(x + 1), x*y*(x + 1), z*(x + 1)]
>>> m2.tolist()
[[[0, 1], [2, 3]], [[4, 5], [6, 7]], [[8, 9], [10, 11]]]
>>> isinstance(m2.tolist(), list)
True
>>> m1.tomatrix()
Matrix([
[0, 1, 2, 3],
[4, 5, 6, 7],
[8, 9, 10, 11]])
Tensor product between arrays Ai1 ,...,in and Bj1 ,...,jm creates the combined array P = A ⊗ B
defined as
Pi1 ,...,in ,j1 ,...,jm := Ai1 ,...,in · Bj1 ,...,jm .
It is available through tensorproduct(...):
Tensor product between a rank-1 array and a matrix creates a rank-3 array:
,→0, z]], [[t, 0, 0, 0], [0, t, 0, 0], [0, 0, t, 0], [0, 0, 0, t]]]
>>> p1[0,:,:]
[[x, 0, 0, 0], [0, x, 0, 0], [0, 0, x, 0], [0, 0, 0, x]]
Tensor contraction sums over the specified axes, for example contracting positions a and b
means
∑
Ai1 ,...,ia ,...,ib ,...,in =⇒ k Ai1 ,...,k,...,k,...,in
Remember that Python indexing is zero starting, to contract the a-th and b-th axes it is there-
fore necessary to specify a − 1 and b − 1
Matrix product is equivalent to a tensor product of two rank-2 arrays, followed by a contrac-
tion of the 2nd and 3rd axes (in Python indexing axes number 1, 2).
∑
Am,n · Bi,j =⇒ k Am,k · Bk,j
or equivalently
>>> C.tomatrix()*D.tomatrix()
Matrix([
[2*x, x - y],
[2*z, -t + z]])
Derivatives by array
The usual derivative operation may be extended to support derivation with respect to arrays,
provided that all elements in the that array are symbols or expressions suitable for derivations.
The definition of a derivative by an array is as follows: given the array Ai1 ,...,iN and the array
Xj1 ,...,jM the derivative of arrays will return a new array B defined by
∂Ai1 ,...,iN
Bj1 ,...,jM ,i1 ,...,iN := ∂Xj1 ,...,jM
>>> derive_by_array(sin(x*y), x)
y*cos(x*y)
∂Am
Deriving array by an array basis: B nm := ∂xn
Classes
class sympy.tensor.array.ImmutableDenseNDimArray
class sympy.tensor.array.ImmutableSparseNDimArray
class sympy.tensor.array.MutableDenseNDimArray
class sympy.tensor.array.MutableSparseNDimArray
Functions
sympy.tensor.array.derive_by_array(expr, dx)
Derivative by arrays. Supports both arrays and scalars.
Given the array Ai1 ,...,iN and the array Xj1 ,...,jM this function will return a new array B
defined by
∂Ai1 ,...,iN
Bj1 ,...,jM ,i1 ,...,iN := ∂Xj1 ,...,jM
Examples
sympy.tensor.array.permutedims(expr, perm)
Permutes the indices of an array.
Parameter specifies the permutation of the indices.
Examples
sympy.tensor.array.tensorcontraction(array, *contraction_axes)
Contraction of an array-like object on the specified axes.
Examples
sympy.tensor.array.tensorproduct(*args)
Tensor product among scalars or array-like objects.
Examples
,→[z, t]]]]
Indexed Objects
Examples
========
>>> x = IndexedBase('x')
>>> M[i, j]*x[j]
M[i, j]*x[j]
To analyze the structure of indexed expressions, you can use the methods
get_indices() and get_contraction_structure():
class sympy.tensor.indexed.Idx
Represents an integer index as an Integer or integer expression.
There are a number of ways to create an Idx object. The constructor takes two argu-
ments:
label An integer or a symbol that labels the index.
range Optionally you can specify a range as either
• Symbol or integer: This is interpreted as a dimension. Lower and upper bounds
are set to 0 and range - 1, respectively.
• tuple: The two elements are interpreted as the lower and upper bounds of the
range, respectively.
Note: bounds of the range are assumed to be either integer or infinite (oo and -oo are
allowed to specify an unbounded range). If n is given as a bound, then n.is_integer
must not return false.
For convenience, if the label is given as a string it is automatically converted to an integer
symbol. (Note: this conversion is not done for range or dimension arguments.)
Examples
If a string is given for the label an integer Symbol is created and the bounds are both
None:
When only a single bound is given it is interpreted as the dimension and the lower bound
defaults to 0:
label
Returns the label (Integer or integer expression) of the Idx object.
Examples
lower
Returns the lower bound of the Idx.
Examples
upper
Returns the upper bound of the Idx.
Examples
class sympy.tensor.indexed.Indexed
Represents a mathematical object with indices.
>>> A = IndexedBase('A')
>>> Indexed('A', i, j) == A[i, j]
True
base
Returns the IndexedBase of the Indexed object.
Examples
indices
Returns the indices of the Indexed object.
Examples
ranges
Returns a list of tuples with lower and upper range of each index.
If an index does not define the data members upper and lower, the corresponding
slot in the list contains None instead of a tuple.
Examples
rank
Returns the rank of the Indexed object.
Examples
shape
Returns a list with dimensions of each index.
Dimensions is a property of the array, not of the indices. Still, if the IndexedBase
does not define a shape attribute, it is assumed that the ranges of the indices corre-
spond to the shape of the array.
class sympy.tensor.indexed.IndexedBase
Represent the base or stem of an indexed object
The IndexedBase class represent an array that contains elements. The main purpose
of this class is to allow the convenient creation of objects of the Indexed class. The
__getitem__ method of IndexedBase returns an instance of Indexed. Alone, without in-
dices, the IndexedBase class can be used as a notation for e.g. matrix equations, re-
sembling what you could do with the Symbol class. But, the IndexedBase class adds
functionality that is not available for Symbol instances:
• An IndexedBase object can optionally store shape information. This can be used in
to check array conformance and conditions for numpy broadcasting. (TODO)
• An IndexedBase object implements syntactic sugar that allows easy symbolic repre-
sentation of array operations, using implicit summation of repeated indices.
• The IndexedBase object symbolizes a mathematical structure equivalent to arrays,
and is recognized as such for code generation and automatic compilation and wrap-
ping.
When an IndexedBase object receives indices, it returns an array with named axes, rep-
resented by an Indexed object:
label
Returns the label of the IndexedBase object.
Examples
offset
Returns the offset for the IndexedBase object.
This is the value added to the resulting index when the 2D Indexed object is unrolled
to a 1D form. Used in code generation.
Examples
shape
Returns the shape of the IndexedBase object.
Examples
Note: If the shape of the IndexedBase is specified, it will override any shape infor-
mation given by the indices.
strides
Returns the strided scheme for the IndexedBase object.
Normally this is a tuple denoting the number of steps to take in the respective di-
mension when traversing an array. For code generation purposes strides=’C’ and
strides=’F’ can also be used.
strides=’C’ would mean that code printer would unroll in row-major order and ‘F’
means unroll in column major order.
Methods
Note: The presence of expressions among the dictinary keys indicates multiple levels
of index contractions. A nested dict displays nested contractions and may itself con-
tain dicts from a deeper level. In practical calculations the summation in the deepest
nested level must be calculated first so that the outer expression can access the resulting
indexed object.
Examples
>>> d[None]
{x[i, i]*y[j, j]}
A parenthesized Add object is also returned as a nested dictionary. The term containing
the parenthesis is a Mul with a contraction among the arguments, so it will be found as
a key in the result. It stores the dictionary resulting from a recursive call on the Add
expression.
Powers with contractions in either base or exponent will also be found as keys in the
dictionary, mapping to a list of results from recursive calls:
The description of the contraction structure may appear complicated when represented
with a string in the above examples, but it is easy to iterate over:
sympy.tensor.index_methods.get_indices(expr)
Determine the outer indices of expression expr
By outer we mean indices that are not summation indices. Returns a set and a dict. The
set contains outer indices and the dict contains information about index symmetries.
Examples
The indices of the total expression is determined, Repeated indices imply a summation,
for instance the trace of a matrix A:
In the case of many terms, the terms are required to have identical outer indices. Else
an IndexConformanceException is raised.
Exceptions
Warning: The concept of outer indices applies recursively, starting on the deepest
level. This implies that dummies inside parenthesis are assumed to be summed first,
so that the following expression is handled gracefully:
>>> get_indices((x[i] + A[i, j]*y[j])*x[j])
({i, j}, {})
This is correct and may appear convenient, but you need to be careful with this as
SymPy will happily .expand() the product, if requested. The resulting expression
would mix the outer j with the dummies inside the parenthesis, which makes it a
different expression. To be on the safe side, it is best to avoid such ambiguities by
using unique indices for all contractions that should be held separate.
Tensor
class sympy.tensor.tensor._TensorManager
Class to manage tensor properties.
Notes
Tensors belong to tensor commutation groups; each group has a label comm; there are
predefined labels:
0 tensors commuting with any other tensor
1 tensors anticommuting among themselves
2 tensors not commuting, apart with those with comm=0
Other groups can be defined using set_comm; tensors in those groups commute with
those with comm=0; by default they do not commute with any other group.
clear()
Clear the TensorManager.
comm_i2symbol(i)
Returns the symbol corresponding to the commutation group number.
comm_symbols2i(i)
get the commutation group number corresponding to i
Notes
i, j can be symbols, strings or numbers, apart from 0, 1 and 2 which are reserved
respectively for commuting, anticommuting tensors and tensors not commuting with
any other group apart with the commuting tensors. For the remaining cases, use
this method to set the commutation rules; by default c=None.
The group commutation number c is assigned in correspondence to the group com-
mutation symbols; it can be
0 commuting
1 anticommuting
None no commutation property
Examples
G and GH do not commute with themselves and commute with each other; A is com-
muting.
set_comms(*args)
set the commutation group numbers c for symbols i, j
Parameters args : sequence of (i, j, c)
class sympy.tensor.tensor.TensorIndexType
A TensorIndexType is characterized by its name and its metric.
Notes
The metric parameter can be: metric = False symmetric metric (in Riemannian ge-
ometry)
metric = True antisymmetric metric (for spinor calculus)
metric = None there is no metric
metric can be an object having name and antisym attributes.
If there is a metric the metric is used to raise and lower indices.
In the case of antisymmetric metric, the following raising and lowering conventions will
be adopted:
psi(a) = g(a, b)*psi(-b); chi(-a) = chi(b)*g(-b, -a)
g(-a, b) = delta(-a, b); g(b, -a) = -delta(a, -b)
where delta(-a, b) = delta(b, -a) is the Kronecker delta (see TensorIndex for the
conventions on indices).
If there is no metric it is not possible to raise or lower indices; e.g. the index of the
defining representation of SU(N) is ‘covariant’ and the conjugate representation is ‘con-
travariant’; for N > 2 they are linearly independent.
eps_dim is by default equal to dim, if the latter is an integer; else it can be assigned (for
use in naive dimensional regularization); if eps_dim is not an integer epsilon is None.
Examples
Examples with metric components data added, this means it is working on a fixed basis:
Attributes
name
metric_name (it is ‘metric’ or metric.name)
metric_antisym
metric (the metric tensor)
delta (Kronecker delta)
epsilon (the Levi-Civita epsilon tensor)
dim
eps_dim
dummy_fmt
data (a property to add ndarray values, to work in a specified basis.)
class sympy.tensor.tensor.TensorIndex
Represents an abstract tensor index.
Parameters name : name of the index, or True if you want it to be automati-
cally assigned
tensortype : TensorIndexType of the index
is_up : flag for contravariant index
Notes
Examples
If you want the index name to be automatically assigned, just put True in the name field,
it will be generated using the reserved character _ in front of its name, in order to avoid
conflicts with possible existing indices:
Attributes
name
tensortype
is_up
sympy.tensor.tensor.tensor_indices(s, typ)
Returns list of tensor indices given their names and their types
Parameters s : string of comma separated names of indices
typ : TensorIndexType of the indices
Examples
class sympy.tensor.tensor.TensorSymmetry
Monoterm symmetry of a tensor
Parameters bsgs : tuple (base, sgs) BSGS of the symmetry of the tensor
See also:
sympy.combinatorics.tensor_can.get_symmetric_group_sgs (page 300)
Notes
A tensor can have an arbitrary monoterm symmetry provided by its BSGS. Multiterm
symmetries, like the cyclic symmetry of the Riemann tensor, are not covered.
Examples
Attributes
sympy.tensor.tensor.tensorsymmetry(*args)
Return a TensorSymmetry object.
One can represent a tensor with any monoterm slot symmetry group using a BSGS.
args can be a BSGS args[0] base args[1] sgs
Usually tensors are in (direct products of) representations of the symmetric group; args
can be a list of lists representing the shapes of Young tableaux
Notes
For instance: [[1]] vector [[1]*n] symmetric tensor of rank n [[n]] antisymmetric
tensor of rank n [[2, 2]] monoterm slot symmetry of the Riemann tensor [[1],[1]]
vector*vector [[2],[1],[1] (antisymmetric tensor)*vector*vector
Notice that with the shape [2, 2] we associate only the monoterm symmetries of the
Riemann tensor; this is an abuse of notation, since the shape [2, 2] corresponds usually
to the irreducible representation characterized by the monoterm symmetries and by the
cyclic symmetry.
Examples
class sympy.tensor.tensor.TensorType
Class of tensor types.
Parameters index_types : list of TensorIndexType of the tensor indices
symmetry : TensorSymmetry of the tensor
Examples
Attributes
index_types
symmetry
types (list of TensorIndexType without repetitions)
class sympy.tensor.tensor.TensorHead
Tensor head of the tensor
Parameters name : name of the tensor
typ : list of TensorIndexType
comm : commutation group number
Notes
Examples
Examples with ndarray values, the components data assigned to the TensorHead object
are assumed to be in a fully-contravariant representation. In case it is necessary to
assign components data which represents the values of a non-fully covariant tensor, see
the other examples.
in order to retrieve data, it is also necessary to specify abstract indices enclosed by round
brackets, then numerical indices inside square brackets.
>>> A.data
[[0, 1, 2, 3], [2, 3, 4, 5], [4, 5, 6, 7], [6, 7, 8, 9]]
Turning to a tensor expression, covariant indices get the corresponding components data
corrected by the metric:
while if all indices are contravariant, the ndarray remains the same
When all indices are contracted and components data are added to the tensor, accessing
the data will return a scalar, no array object. In fact, arrays are dropped to scalars if
they contain only one element.
It is also possible to assign components data to an indexed tensor, i.e. a tensor with
specified covariant and contravariant components. In this example, the covariant com-
ponents data of the Electromagnetic tensor are injected into A:
>>> F.data.tomatrix()
Matrix([
[ 0, -E_x/c, -E_y/c, -E_z/c],
[E_x/c, 0, -B_z, B_y],
[E_y/c, B_z, 0, -B_x],
[E_z/c, -B_y, B_x, 0]])
Still notice, in this last example, that accessing components data from a tensor without
specifying the indices is equivalent to assume that all indices are contravariant.
It is also possible to store symbolic components data inside a tensor, for example, define
a four-momentum-like tensor:
>>> P(i0).data
[E, p_x, p_y, p_z]
>>> P(-i0).data
[E, -p_x, -p_y, -p_z]
>>> P(i0)**2
E**2 - p_x**2 - p_y**2 - p_z**2
>>> P**2
E**2 - p_x**2 - p_y**2 - p_z**2
Attributes
name
index_types
rank
types ( equal to typ.types)
symmetry (equal to typ.symmetry)
comm (commutation group)
commutes_with(other)
Returns 0 if self and other commute, 1 if they anticommute.
Returns None if self and other neither commute nor anticommute.
class sympy.tensor.tensor.TensExpr
Abstract base class for tensor expressions
Notes
Examples
get_matrix()
Returns ndarray components data as a matrix, if components data are available and
ndarray dimension does not exceed 2.
Examples
The tensor A is symmetric in its indices, as can be deduced by the [1, 1] Young
tableau when constructing sym2. One has to be careful to assign symmetric compo-
nent data to A, as the symmetry properties of data are currently not checked to be
compatible with the defined tensor symmetry.
class sympy.tensor.tensor.TensAdd
Sum of tensors
Parameters free_args : list of the free indices
Notes
Sum of more than one tensor are put automatically in canonical form.
Examples
>>> (p(a)*p(-a)).data
-58
>>> p(a)**2
-58
Attributes
canon_bp()
canonicalize using the Butler-Portugal algorithm for canonicalization under
monoterm symmetries.
contract_metric(g)
Raise or lower indices with the metric g
Parameters g : metric
contract_all : if True, eliminate all g which are contracted
Notes
Examples
substitute_indices(*index_tuples)
Return a tensor with free indices substituted according to index_tuples
Parameters index_types : list of tuples (old_index, new_index)
Examples
class sympy.tensor.tensor.TensMul
Product of tensors
Parameters coeff : SymPy coefficient of the tensor
args
Notes
Attributes
canon_bp()
Canonicalize using the Butler-Portugal algorithm for canonicalization under
monoterm symmetries.
Examples
contract_metric(g)
Raise or lower indices with the metric g
Parameters g : metric
Notes
Examples
get_free_indices()
Returns the list of free indices of the tensor
The indices are listed in the order in which they appear in the component tensors.
Examples
get_indices()
Returns the list of indices of the tensor
The indices are listed in the order in which they appear in the component tensors.
The dummy indices are given a name which does not collide with the names of the
free indices.
Examples
perm2tensor(g, is_canon_bp=False)
Returns the tensor corresponding to the permutation g
For further details, see the method in TIDS with the same name.
sorted_components()
Returns a tensor product with sorted components.
split()
Returns a list of tensors, whose product is self
Dummy indices contracted among different tensor components become free indices
with the same name as the one used to represent the dummy indices.
Examples
sympy.tensor.tensor.canon_bp(p)
Butler-Portugal canonicalization
sympy.tensor.tensor.tensor_mul(*a)
product of tensors
sympy.tensor.tensor.riemann_cyclic_replace(t_r)
replace Riemann tensor with an equivalent expression
R(m,n,p,q) -> 2/3*R(m,n,p,q) - 1/3*R(m,q,n,p) + 1/3*R(m,p,n,q)
sympy.tensor.tensor.riemann_cyclic(t2)
replace each Riemann tensor with an equivalent expression satisfying the cyclic identity.
This trick is discussed in the reference guide to Cadabra.
Examples
5.37 Utilities
This module contains some general purpose utilities that are used across SymPy.
Contents:
The autowrap module works very well in tandem with the Indexed classes of the Tensor Mod-
ule (page 1399). Here is a simple example that shows how to setup a binary routine that
calculates a matrix-vector product.
Because the code printers treat Indexed objects with repeated indices as a summation, the
above equality instance will be translated to low-level code for a matrix vector product. This
is how you tell SymPy to generate the code, compile it and wrap it as a python function:
That’s it. Now let’s test it with some numpy arrays. The default wrapper backend is f2py. The
wrapper function it provides is set up to accept python lists, which it will silently convert to
numpy arrays. So we can test the matrix vector product like this:
Implementation details
API Reference
Module for compiling codegen output, and wrap the binary for use in python.
This module provides a common interface for different external backends, such as f2py, fwrap,
Cython, SWIG(?) etc. (Currently only f2py and Cython are implemented) The goal is to provide
access to compiled binaries of acceptable performance with a one-button user interface, i.e.
The callable returned from autowrap() is a binary python function, not a SymPy object. If
it is desired to use the compiled function in symbolic expressions, it is better to use bi-
nary_function() which returns a SymPy Function object. The binary callable is attached as the
_imp_ attribute and invoked when a numerical evaluation is requested with evalf(), or with
lambdify().
The idea is that a SymPy user will primarily be interested in working with mathematical
expressions, and should not have to learn details about wrapping tools in order to evaluate
expressions numerically, even if they are computationally expensive.
When is this useful?
1) For computations on large arrays, Python iterations may be too slow, and depending on
the mathematical expression, it may be difficult to exploit the advanced index operations
provided by NumPy.
2) For really long expressions that will be called repeatedly, the compiled binary should be
significantly faster than SymPy’s .evalf()
3) If you are generating code with the codegen utility in order to use it in another project,
the automatic python wrappers let you test the binaries immediately from within SymPy.
4) To create customized ufuncs for use with numpy arrays. See ufuncify.
When is this module NOT the best approach?
1) If you are really concerned about speed or memory optimizations, you will probably
get better results by working directly with the wrapper tools and the low level code.
However, the files generated by this utility may provide a useful starting point and
reference code. Temporary files will be left intact if you supply the keyword tem-
pdir=”path/to/files/”.
2) If the array computation can be handled easily by numpy, and you don’t need the binaries
for another project.
class sympy.utilities.autowrap.CodeWrapper(generator, filepath=None, flags=[],
verbose=False)
Base Class for code wrappers
class sympy.utilities.autowrap.CythonCodeWrapper(*args, **kwargs)
Wrapper that uses Cython
dump_pyx(routines, f, prefix)
Write a Cython file with python wrappers
This file contains all the definitions of the routines in c code and refers to the header
file.
Arguments
Arguments
Examples
Examples
References
[1] http://docs.scipy.org/doc/numpy/reference/ufuncs.html
Examples
For the ‘f2py’ and ‘cython’ backends, inputs are required to be equal length 1-
dimensional arrays. The ‘f2py’ backend will perform type conversion, but the Cython
backend will error if the inputs are not of the expected type.
Note
The default backend (‘numpy’) will create actual instances of numpy.ufunc. These sup-
port ndimensional broadcasting, and implicit type conversion. Use of the other backends
will result in a “ufunc-like” function, which requires equal length 1-dimensional arrays
for all arguments, and will not perform any type conversions.
5.37.2 Codegen
This module provides functionality to generate directly compilable code from SymPy expres-
sions. The codegen function is the user interface to the code generation functionality in
SymPy. Some details of the implementation is given below for advanced users that may want
to use the framework directly.
Note: The codegen callable is not in the sympy namespace automatically, to use it you must
first execute
Implementation Details
Here we present the most important pieces of the internal structure, as advanced users may
want to use it directly, for instance by subclassing a code generator for a specialized applica-
tion. It is very likely that you would prefer to use the codegen() function documented
above.
Basic assumptions:
• A generic Routine data structure describes the routine that must be translated into
C/Fortran/… code. This data structure covers all features present in one or more of
the supported languages.
• Descendants from the CodeGen class transform multiple Routine instances into compil-
able code. Each derived class translates into a specific language.
• In many cases, one wants a simple workflow. The friendly functions in the last part are
a simple api on top of the Routine/CodeGen stuff. They are easier to use, but are less
powerful.
Routine
The Routine class is a very important piece of the codegen module. Viewing the codegen
utility as a translator of mathematical expressions into a set of statements in a programming
language, the Routine instances are responsible for extracting and storing information about
how the math can be encapsulated in a function call. Thus, it is the Routine constructor that
decides what arguments the routine will need and if there should be a return value.
API Reference
module for generating C, C++, Fortran77, Fortran90, Julia, Rust and Octave/Matlab routines
that evaluate sympy expressions. This module is work in progress. Only the milestones with
a ‘+’ character in the list below have been completed.
— How is sympy.utilities.codegen different from sympy.printing.ccode? —
We considered the idea to extend the printing routines for sympy functions in such a way that
it prints complete compilable code, but this leads to a few unsurmountable issues that can
only be tackled with dedicated code generator:
• For C, one needs both a code and a header file, while the printing routines generate just
one string. This code generator can be extended to support .pyf files for f2py.
• SymPy functions are not concerned with programming-technical issues, such as input,
output and input-output arguments. Other examples are contiguous or non-contiguous
arrays, including headers of other libraries such as gsl or others.
• It is highly interesting to evaluate several sympy functions in one C routine, eventually
sharing common intermediate results with the help of the cse routine. This is more than
just printing.
• From the programming perspective, expressions with constants should be evaluated in
the code generator as much as possible. This is different for printing.
— Basic assumptions —
• A generic Routine data structure describes the routine that must be translated into
C/Fortran/… code. This data structure covers all features present in one or more of
the supported languages.
• Descendants from the CodeGen class transform multiple Routine instances into compil-
able code. Each derived class translates into a specific language.
• In many cases, one wants a simple workflow. The friendly functions in the last part are
a simple api on top of the Routine/CodeGen stuff. They are easier to use, but are less
powerful.
— Milestones —
• First working version with scalar input arguments, generating C code, tests
• Friendly functions that are easier to use than the rigorous Routine/CodeGen workflow.
• Integer and Real numbers as input and output
• Output arguments
• InputOutput arguments
• Sort input/output arguments properly
• Contiguous array arguments (numpy matrices)
• Also generate .pyf code for f2py (in autowrap module)
• Isolate constants and evaluate them beforehand in double precision
• Fortran 90
• Octave/Matlab
• Common Subexpression Elimination
• User defined comments in the generated code
• Optional extra include lines for libraries/objects that can eval special functions
• Test other C compilers and libraries: gcc, tcc, libtcc, gcc+gsl, …
• Contiguous array arguments (sympy matrices)
• Non-contiguous array arguments (sympy matrices)
• ccode must raise an error when it encounters something that can not be translated into
c. ccode(integrate(sin(x)/x, x)) does not make sense.
• Complex numbers as input and output
• A default complex datatype
• Include extra information in the header: date, user, hostname, sha1 hash, …
• Fortran 77
• C++
• Python
• Julia
• Rust
• …
class sympy.utilities.codegen.Routine(name, arguments, results, local_vars,
global_vars)
Generic description of evaluation routine for set of expressions.
A CodeGen class can translate instances of this class into code in a particular language.
The routine specification covers all the features present in these languages. The Code-
Gen part must raise an exception when certain features are not present in the target
language. For example, multiple return values are possible in Python, but not in C or
Fortran. Another example: Fortran and Python support complex numbers, while C does
not.
result_variables
Returns a list of OutputArgument, InOutArgument and Result.
If return values are present, they are at the end ot the list.
variables
Returns a set of all variables possibly used in the routine.
For routines with unnamed return values, the dummies that may or may not be used
will be included in the set.
class sympy.utilities.codegen.DataType(cname, fname, pyname, jlname, octname,
rsname)
Holds strings for a certain datatype in different languages.
sympy.utilities.codegen.get_default_datatype(expr)
Derives an appropriate datatype based on the expression.
class sympy.utilities.codegen.Argument(name, datatype=None, dimen-
sions=None, precision=None)
An abstract Argument data structure: a name and a data type.
This structure is refined in the descendants below.
Attributes
printer
ment_sequence is None, arguments will be ordered alphabetically, but with all In-
putArguments first, and then OutputArgument and InOutArguments.
write(routines, prefix, to_files=False, header=True, empty=True)
Writes all the source code files for the given routines.
The generated source is returned as a list of (filename, contents) tuples, or is written
to files (see below). Each filename consists of the given prefix, appended with an
appropriate extension.
Parameters routines : list
A list of Routine instances to be written
prefix : string
The prefix for the output files
to_files : bool, optional
When True, the output is written to files. Otherwise, a list of (file-
name, contents) tuples is returned. [default: False]
header : bool, optional
When True, a header comment is included on top of each source file.
[default: True]
empty : bool, optional
When True, empty lines are included to structure the source files.
[default: True]
class sympy.utilities.codegen.CCodeGen(project=’project’, printer=None, prepro-
cessor_statements=None, cse=False)
Generator for C code.
The .write() method inherited from CodeGen will output a code file and an interface file,
<prefix>.c and <prefix>.h respectively.
Attributes
printer
Attributes
printer
Attributes
printer
Attributes
printer
The filename prefix, used to refer to the proper header file. Only the
basename of the prefix is used.
header : bool, optional
When True, a header comment is included on top of each source file.
[default : True]
empty : bool, optional
When True, empty lines are included to structure the source files.
[default : True]
routine(name, expr, argument_sequence, global_vars)
Specialized Routine creation for Octave.
class sympy.utilities.codegen.RustCodeGen(project=’project’, printer=None)
Generator for Rust code.
The .write() method inherited from CodeGen will output a code file <prefix>.rs
Attributes
printer
Examples
Another example using Equality objects to give named outputs. Here the filename (pre-
fix) is taken from the first (name, expr) pair.
If the generated function(s) will be part of a larger project where various global variables
have been defined, the ‘global_vars’ option can be used to remove the specified variables
from the function signature
Examples
[f, out_8598435338387848786]
>>> [a.expr for a in r.arguments if isinstance(a, OutputArgument)]
[1, Matrix([[x, 2]])]
5.37.3 Decorator
Adds metadata about the dependencies which need to be met for doctesting the doc-
strings of the decorated objects.
sympy.utilities.decorator.memoize_property(storage)
Create a property, where the lookup is stored in storage
class sympy.utilities.decorator.no_attrs_in_subclass(cls, f)
Don’t ‘inherit’ certain attributes from a base class
>>> from sympy.utilities.decorator import no_attrs_in_subclass
sympy.utilities.decorator.public(obj)
Append obj’s name to global __all__ variable (call site).
By using this decorator on functions or classes you achieve the same goal as by fill-
ing __all__ variables manually, you just don’t have to repeat yourself (object’s name).
You also know if object is public at definition site, not at some random location (where
__all__ was set).
Note that in multiple decorator setup (in almost all cases) @public decorator must be
applied before any other decorators, because it relies on the pointer to object’s global
namespace. If you apply other decorators first, @public may end up modifying the wrong
namespace.
Examples
>>> __all__
Traceback (most recent call last):
...
NameError: name '__all__' is not defined
>>> @public
... def some_function():
... pass
>>> __all__
['some_function']
sympy.utilities.decorator.threaded(func)
Apply func to sub–elements of an object, including Add.
This decorator is intended to make it uniformly possible to apply a function to all elements
of composite objects, e.g. matrices, lists, tuples and other iterable containers, or just
expressions.
This version of threaded() (page 1450) decorator allows threading over elements of Add
class. If this behavior is not desirable use xthreaded() (page 1451) decorator.
Functions using this decorator must have the following signature:
@threaded
def function(expr, *args, **kwargs):
sympy.utilities.decorator.threaded_factory(func, use_add)
A factory for threaded decorators.
sympy.utilities.decorator.xthreaded(func)
Apply func to sub–elements of an object, excluding Add.
This decorator is intended to make it uniformly possible to apply a function to all elements
of composite objects, e.g. matrices, lists, tuples and other iterable containers, or just
expressions.
This version of threaded() (page 1450) decorator disallows threading over elements of
Add class. If this behavior is not desirable use threaded() (page 1450) decorator.
Functions using this decorator must have the following signature:
@xthreaded
def function(expr, *args, **kwargs):
5.37.4 Enumerative
This module includes functions and classes for enumerating and counting multiset partitions.
sympy.utilities.enumerative.multiset_partitions_taocp(multiplicities)
Enumerates partitions of a multiset.
Parameters multiplicities
list of integer multiplicities of the components of the multiset.
Yields state
Internal data structure which encodes a particular partition. This out-
put is then usually processed by a vistor function which combines the
information from this data structure with the components themselves
to produce an actual partition.
Unless they wish to create their own visitor function, users will have
little need to look inside this data structure. But, for reference, it is
a 3-element list with components:
f is a frame array, which is used to divide pstack into parts.
lpart points to the base of the topmost part.
pstack is an array of PartComponent objects.
The state output offers a peek into the internal data structures of
the enumeration function. The client should treat this as read-only;
any modification of the data structure will cause unpredictable (and
almost certainly incorrect) results. Also, the components of state
are modified in place at each iteration. Hence, the visitor must be
called at each loop iteration. Accumulating the state instances and
processing them later will not work.
See also:
Examples
sympy.utilities.enumerative.factoring_visitor(state, primes)
Use with multiset_partitions_taocp to enumerate the ways a number can be expressed
as a product of factors. For this usage, the exponents of the prime factors of a number
are arguments to the partition enumerator, while the corresponding prime factors are
input here.
Examples
To enumerate the factorings of a number we can think of the elements of the partition
as being the prime factors and the multiplicities as being their exponents.
sympy.utilities.enumerative.list_visitor(state, components)
Return a list of lists to represent the partition.
Examples
References
Examples
count_partitions(multiplicities)
Returns the number of partitions of a multiset whose components have the multi-
plicities given in multiplicities.
For larger counts, this method is much faster than calling one of the enumerators
and counting the result. Uses dynamic programming to cut down on the number of
nodes actually explored. The dictionary used in order to accelerate the counting pro-
cess is stored in the MultisetPartitionTraverser object and persists across calls.
If the user does not expect to call count_partitions for any additional multisets, the
object should be cleared to save memory. On the other hand, the cache built up from
one count run can significantly speed up subsequent calls to count_partitions, so
it may be advantageous not to clear the object.
Notes
Examples
enum_all(multiplicities)
Enumerate the partitions of a multiset.
See also:
Examples
enum_large(multiplicities, lb)
Enumerate the partitions of a multiset with lb < num(parts)
Equivalent to enum_range(multiplicities, lb, sum(multiplicities))
Parameters multiplicities
list of multiplicities of the components of the multiset.
lb
Number of parts in the partition must be greater than this lower
bound.
See also:
enum_all (page 1454), enum_small (page 1456), enum_range (page 1455)
Examples
Examples
enum_small(multiplicities, ub)
Enumerate multiset partitions with no more than ub parts.
Equivalent to enum_range(multiplicities, 0, ub)
Parameters multiplicities
list of multiplicities of the components of the multiset.
ub
Maximum number of parts
See also:
enum_all (page 1454), enum_large (page 1455), enum_range (page 1455)
Examples
The implementation is based, in part, on the answer given to exercise 69, in Knuth
[AOCP].
5.37.5 Iterables
cartes
variations
partitions
Although the combinatorics module contains Partition and IntegerPartition classes for inves-
tigation and manipulation of partitions, there are a few functions to generate partitions that
can be used as low-level tools for routines: partitions and multiset_partitions. The for-
mer gives integer partitions, and the latter gives enumerated partitions of elements. There
is also a routine kbins that will give a variety of permutations of partions.
partitions:
multiset_partitions:
kbins:
Docstring
sympy.utilities.iterables.binary_partitions(n)
Generates the binary partition of n.
A binary partition consists only of numbers that are powers of two. Each step reduces a
2**(k+1) to 2**k and 2**k. Thus 16 is converted to 8 and 8.
Reference: TAOCP 4, section 7.2.1.5, problem 64
Examples
sympy.utilities.iterables.bracelets(n, k)
Wrapper to necklaces to return a free (unrestricted) necklace.
sympy.utilities.iterables.capture(func)
Return the printed output of func().
f unc should be a function without arguments that produces output with print statements.
sympy.utilities.iterables.common_prefix(*seqs)
Return the subsequence that is a common start of sequences in seqs.
sympy.utilities.iterables.common_suffix(*seqs)
Return the subsequence that is a common ending of sequences in seqs.
sympy.utilities.iterables.dict_merge(*dicts)
Merge dictionaries into a single dictionary.
sympy.utilities.iterables.filter_symbols(iterator, exclude)
Only yield elements from iterator that do not occur in exclude.
Parameters iterator : iterable
iterator to take elements from
exclude : iterable
elements to exclude
Returns iterator : iterator
filtered iterator
sympy.utilities.iterables.flatten(iterable, levels=None, cls=None)
Recursively denest iterable containers.
If you want to denest only a specified number of levels of nested containers, then set
levels flag to the desired number of levels:
If cls argument is specified, it will only flatten instances of that class, for example:
References
• http://en.wikipedia.org/wiki/Method_ringing
• http://stackoverflow.com/questions/4856615/recursive-permutation/4857018
• http://programminggeeks.com/bell-algorithm-for-permutation/
• http://en.wikipedia.org/wiki/Steinhaus%E2%80%93Johnson%E2%80%93Trotter_
algorithm
• Generating involutions, derangements, and relatives by ECO Vincent Vajnovszki,
DMTCS vol 1 issue 12, 2010
Examples
This is the sort of permutation used in the ringing of physical bells, and does not produce
permutations in lexicographical order. Rather, the permutations differ from each other
by exactly one inversion, and the position at which the swapping occurs varies period-
ically in a simple fashion. Consider the first few permutations of 4 elements generated
by permutations and generate_bell:
>>> list(permutations(range(4)))[:5]
[(0, 1, 2, 3), (0, 1, 3, 2), (0, 2, 1, 3), (0, 2, 3, 1), (0, 3, 1, 2)]
>>> list(generate_bell(4))[:5]
[(0, 1, 2, 3), (0, 1, 3, 2), (0, 3, 1, 2), (3, 0, 1, 2), (3, 0, 2, 1)]
Notice how the 2nd and 3rd lexicographical permutations have 3 elements out of place
whereas each “bell” permutation always has only two elements out of place relative to
the previous permutation (and so the signature (+/-1) of a permutation is opposite of the
signature of the previous permutation).
How the position of inversion varies across the elements can be seen by tracing out
where the largest number appears in the permutations:
sympy.utilities.iterables.generate_derangements(perm)
Routine to generate unique derangements.
TODO: This will be rewritten to use the ECO operator approach once the permutations
branch is in master.
See also:
sympy.functions.combinatorial.factorials.subfactorial (page 486)
Examples
sympy.utilities.iterables.generate_involutions(n)
Generates involutions.
An involution is a permutation that when multiplied by itself equals the identity permu-
tation. In this implementation the involutions are generated using Fixed Points.
Alternatively, an involution can be considered as a permutation that does not contain any
cycles with a length that is greater than two.
Reference: http://mathworld.wolfram.com/PermutationInvolution.html
Examples
sympy.utilities.iterables.generate_oriented_forest(n)
This algorithm generates oriented forests.
An oriented graph is a directed graph having no symmetric pair of directed edges. A
forest is an acyclic graph, i.e., it has no cycles. A forest can also be described as a
disjoint union of trees, which are graphs in which any two vertices are connected by
exactly one simple path.
Reference: [1] T. Beyer and S.M. Hedetniemi: constant time generation of rooted
trees, SIAM J. Computing Vol. 9, No. 4, November 1980 [2] http://stackoverflow.com/
questions/1633833/oriented-forest-taocp-algorithm-in-python
Examples
sympy.utilities.iterables.group(seq, multiple=True)
Splits a sequence into a list of lists of equal, adjacent elements.
See also:
multiset (page 1465)
Examples
sympy.utilities.iterables.has_dups(seq)
Return True if there are any duplicate elements in seq.
Examples
sympy.utilities.iterables.has_variety(seq)
Return True if there are any different elements in seq.
Examples
Examples
sympy.utilities.iterables.interactive_traversal(expr)
Traverse a tree asking a user which branch to choose.
sympy.utilities.iterables.kbins(l, k, ordered=None)
Return sequence l partitioned into k bins.
See also:
partitions (page 1470), multiset_partitions (page 1466)
Examples
The default is to give the items in the same order, but grouped into k partitions without
any reordering:
>>> from __future__ import print_function
>>> for p in kbins(list(range(5)), 2):
... print(p)
...
[[0], [1, 2, 3, 4]]
[[0, 1], [2, 3, 4]]
[[0, 1, 2], [3, 4]]
[[0, 1, 2, 3], [4]]
The ordered flag is either None (to give the simple partition of the elements) or is a 2
digit integer indicating whether the order of the bins and the order of the items in the
bins matters. Given:
A = [[0], [1, 2]]
B = [[1, 2], [0]]
C = [[2, 1], [0]]
D = [[0], [2, 1]]
Examples
sympy.utilities.iterables.multiset(seq)
Return the hashable sequence in multiset form with values being the multiplicity of the
item in the sequence.
See also:
group (page 1462)
Examples
sympy.utilities.iterables.multiset_combinations(m, n, g=None)
Return the unique combinations of size n from multiset m.
Examples
The number of combinations depends on the number of letters; the number of unique
combinations depends on how the letters are repeated.
>>> s1 = 'abracadabra'
>>> s2 = 'banana tree'
>>> count(combinations, s1), count(multiset_combinations, s1)
(165, 23)
>>> count(combinations, s2), count(multiset_combinations, s2)
(165, 54)
sympy.utilities.iterables.multiset_partitions(multiset, m=None)
Return unique partitions of the given multiset (in list form). If m is None, all multisets
will be returned, otherwise only partitions with m parts will be returned.
If multiset is an integer, a range [0, 1, …, multiset - 1] will be supplied.
See also:
partitions (page 1470), sympy.combinatorics.partitions.Partition (page 208),
sympy.combinatorics.partitions.IntegerPartition (page 210), sympy.functions.
combinatorial.numbers.nT
Notes
When all the elements are the same in the multiset, the order of the returned partitions
is determined by the partitions routine. If one is counting partitions then it is better
to use the nT function.
Examples
Only unique partitions are returned and these will be returned in a canonical order re-
gardless of the order of the input:
>>> a = [1, 2, 2, 1]
>>> ans = list(multiset_partitions(a, 2))
>>> a.sort()
>>> list(multiset_partitions(a, 2)) == ans
True
>>> a = range(3, 1, -1)
>>> (list(multiset_partitions(a)) ==
... list(multiset_partitions(sorted(a))))
True
Counting
The number of partitions of length k from a set of size n is given by the Stirling Number
of the 2nd kind:
Examples
sympy.utilities.iterables.necklaces(n, k, free=False)
A routine to generate necklaces that may (free=True) or may not (free=False) be turned
over to be viewed. The “necklaces” returned are comprised of n integers (beads) with k
different values (colors). Only unique necklaces are returned.
References
http://mathworld.wolfram.com/Necklace.html
Examples
The “unrestricted necklace” is sometimes also referred to as a “bracelet” (an object that
can be turned over, a sequence that can be reversed) and the term “necklace” is used to
imply a sequence that cannot be reversed. So ACB == ABC for a bracelet (rotate and
reverse) while the two are different for a necklace since rotation alone cannot make the
two sequences the same.
(mnemonic: Bracelets can be viewed Backwards, but Not Necklaces.)
References
[R597], [R598]
Examples
When m is given, a given list objects will be used more than once for speed reasons so
you will not see the correct partitions unless you make a copy of each as it is generated:
When n is a multiple of m, the elements are still sorted but the partitions themselves will
be unordered if sort is False; the default is to return them in ascending lexicographical
order.
>>> for p in ordered_partitions(6, 2):
... print(p)
[1, 5]
[2, 4]
[3, 3]
But if speed is more important than ordering, sort can be set to False:
>>> for p in ordered_partitions(6, 2, sort=False):
... print(p)
[1, 5]
[3, 3]
[2, 4]
Examples
The numbers appearing in the partition (the key of the returned dict) are limited with k:
>>> for p in partitions(6, k=2): # doctest: +SKIP
... print(p)
{2: 3}
(continues on next page)
The maximum number of parts in the partition (the sum of the values in the returned
dict) are limited with m (default value, None, gives partitions from 1 through n):
Note that the _same_ dictionary object is returned each time. This is for speed: gener-
ating each partition goes quickly, taking constant time, independent of n.
If you want to build a list of the returned dictionaries then make a copy of them:
Reference: modified from Tim Peter’s version to allow for k and m values:
code.activestate.com/recipes/218332-generator-for-integer-partitions/
sympy.utilities.iterables.permute_signs(t)
Return iterator in which the signs of non-zero elements of t are permuted.
Examples
sympy.utilities.iterables.postfixes(seq)
Generate all postfixes of a sequence.
Examples
>>> list(postfixes([1,2,3,4]))
[[4], [3, 4], [2, 3, 4], [1, 2, 3, 4]]
sympy.utilities.iterables.postorder_traversal(node, keys=None)
Do a postorder traversal of a tree.
This generator recursively yields nodes that it has visited in a postorder fashion. That
is, it descends through the tree depth-first to yield all of a node’s children’s postorder
traversal before yielding the node itself.
Parameters node : sympy expression
The expression to traverse.
keys : (default None) sort key(s)
The key(s) used to sort args of Basic objects. When None, args of
Basic objects are processed in arbitrary order. If key is defined, it
will be passed along to ordered() as the only key(s) to use to sort the
arguments; if key is simply True then the default keys of ordered will
be used (node count and default_sort_key).
Yields subtree : sympy expression
All of the subtrees in the tree.
Examples
The nodes are returned in the order that they are encountered unless key is given; simply
passing key=True will guarantee that the traversal is unique.
sympy.utilities.iterables.prefixes(seq)
Generate all prefixes of a sequence.
Examples
>>> list(prefixes([1,2,3,4]))
[[1], [1, 2], [1, 2, 3], [1, 2, 3, 4]]
sympy.utilities.iterables.reshape(seq, how)
Reshape the sequence according to the template in how.
Examples
sympy.utilities.iterables.rotate_left(x, y)
Left rotates a list x by the number of steps specified in y.
Examples
sympy.utilities.iterables.rotate_right(x, y)
Right rotates a list x by the number of steps specified in y.
Examples
Examples
Examples
sift() returns a defaultdict() object, so any key that has no matches will give [].
Sometimes you won’t know how many keys you will get:
Sometimes you expect the results to be binary; the results can be unpacked by setting
binary to True:
A ValueError is raised if the predicate was not actually binary (which is a good test for
the logic where sifting is used and binary results were expected):
If you need to sort the sifted items it might be better to use ordered which can econom-
ically apply multiple sort keys to a squence while sorting.
sympy.utilities.iterables.signed_permutations(t)
Return iterator in which the signs of non-zero elements of t and the order of the elements
are permuted.
Examples
Examples
subsets(seq, k) will return the n!/k!/(n - k)! k-subsets (combinations) without repetition,
i.e. once an item has been removed, it can no longer be “taken”:
If you ask for more items than are in the set you get the empty set unless you allow
repetitions:
sympy.utilities.iterables.take(iter, n)
Return n items from iter iterator.
sympy.utilities.iterables.topological_sort(graph, key=None)
Topological sort of graph’s vertices.
Parameters ‘‘graph‘‘ : tuple[list, list[tuple[T, T]]
A tuple consisting of a list of vertices and a list of edges of a graph to
be sorted topologically.
‘‘key‘‘ : callable[T] (optional)
Ordering key for vertices on the same level. By default the natural
(e.g. lexicographic) ordering is used (in this case the base type must
implement ordering relations).
Examples
Consider a graph:
where vertices are integers. This graph can be encoded using elementary Python’s data
structures as follows:
Only acyclic graphs can be sorted. If the input graph has a cycle, then ValueError will
be raised:
>>> topological_sort((V, E + [(10, 7)]))
Traceback (most recent call last):
...
ValueError: cycle detected
See also:
http://en.wikipedia.org/wiki/Topological_sorting
sympy.utilities.iterables.unflatten(iter, n=2)
Group iter into tuples of length n. Raise an error if the length of iter is not a multiple
of n.
sympy.utilities.iterables.uniq(seq, result=None)
Yield unique elements from seq as an iterator. The second parameter result is used
internally; it is not necessary to pass anything for this.
Examples
>>> list(uniq(dat))
[1, 4, 5, 2]
>>> list(uniq(x for x in dat))
[1, 4, 5, 2]
>>> list(uniq([[1], [2, 1], [1]]))
[[1], [2, 1]]
sympy.utilities.iterables.variations(seq, n, repetition=False)
Returns a generator of the n-sized variations of seq (size N). repetition controls
whether items in seq can appear more than once;
See also:
sympy.core.compatibility.permutations, sympy.core.compatibility.product
Examples
variations(seq, n, True) will return the N**n permutations obtained by allowing repeti-
tion of elements:
If you ask for more items than are in the set you get the empty set unless you allow
repetitions:
5.37.6 Lambdify
This module provides convenient functions to transform sympy expressions to lambda func-
tions which can be used to calculate numerical values very fast.
sympy.utilities.lambdify.implemented_function(symfunc, implementation)
Add numerical implementation to function symfunc.
symfunc can be an UndefinedFunction instance, or a name string. In the latter case we
create an UndefinedFunction instance with that name.
Be aware that this is a quick workaround, not a general method to create special symbolic
functions. If you want to create a symbolic function to be used by all the machinery of
SymPy you should subclass the Function class.
Parameters symfunc : str or UndefinedFunction instance
If str, then create new UndefinedFunction with this as name. If
symf unc is an Undefined function, create a new function with the
same name and the implemented function attached.
implementation : callable
numerical implementation to be called by evalf() or lambdify
Returns afunc : sympy.FunctionClass instance
function with attached implementation
Examples
Examples
Although tuples may not appear as arguments to lambda in Python 3, lambdastr will cre-
ate a lambda function that will unpack the original arguments so that nested arguments
can be handled:
Warning: Note that this function uses eval, and thus shouldn’t be used on unsani-
tized input.
Arguments in the provided expression that are not valid Python identifiers are substitued
with dummy symbols. This allows for applied functions (e.g. f(t)) to be supplied as
arguments. Call the function with dummify=True to replace all arguments with dummy
symbols (if args is not a string) - for example, to ensure that the arguments do not redefine
any built-in names.
For functions involving large array calculations, numexpr can provide a significant
speedup over numpy. Please note that the available functions for numexpr are more
limited than numpy but can be expanded with implemented_function and user defined
subclasses of Function. If specified, numexpr may be the only option in modules. The
official list of numexpr functions can be found at: https://github.com/pydata/numexpr#
supported-functions
Examples
Tuple arguments are handled and the lambdified function should be called with the same
type of arguments as were used to create the function.:
A more robust way of handling this is to always work with flattened arguments:
Functions present in expr can also carry their own numerical implementations, in a
callable attached to the _imp_ attribute. Usually you attach this using the imple-
mented_function factory:
Usage
Attention: Functions that are not in the math module will throw a name
error when the function definition is evaluated! So this would be better:
In the above examples, the generated functions can accept scalar values or numpy
arrays as arguments. However, in some cases the generated function relies on the
input being a numpy array:
5.37.7 Memoization
sympy.utilities.memoization.assoc_recurrence_memo(base_seq)
Memo decorator for associated sequences defined by recurrence starting from base
base_seq(n) – callable to get base sequence elements
XXX works only for Pn0 = base_seq(0) cases XXX works only for m <= n cases
sympy.utilities.memoization.recurrence_memo(initial)
Memo decorator for sequences defined by recurrence
See usage examples e.g. in the specfun/combinatorial module
5.37.8 Miscellaneous
sympy.utilities.misc.filldedent(s, w=70)
Strips leading and trailing empty lines from a copy of s, then dedents, fills and returns
it.
Empty line stripping serves to deal with docstrings like this one that start with a newline
after the initial triple quote, inserting an empty line at the beginning of the string.
sympy.utilities.misc.find_executable(executable, path=None)
Try to find ‘executable’ in the directories listed in ‘path’ (a string listing directories sep-
arated by ‘os.pathsep’; defaults to os.environ[‘PATH’]). Returns the complete filename
or None if not found
sympy.utilities.misc.func_name(x, short=False)
Return function name of x (if defined) else the type(x). If short is True and there is a
shorter alias for the result, return the alias.
See also:
sympy.core.compatibility (page 204)
Examples
sympy.utilities.misc.rawlines(s)
Return a cut-and-pastable string that, when printed, is equivalent to the input. The
string returned is formatted so it can be indented nicely within tests; in some cases it is
wrapped in the dedent function which has to be imported from textwrap.
Examples
Note: because there are characters in the examples below that need to be escaped be-
cause they are themselves within a triple quoted docstring, expressions below look more
complicated than they would be if they were printed in an interpreter window.
>>> print(rawlines('''this
... that
... '''))
dedent('''\
this
that
''')
>>> s = ”””this
... is a triple '''
... ”””
>>> print(rawlines(s))
dedent(”””\
this
is a triple '''
”””)
>>> print(rawlines('''this
... that
... '''))
(
'this\n'
'that\n'
' '
)
sympy.utilities.misc.replace(string, *reps)
Return string with all keys in reps replaced with their corresponding values, longer
strings first, irrespective of the order they are given. reps may be passed as tuples or a
single mapping.
References
[R599]
Examples
There is no guarantee that a unique answer will be obtained if keys in a mapping overlap
(i.e. are the same length and have some identical sequence at the beginning/end):
>>> reps = [
... ('ab', 'x'),
... ('bc', 'y')]
>>> replace('abc', *reps) in ('xc', 'ay')
True
Examples
There is no guarantee that a unique answer will be obtained if keys in a mapping overlap
are the same length and have some identical sequences at the beginning/end:
Syntax
5.37.9 PKGDATA
pkgdata is a simple, extensible way for a package to acquire data file resources.
The getResource function is equivalent to the standard idioms, such as the following minimal
implementation:
import sys, os
When a __loader__ is present on the module given by __name__, it will defer getResource to
its get_data implementation and return it as a file-like object (such as StringIO).
sympy.utilities.pkgdata.get_resource(identifier, pkgname=’sympy.utilities.pkgdata’)
Acquire a readable object for a given package name and identifier. An IOError will be
raised if the resource can not be found.
For example:
mydata = get_resource('mypkgdata.jpg').read()
Note that the package name must be fully qualified, if given, such that it would be found
in sys.modules.
In some cases, getResource will return a real file object. In that case, it may be useful to
use its name attribute to get the path rather than use it as a file-like object. For example,
you may be handing data off to a C API.
5.37.10 pytest
Examples
Note that you cannot test multiple statements via with raises:
This is just what with is supposed to do: abort the contained statement sequence at the
first exception and let the context manager deal with the exception.
Examples
Examples
collection. If you would like to examine the namespace after the test completes, then
use clear_globs=False.
compileflags gives the set of flags that should be used by the Python compiler when
running the examples. If not specified, then it will default to the set of future-import
flags that apply to globs.
The output of each example is checked using SymPyDocTestRunner.check_output,
and the results are formatted by the SymPyDocTestRunner.report_* methods.
class sympy.utilities.runtests.SymPyOutputChecker
Compared to the OutputChecker from the stdlib our OutputChecker class supports nu-
merical comparison of floats occurring in the output of the doctest examples
check_output(want, got, optionflags)
Return True iff the actual output from an example (got) matches the expected out-
put (want). These strings are always considered to match if they are identical; but
depending on what option flags the test runner is using, several non-exact match
types are also possible. See the documentation for T estRunner for more information
about option flags.
sympy.utilities.runtests.SymPyTestResults
alias of sympy.utilities.runtests.TestResults
sympy.utilities.runtests.convert_to_native_paths(lst)
Converts a list of ‘/’ separated paths into a list of native (os.sep separated) paths and
converts to lowercase if the system is case insensitive.
sympy.utilities.runtests.doctest(*paths, **kwargs)
Runs doctests in all *.py files in the sympy directory which match any of the given strings
in paths or all tests if paths=[].
Notes:
• Paths can be entered in native system format or in unix, forward-slash format.
• Files that are on the blacklist can be tested by providing their path; they are only
excluded if no paths are given.
Examples
The split option can be passed to split the test run into parts. The split currently only
splits the test files, though this may change in the future. split should be a string of
the form ‘a/b’, which will run part a of b. Note that the regular doctests and the Sphinx
doctests are split independently. For instance, to run the first half of the test suite:
The subprocess and verbose options are the same as with the function test(). See the
docstring of that function for more information.
sympy.utilities.runtests.get_sympy_dir()
Returns the root sympy directory and set the global value indicating whether the system
is case sensitive or not.
sympy.utilities.runtests.run_all_tests(test_args=(), test_kwargs=None,
doctest_args=(), doctest_kwargs=None,
examples_args=(), exam-
ples_kwargs=None)
Run all tests.
Right now, this runs the regular tests (bin/test), the doctests (bin/doctest), the examples
(examples/all.py), and the sage tests (see sympy/external/tests/test_sage.py).
This is what setup.py test uses.
You can pass arguments and keyword arguments to the test functions that support them
(for now, test, doctest, and the examples). See the docstrings of those functions for a
description of the available options.
For example, to run the solvers tests with colors turned off:
sympy.utilities.runtests.run_in_subprocess_with_hash_randomization(function,
func-
tion_args=(),
func-
tion_kwargs=None,
command=’/opt/conda/envs
release-
docs/bin/python’,
mod-
ule=’sympy.utilities.runtest
force=False)
Run a function in a Python subprocess with hash randomization enabled.
If hash randomization is not supported by the version of Python given, it returns False.
Otherwise, it returns the exit value of the command. The function is passed to sys.exit(),
so the return value of the function will be the return value.
The environment variable PYTHONHASHSEED is used to seed Python’s hash random-
ization. If it is set, this function will return False, because starting a new subprocess
is unnecessary in that case. If it is not set, one is set at random, and the tests are run.
Note that if this environment variable is set when Python starts, hash randomization is
Examples
sympy.utilities.runtests.sympytestfile(filename, module_relative=True,
name=None, package=None,
globs=None, verbose=None, re-
port=True, optionflags=0, extra-
globs=None, raise_on_error=False,
parser=<doctest.DocTestParser object>,
encoding=None)
Optional keyword arg encoding specifies an encoding that should be used to convert the
file to unicode.
Advanced tomfoolery: testmod runs methods of a local instance of class doctest.Tester,
then merges the results into (or creates) global Tester instance doctest.master. Methods
of doctest.master can be called directly too, if you want to do something unusual. Passing
report=0 to testmod is especially useful then, to delay displaying a summary. Invoke
doctest.master.summarize(verbose) when you’re done fiddling.
sympy.utilities.runtests.test(*paths, **kwargs)
Run tests in the specified test_*.py files.
Tests in a particular test_*.py file are run if any of the given strings in paths matches a
part of the test file’s path. If paths=[], tests in all test_*.py files are run.
Notes:
• If sort=False, tests are run in random order (not default).
• Paths can be entered in native system format or in unix, forward-slash format.
• Files that are on the blacklist can be tested by providing their path; they are only
excluded if no paths are given.
Explanation of test results
Out- Meaning
put
. passed
F failed
X XPassed (expected to fail but passed)
f XFAILed (expected to fail and indeed failed)
s skipped
w slow
T timeout (e.g., when --timeout is used)
K KeyboardInterrupt (when running the slow tests with --slow, you can interrupt
one of them without killing the test runner)
Colors have no additional meaning and are used just to facilitate interpreting the output.
Examples
>>> sympy.test(”sympy/core/tests/test_basic.py”,
... kw=”test_equality”) # doctest: +SKIP
Turn off colors:
Force colors, even when the output is not to a terminal (this is useful, e.g., if you are
piping to less -r and you still want colors)
The split option can be passed to split the test run into parts. The split currently only
splits the test files, though this may change in the future. split should be a string of the
form ‘a/b’, which will run part a of b. For instance, to run the first half of the test suite:
You can disable running the tests in a separate subprocess using subprocess=False.
This is done to support seeding hash randomization, which is enabled by default in the
Python versions where it is supported. If subprocess=False, hash randomization is en-
abled/disabled according to whether it has been enabled or not in the calling Python
process. However, even if it is enabled, the seed cannot be printed unless it is called
from a new Python process.
Hash randomization was added in the minor Python versions 2.6.8, 2.7.3, 3.1.5, and
3.2.3, and is enabled by default in all Python versions after and including 3.3.0.
If hash randomization is not supported subprocess=False is used automatically.
To set the hash randomization seed, set the environment variable PYTHONHASHSEED before
running the tests. This can be done from within Python using
>>> import os
>>> os.environ['PYTHONHASHSEED'] = '42' # doctest: +SKIP
This module adds several functions for interactive source code inspection.
sympy.utilities.source.get_class(lookup_view)
Convert a string version of a class name to the object.
For example, get_class(‘sympy.core.Basic’) will return class Basic located in module
sympy.core
sympy.utilities.source.get_mod_func(callback)
splits the string path to a class into a string path to the module and the name of the class.
For example:
sympy.utilities.source.source(object)
Prints the source code of a given object.
Simple tools for timing functions’ execution, when IPython is not available.
sympy.utilities.timeutils.timed(func, setup=’pass’, limit=None)
Adaptively measure execution time of a function.
Examples
In addition the order of the arguments will not be made canonical. This feature allows
one to tell exactly how the expression was entered:
sympy.parsing.sympy_parser.split_symbols_custom(predicate)
Creates a transformation that splits symbol names.
predicate should return True if the symbol name is to be split.
For instance, to retain the default behavior but avoid splitting certain symbol names, a
predicate like this would work:
sympy.parsing.sympy_parser.implicit_multiplication(result, local_dict,
global_dict)
Makes the multiplication operator optional in most cases.
Use this before implicit_application() (page 1498), otherwise expressions like sin
2x will be parsed as x * sin(2) rather than sin(2*x).
Examples
Examples
sympy.parsing.sympy_parser.function_exponentiation(tokens, local_dict,
global_dict)
Allows functions to be exponentiated, e.g. cos**2(x).
Examples
sympy.parsing.sympy_parser.implicit_multiplication_application(result, lo-
cal_dict,
global_dict)
Allows a slightly relaxed syntax.
• Parentheses for single-argument method calls are optional.
• Multiplication is implicit.
• Symbol names can be split (i.e. spaces are not needed between symbols).
• Functions can be exponentiated.
Examples
LATEX parsing was ported from latex2sympy. While functional and its API should remain stable,
the parsing behavior or backend may change in future releases.
The current implementation is experimental. The behavior, parser backend and API might
change in the future. Unlike some of the other parsers, LATEX is designed as a type-setting
language, not a computer algebra system and so can contain typographical conventions that
might be interpreted multiple ways.
In its current definition, the parser will at times will fail to fully parse the expression, but not
throw a warning:
parse_latex(r'x -')
Will simply find x. What is covered by this behavior will almost certainly change between
releases, and become stricter, more relaxed, or some mix.
sympy.parsing.latex.parse_latex(s)
Converts the string s to a SymPy Expr
Parameters s : str
The LaTeX string to parse. In Python source containing LaTeX, *raw
strings*
(denoted with ‘r’‘‘, like this one) are preferred, as LaTeX makes lib-
eral
use of the ‘‘‘‘ character, which would trigger escaping in normal
Python
strings.
>>> from sympy.parsing.latex import parse_latex # doctest: +SKIP
>>> expr = parse_latex(r”frac {1 + sqrt {a}} {b}”) # doctest: +SKIP
>>> expr # doctest: +SKIP
(sqrt(a) + 1)/b
>>> expr.evalf(4, subs=dict(a=5, b=2)) # doctest: +SKIP
1.618
class sympy.parsing.latex.LaTeXParsingError
The currently-packaged parser backend is partially generated with ANTLR4, but to use the
parser, you only need the antlr4 Python package available.
Depending on your package manager, you can install the right package with, for example,
pip3 (Python 3 only):
5.39 Calculus
Calculus-related methods.
This module implements a method to find Euler-Lagrange Equations for given Lagrangian.
sympy.calculus.euler.euler_equations(L, funcs=(), vars=())
Find the Euler-Lagrange equations [R22] for a given Lagrangian.
Parameters L : Expr
The Lagrangian that should be a function of the functions listed in
the second argument and their derivatives.
For example, in the case of two functions f (x, y), g(x, y) and two inde-
pendent variables x, y the Lagrangian would have the form:
( )
∂f (x, y) ∂f (x, y) ∂g(x, y) ∂g(x, y)
L f (x, y), g(x, y), , , , , x, y
∂x ∂y ∂x ∂y
References
[R22]
Examples
5.39.1 Singularities
This module implements algorithms for finding singularities for a function and identifying
types of functions.
The differential calculus methods in this module include methods to identify the following
function types in the given Interval: - Increasing - Strictly Increasing - Decreasing - Strictly
Decreasing - Monotonic
sympy.calculus.singularities.is_decreasing(expression, interval=Reals, sym-
bol=None)
Return whether the function is decreasing in the given interval.
Examples
Examples
Examples
sympy.calculus.singularities.is_strictly_decreasing(expression, inter-
val=Reals, sym-
bol=None)
Return whether the function is strictly decreasing in the given interval.
Examples
sympy.calculus.singularities.is_strictly_increasing(expression, inter-
val=Reals, sym-
bol=None)
Return whether the function is strictly increasing in the given interval.
Examples
Notes
This function does not find nonisolated singularities nor does it find branch points of the
expression.
References
[R23]
Examples
This module implements an algorithm for efficient generation of finite difference weights for
ordinary differentials of functions for derivatives from 0 (interpolation) up to arbitrary order.
The core algorithm is provided in the finite difference weight generating function
(finite_diff_weights), and two convenience functions are provided for:
• estimating a derivative (or interpolate) directly from a series of points is also
provided (apply_finite_diff).
• differentiating by using finite difference approximations
(differentiate_finite).
sympy.calculus.finite_diff.apply_finite_diff(order, x_list, y_list, x0=0)
Calculates the finite difference approximation of the derivative of requested order at x0
from points provided in x_list and y_list.
Parameters order: int
order of derivative to approximate. 0 corresponds to interpolation.
x_list: sequence
Sequence of (unique) values for the independent variable.
y_list: sequence
The function value at corresponding values for the independent vari-
able in x_list.
x0: Number or Symbol
At what value of the independent variable the derivative should be
evaluated. Defaults to S(0).
Returns sympy.core.add.Add or sympy.core.numbers.Number
The finite difference expression approximating the requested deriva-
tive order at x0.
See also:
sympy.calculus.finite_diff.finite_diff_weights (page 1508)
Notes
Order = 0 corresponds to interpolation. Only supply so many points you think makes
sense to around x0 when extracting the derivative (the function need to be well behaved
within that region). Also beware of Runge’s phenomenon.
References
Examples
we see that the example above only contain rounding errors. apply_finite_diff can also
be used on more abstract objects:
Examples
>>> from sympy import symbols, Function, exp, sqrt, Symbol, as_finite_diff
>>> from sympy.utilities.exceptions import SymPyDeprecationWarning
>>> import warnings
>>> warnings.simplefilter(”ignore”, SymPyDeprecationWarning)
>>> x, h = symbols('x h')
>>> f = Function('f')
>>> as_finite_diff(f(x).diff(x))
-f(x - 1/2) + f(x + 1/2)
The default step size and number of points are 1 and order + 1 respectively. We can
change the step size by passing a symbol as a parameter:
>>> as_finite_diff(f(x).diff(x), h)
-f(-h/2 + x)/h + f(h/2 + x)/h
The algorithm is not restricted to use equidistant spacing, nor do we need to make the
approximation around x0, but we can get an expression estimating the derivative at an
offset:
>>> y = Symbol('y')
>>> d2fdxdy=f(x,y).diff(x,y)
>>> as_finite_diff(d2fdxdy, wrt=x)
-Derivative(f(x - 1/2, y), y) + Derivative(f(x + 1/2, y), y)
Examples
Note that the above form preserves the product rule in discrete form. If we want we can
pass evaluate=True to get another form (which is usually not what we want):
Notes
If weights for a finite difference approximation of 3rd order derivative is wanted, weights
for 0th, 1st and 2nd order are calculated “for free”, so are formulae using subsets of
x_list. This is something one can take advantage of to save computational cost. Be
aware that one should define x_list from nearest to farest from x0. If not, subsets of
x_list will yield poorer approximations, which might not grand an order of accuracy of
len(x_list) - order.
References
[R24]
Examples
Each sublist contains the most accurate formula at the end. Note, that in the above
example res[1][1] is the same as res[1][2]. Since res[1][2] has an order of accuracy
of len(x_list[:3]) - order = 3 - 1 = 2, the same is true for res[1][1]!
>>> from sympy import S
>>> from sympy.calculus import finite_diff_weights
>>> res = finite_diff_weights(1, [S(0), S(1), -S(1), S(2), -S(2)], 0)[1]
>>> res
[[0, 0, 0, 0, 0],
[-1, 1, 0, 0, 0],
[0, 1/2, -1/2, 0, 0],
[-1/2, 1, -1/3, -1/6, 0],
[0, 2/3, -2/3, -1/12, 1/12]]
>>> res[0] # no approximation possible, using x_list[0] only
[0, 0, 0, 0, 0]
>>> res[1] # classic forward step approximation
(continues on next page)
Let us compare this to a differently defined x_list. Pay attention to foo[i][k] corre-
sponding to the gridpoint defined by x_list[k].
Note that, unless you plan on using approximations based on subsets of x_list, the
order of gridpoints does not matter.
The capability to generate weights at arbitrary points can be used e.g. to minimize
Runge’s phenomenon by using Chebyshev nodes:
5.40.1 Contents
Hydrogen Wavefunctions
sympy.physics.hydrogen.E_nl(n, Z=1)
Returns the energy of the state (n, l) in Hartree atomic units.
The energy doesn’t depend on “l”.
Examples
Examples
>>> E_nl_dirac(2, 0)
-0.125002080189006
>>> E_nl_dirac(2, 1)
-0.125000416028342
>>> E_nl_dirac(2, 1, False)
-0.125002080189006
>>> E_nl_dirac(3, 0)
-0.0555562951740285
>>> E_nl_dirac(3, 1)
(continues on next page)
Examples
Integrating the absolute square of a hydrogen wavefunction psi_{nlm} over the whole
space leads 1.
The normalization of the hydrogen wavefunctions Psi_nlm is:
sympy.physics.hydrogen.R_nl(n, l, r, Z=1)
Returns the Hydrogen radial wavefunction R_{nl}.
n, l quantum numbers ‘n’ and ‘l’
r radial coordinate
Z atomic number (1 for Hydrogen, 2 for Helium, …)
Everything is in Hartree atomic units.
Examples
For Hydrogen atom, you can just use the default value of Z=1:
>>> R_nl(1, 0, r)
2*exp(-r)
>>> R_nl(2, 0, r)
sqrt(2)*(-r + 2)*exp(-r/2)/4
>>> R_nl(3, 0, r)
2*sqrt(3)*(2*r**2/9 - 2*r + 3)*exp(-r/3)/27
Matrices
References
[R462]
Examples
sympy.physics.matrices.mgamma(mu, lower=False)
Returns a Dirac gamma matrix γ µ in the standard (Dirac) representation.
If you want γµ , use gamma(mu, True).
We use a convention:
γ5 = i · γ0 · γ1 · γ2 · γ3
γ5 = i · γ0 · γ1 · γ2 · γ3 = −γ 5
References
[R463]
Examples
sympy.physics.matrices.msigma(i)
Returns a Pauli matrix σi with i = 1, 2, 3
References
[R464]
Examples
Examples
To translate a body having a mass of 2 units a distance of 1 unit along the x-axis we get:
Pauli Algebra
This module implements Pauli algebra by subclassing Symbol. Only algebraic properties of
Pauli matrices are used (we don’t use the Matrix class).
See the documentation to the class Pauli for examples.
References
sympy.physics.paulialgebra.evaluate_pauli_product(arg)
Help function to evaluate Pauli matrices product with symbolic objects
Parameters arg: symbolic expression that contains Paulimatrices
Examples
sympy.physics.qho_1d.E_n(n, omega)
Returns the Energy of the One-dimensional harmonic oscillator
n the “nodal” quantum number
omega the harmonic oscillator angular frequency
The unit of the returned value matches the unit of hw, since the energy is calculated as:
E_n = hbar * omega*(n + 1/2)
Examples
sympy.physics.qho_1d.coherent_state(n, alpha)
Returns <n|alpha> for the coherent states of 1D harmonic oscillator. See http://en.
wikipedia.org/wiki/Coherent_states
n the “nodal” quantum number
alpha the eigen value of annihilation operator
sympy.physics.qho_1d.psi_n(n, x, m, omega)
Returns the wavefunction psi_{n} for the One-dimensional harmonic oscillator.
n the “nodal” quantum number. Corresponds to the number of nodes in the wavefunc-
tion. n >= 0
x x coordinate
m mass of the particle
omega angular frequency of the oscillator
Examples
sympy.physics.sho.E_nl(n, l, hw)
Returns the Energy of an isotropic harmonic oscillator
n the “nodal” quantum number
Examples
sympy.physics.sho.R_nl(n, l, nu, r)
Returns the radial wavefunction R_{nl} for a 3d isotropic harmonic oscillator.
n the “nodal” quantum number. Corresponds to the number of nodes in the wavefunc-
tion. n >= 0
l the quantum number for orbital angular momentum
nu mass-scaled frequency: nu = m*omega/(2*hbar) where m is the mass and omega the
frequency of the oscillator. (in atomic units nu == omega/2)
r Radial coordinate
Examples
Second Quantization
Examples
classmethod eval(arg)
Evaluates the Dagger instance.
Examples
References
[R487]
Examples
Symbolic indices:
classmethod eval(i, j)
Evaluates the discrete delta function.
Examples
>>> KroneckerDelta(i, j)
KroneckerDelta(i, j)
>>> KroneckerDelta(i, i)
1
>>> KroneckerDelta(i, i + 1)
0
>>> KroneckerDelta(i, i + 1 + k)
KroneckerDelta(i, i + k + 1)
# indirect doctest
indices_contain_equal_information
Returns True if indices are either both above or below fermi.
Examples
is_above_fermi
True if Delta can be non-zero above fermi
See also:
is_below_fermi (page 1520), is_only_below_fermi (page 1521),
is_only_above_fermi (page 1521)
Examples
is_below_fermi
True if Delta can be non-zero below fermi
See also:
is_above_fermi (page 1520), is_only_above_fermi (page 1521),
is_only_below_fermi (page 1521)
Examples
is_only_above_fermi
True if Delta is restricted to above fermi
See also:
is_above_fermi (page 1520), is_below_fermi (page 1520), is_only_below_fermi
(page 1521)
Examples
is_only_below_fermi
True if Delta is restricted to below fermi
See also:
is_above_fermi (page 1520), is_below_fermi (page 1520), is_only_above_fermi
(page 1521)
Examples
killable_index
Returns the index which is preferred to substitute in the final expression.
The index to substitute is the index with less information regarding fermi level. If
indices contain same information, ‘a’ is preferred before ‘b’.
See also:
preferred_index (page 1522)
Examples
preferred_index
Returns the index which is preferred to keep in the final expression.
The preferred index is the index with more information regarding fermi level. If
indices contain same information, ‘a’ is preferred before ‘b’.
See also:
killable_index (page 1521)
Examples
class sympy.physics.secondquant.AnnihilateBoson
Bosonic annihilation operator.
Examples
apply_operator(state)
Apply state to self if self is not symbolic and state is a FockStateKet, else multiply
self by state.
Examples
class sympy.physics.secondquant.CreateBoson
Bosonic creation operator.
apply_operator(state)
Apply state to self if self is not symbolic and state is a FockStateKet, else multiply
self by state.
Examples
class sympy.physics.secondquant.AnnihilateFermion
Fermionic annihilation operator.
apply_operator(state)
Apply state to self if self is not symbolic and state is a FockStateKet, else multiply
self by state.
Examples
is_only_q_annihilator
Always destroy a quasi-particle? (annihilate hole or annihilate particle)
>>> from sympy import Symbol
>>> from sympy.physics.secondquant import F
>>> a = Symbol('a', above_fermi=True)
>>> i = Symbol('i', below_fermi=True)
>>> p = Symbol('p')
>>> F(a).is_only_q_annihilator
True
>>> F(i).is_only_q_annihilator
False
>>> F(p).is_only_q_annihilator
False
is_only_q_creator
Always create a quasi-particle? (create hole or create particle)
>>> F(a).is_only_q_creator
False
>>> F(i).is_only_q_creator
True
>>> F(p).is_only_q_creator
False
is_q_annihilator
Can we destroy a quasi-particle? (annihilate hole or annihilate particle) If so, would
that be above or below the fermi surface?
>>> F(a).is_q_annihilator
1
>>> F(i).is_q_annihilator
0
>>> F(p).is_q_annihilator
1
is_q_creator
Can we create a quasi-particle? (create hole or create particle) If so, would that be
above or below the fermi surface?
>>> F(a).is_q_creator
0
>>> F(i).is_q_creator
-1
>>> F(p).is_q_creator
-1
class sympy.physics.secondquant.CreateFermion
Fermionic creation operator.
apply_operator(state)
Apply state to self if self is not symbolic and state is a FockStateKet, else multiply
self by state.
Examples
is_only_q_annihilator
Always destroy a quasi-particle? (annihilate hole or annihilate particle)
>>> from sympy import Symbol
>>> from sympy.physics.secondquant import Fd
>>> a = Symbol('a', above_fermi=True)
>>> i = Symbol('i', below_fermi=True)
>>> p = Symbol('p')
>>> Fd(a).is_only_q_annihilator
False
>>> Fd(i).is_only_q_annihilator
True
>>> Fd(p).is_only_q_annihilator
False
is_only_q_creator
Always create a quasi-particle? (create hole or create particle)
>>> from sympy import Symbol
>>> from sympy.physics.secondquant import Fd
>>> a = Symbol('a', above_fermi=True)
>>> i = Symbol('i', below_fermi=True)
>>> p = Symbol('p')
>>> Fd(a).is_only_q_creator
True
>>> Fd(i).is_only_q_creator
False
>>> Fd(p).is_only_q_creator
False
is_q_annihilator
Can we destroy a quasi-particle? (annihilate hole or annihilate particle) If so, would
that be above or below the fermi surface?
>>> from sympy import Symbol
>>> from sympy.physics.secondquant import Fd
>>> a = Symbol('a', above_fermi=1)
(continues on next page)
>>> Fd(a).is_q_annihilator
0
>>> Fd(i).is_q_annihilator
-1
>>> Fd(p).is_q_annihilator
-1
is_q_creator
Can we create a quasi-particle? (create hole or create particle) If so, would that be
above or below the fermi surface?
>>> Fd(a).is_q_creator
1
>>> Fd(i).is_q_creator
0
>>> Fd(p).is_q_creator
1
class sympy.physics.secondquant.FockState
Many particle Fock state with a sequence of occupation numbers.
Anywhere you can have a FockState, you can also have S.Zero. All code must check for
this!
Base class to represent FockStates.
class sympy.physics.secondquant.FockStateBra
Representation of a bra.
class sympy.physics.secondquant.FockStateKet
Representation of a ket.
class sympy.physics.secondquant.FockStateBosonKet
Many particle Fock state with a sequence of occupation numbers.
Occupation numbers can be any integer >= 0.
Examples
class sympy.physics.secondquant.FockStateBosonBra
Describes a collection of BosonBra particles.
Examples
sympy.physics.secondquant.BBra
alias of sympy.physics.secondquant.FockStateBosonBra (page 1526)
sympy.physics.secondquant.BKet
alias of sympy.physics.secondquant.FockStateBosonKet (page 1526)
sympy.physics.secondquant.FBra
alias of sympy.physics.secondquant.FockStateFermionBra
sympy.physics.secondquant.FKet
alias of sympy.physics.secondquant.FockStateFermionKet
sympy.physics.secondquant.F
alias of sympy.physics.secondquant.AnnihilateFermion (page 1523)
sympy.physics.secondquant.Fd
alias of sympy.physics.secondquant.CreateFermion (page 1524)
sympy.physics.secondquant.B
alias of sympy.physics.secondquant.AnnihilateBoson (page 1522)
sympy.physics.secondquant.Bd
alias of sympy.physics.secondquant.CreateBoson (page 1523)
sympy.physics.secondquant.apply_operators(e)
Take a sympy expression with operators and states and apply the operators.
Examples
class sympy.physics.secondquant.InnerProduct
An unevaluated inner product between a bra and ket.
Currently this class just reduces things to a product of Kronecker Deltas. In the future,
we could introduce abstract states like |a> and |b>, and leave the inner product uneval-
uated as <a|b>.
bra
Returns the bra part of the state
ket
Returns the ket part of the state
class sympy.physics.secondquant.BosonicBasis
Base class for a basis set of bosonic Fock states.
class sympy.physics.secondquant.VarBosonicBasis(n_max)
A single state, variable particle number basis set.
Examples
index(state)
Returns the index of state in basis.
Examples
state(i)
The state of a single basis.
Examples
Examples
index(state)
Returns the index of state in basis.
Examples
state(i)
Returns the state that lies at index i of the basis
Examples
class sympy.physics.secondquant.Commutator
The Commutator: [A, B] = A*B - B*A
The arguments are ordered according to .__cmp__()
>>> Commutator(Fd(a),Fd(i))
2*NO(CreateFermion(a)*CreateFermion(i))
But for more complicated expressions, the evaluation is triggered by a call to .doit()
doit(**hints)
Enables the computation of complex expressions.
Examples
classmethod eval(a, b)
The Commutator [A,B] is on canonical form if A < B.
Examples
sympy.physics.secondquant.matrix_rep(op, basis)
Find the representation of an operator in a basis.
Examples
sympy.physics.secondquant.contraction(a, b)
Calculates contraction of Fermionic operators a and b.
Examples
>>> contraction(F(a),Fd(b))
KroneckerDelta(a, b)
>>> contraction(Fd(i),F(j))
KroneckerDelta(i, j)
For general indices a non-zero result restricts the indices to below/above the fermi sur-
face:
>>> contraction(Fd(p),F(q))
KroneckerDelta(_i, q)*KroneckerDelta(p, q)
>>> contraction(F(p),Fd(q))
KroneckerDelta(_a, q)*KroneckerDelta(p, q)
>>> contraction(F(p),F(q))
0
>>> contraction(Fd(p),Fd(q))
0
sympy.physics.secondquant.wicks(e, **kw_args)
Returns the normal ordered equivalent of an expression using Wicks Theorem.
Examples
class sympy.physics.secondquant.NO
This Object is used to represent normal ordering brackets.
i.e. {abcd} sometimes written :abcd:
Applying the function NO(arg) to an argument means that all operators in the argument
will be assumed to anticommute, and have vanishing contractions. This allows an imme-
diate reordering to canonical form upon object creation.
Note: If you want to generate a normal ordered equivalent of an expression, you should
use the function wicks(). This class only indicates that all operators inside the brackets
anticommute, and have vanishing contractions. Nothing more, nothing less.
doit(**kw_args)
Either removes the brackets or enables complex computations in its arguments.
Examples
get_subNO(i)
Returns a NO() without FermionicOperator at index i.
Examples
has_q_annihilators
Return 0 if the rightmost argument of the first argument is a not a q_annihilator,
else 1 if it is above fermi or -1 if it is below fermi.
Examples
has_q_creators
Return 0 if the leftmost argument of the first argument is a not a q_creator, else 1 if
it is above fermi or -1 if it is below fermi.
Examples
iter_q_annihilators()
Iterates over the annihilation operators.
Examples
>>> no.iter_q_creators()
<generator object... at 0x...>
>>> list(no.iter_q_creators())
[0, 1]
>>> list(no.iter_q_annihilators())
[3, 2]
iter_q_creators()
Iterates over the creation operators.
Examples
>>> no.iter_q_creators()
<generator object... at 0x...>
>>> list(no.iter_q_creators())
[0, 1]
>>> list(no.iter_q_annihilators())
[3, 2]
sympy.physics.secondquant.evaluate_deltas(e)
We evaluate KroneckerDelta symbols in the expression assuming Einstein summation.
If one index is repeated it is summed over and in effect substituted with the other one.
If both indices are repeated we substitute according to what is the preferred index. this
is determined by KroneckerDelta.preferred_index and KroneckerDelta.killable_index.
In case there are no possible substitutions or if a substitution would imply a loss of
information, nothing is done.
In case an index appears in more than one KroneckerDelta, the resulting substitution
depends on the order of the factors. Since the ordering is platform dependent, the literal
expression resulting from this function may be hard to predict.
Examples
The order of preference for these indices according to KroneckerDelta is (a, b, i, j, p, q).
Trivial cases:
>>> evaluate_deltas(KroneckerDelta(i,p)*t(a,i)*f(p,q))
f(_i, _q)*t(_a, _i)
>>> evaluate_deltas(KroneckerDelta(a,p)*t(a,i)*f(p,q))
f(_a, _q)*t(_a, _i)
>>> evaluate_deltas(KroneckerDelta(p,q)*f(p,q))
f(_p, _p)
Finally, here are some cases where nothing is done, because that would imply a loss of
information:
>>> evaluate_deltas(KroneckerDelta(i,p)*f(q))
f(_q)*KroneckerDelta(_i, _p)
>>> evaluate_deltas(KroneckerDelta(i,p)*f(i))
f(_i)*KroneckerDelta(_i, _p)
class sympy.physics.secondquant.AntiSymmetricTensor
Stores upper and lower indices in separate Tuple’s.
Each group of indices is assumed to be antisymmetric.
Examples
As you can see, the indices are automatically sorted to a canonical form.
doit(**kw_args)
Returns self.
Examples
lower
Returns the lower indices.
Examples
symbol
Returns the symbol of the tensor.
Examples
upper
Returns the upper indices.
Examples
sympy.physics.secondquant.substitute_dummies(expr, new_indices=False,
pretty_indices={})
Collect terms by substitution of dummy variables.
This routine allows simplification of Add expressions containing terms which differ only
due to dummy variables.
The idea is to substitute all dummy variables consistently depending on the structure of
the term. For each term, we obtain a sequence of all dummy variables, where the order is
determined by the index range, what factors the index belongs to and its position in each
factor. See _get_ordered_dummies() for more inforation about the sorting of dummies.
The index sequence is then substituted consistently in each term.
Examples
Since a, b, c and d are equivalent summation indices, the expression can be simplified
to a single term (for which the dummy indices are still summed over)
>>> substitute_dummies(expr)
2*f(_a, _b)
Controlling output:
By default the dummy symbols that are already present in the expression will be reused in
a different permutation. However, if new_indices=True, new dummies will be generated
and inserted. The keyword ‘pretty_indices’ can be used to control this generation of new
symbols.
By default the new dummies will be generated on the form i_1, i_2, a_1, etc. If you supply
a dictionary with key:value pairs in the form:
{ index_group: string_of_letters }
The letters will be used as labels for the new dummy symbols. The index_groups must
be one of ‘above’, ‘below’ or ‘general’.
>>> expr = f(a,b,i,j)
>>> my_dummies = { 'above':'st', 'below':'uv' }
>>> substitute_dummies(expr, new_indices=True, pretty_indices=my_dummies)
f(_s, _t, _u, _v)
If we run out of letters, or if there is no keyword for some index_group the default dummy
generator will be used as a fallback:
>>> p,q = symbols('p q', cls=Dummy) # general indices
>>> expr = f(p,q)
>>> substitute_dummies(expr, new_indices=True, pretty_indices=my_dummies)
f(_p_0, _p_1)
class sympy.physics.secondquant.PermutationOperator
Represents the index permutation operator P(ij).
P(ij)*f(i)*g(j) = f(i)*g(j) - f(j)*g(i)
get_permuted(expr)
Returns -expr with permuted indices.
>>> from sympy import symbols, Function
>>> from sympy.physics.secondquant import PermutationOperator
>>> p,q = symbols('p,q')
>>> f = Function('f')
>>> PermutationOperator(p,q).get_permuted(f(p,q))
-f(q, p)
sympy.physics.secondquant.simplify_index_permutations(expr, permuta-
tion_operators)
Performs simplification by introducing PermutationOperators where appropriate.
Schematically: [abij] - [abji] - [baij] + [baji] -> P(ab)*P(ij)*[abij]
permutation_operators is a list of PermutationOperators to consider.
If permutation_operators=[P(ab),P(ij)] we will try to introduce the permutation operators
P(ij) and P(ab) in the expression. If there are other possible simplifications, we ignore
them.
Wigner Symbols
References
This code was taken from Sage with the permission of all authors:
https://groups.google.com/forum/#!topic/sage-devel/M4NZdu-7O38
AUTHORS:
• Jens Rasch (2009-03-24): initial version for Sage
• Jens Rasch (2009-05-31): updated to sage-4.0
Copyright (C) 2008 Jens Rasch <[email protected]>
sympy.physics.wigner.clebsch_gordan(j_1, j_2, j_3, m_1, m_2, m_3)
Calculates the Clebsch-Gordan coefficient ⟨j1 m1 j2 m2 |j3 m3 ⟩.
NOTES:
The Clebsch-Gordan coefficient will be evaluated via its relation to Wigner 3j symbols:
√
⟨j1 m1 j2 m2 |j3 m3 ⟩ = (−1)j1 −j2 +m3 2j3 + 1 Wigner3j(j1 , j2 , j3 , m1 , m2 , −m3 )
See also the documentation on Wigner 3j symbols which exhibit much higher symmetry
relations than the Clebsch-Gordan coefficient.
AUTHORS:
• Jens Rasch (2009-03-24): initial version
sympy.physics.wigner.dot_rot_grad_Ynm(j, p, l, m, theta, phi)
Returns dot product of rotational gradients of spherical harmonics.
This function returns the right hand side of the following expression:
∑
l+j
1
⃗ p · RY
RY ⃗ ml = (−1)
m+p
Y m+p ∗ αl,m,j,p,k ∗ (k 2 − j 2 − l2 + k − j − l)
j k
2
k=|l−j|
Arguments
Example
Examples
sage: gaunt(1.2,0,1.2,0,0,0)
Traceback (most recent call last):
...
ValueError: l values must be integer
sage: gaunt(1,0,1,1.1,0,-1.1)
Traceback (most recent call last):
...
ValueError: m values must be integer
NOTES:
The Gaunt coefficient obeys the following symmetry rules:
• invariant under any permutation of the columns
Y (l1 , l2 , l3 , m1 , m2 , m3 ) = Y (l3 , l1 , l2 , m3 , m1 , m2 )
= Y (l2 , l3 , l1 , m2 , m3 , m1 )
= Y (l3 , l2 , l1 , m3 , m2 , m1 )
= Y (l1 , l3 , l2 , m1 , m3 , m2 )
= Y (l2 , l1 , l3 , m2 , m1 , m3 )
• symmetric with respect to the 72 Regge symmetries as inherited for the 3j symbols
[Regge58]
• zero for l1 , l2 , l3 not fulfilling triangle relation
• zero for violating any one of the conditions: l1 ≥ |m1 |, l2 ≥ |m2 |, l3 ≥ |m3 |
• non-zero only for an even sum of the li , i.e. L = l1 + l2 + l3 = 2n for n in N
ALGORITHM:
This function uses the algorithm of [Liberatodebrito82] to calculate the value of the
Gaunt coefficient exactly. Note that the formula contains alternating sums over large
factorials and is therefore unsuitable for finite precision arithmetic and only useful for a
computer algebra system [Rasch03].
REFERENCES:
AUTHORS:
• Jens Rasch (2009-03-24): initial version for Sage
sympy.physics.wigner.racah(aa, bb, cc, dd, ee, ff, prec=None)
Calculate the Racah symbol W (a, b, c, d; e, f ).
INPUT:
• a, …, f - integer or half integer
• prec - precision, default: None. Providing a precision can drastically speed up the
calculation.
OUTPUT:
Rational number times the square root of a rational number (if prec=None), or real num-
ber if a precision is given.
Examples
NOTES:
The Racah symbol is related to the Wigner 6j symbol:
Please see the 6j symbol for its much richer symmetries and for additional properties.
ALGORITHM:
This function uses the algorithm of [Edmonds74] to calculate the value of the 6j symbol
exactly. Note that the formula contains alternating sums over large factorials and is
therefore unsuitable for finite precision arithmetic and only useful for a computer alge-
bra system [Rasch03].
AUTHORS:
• Jens Rasch (2009-03-24): initial version
Examples
It is an error to have arguments that are not integer or half integer values:
sage: wigner_3j(2.1, 6, 4, 0, 0, 0)
Traceback (most recent call last):
...
ValueError: j values must be integer or half integer
sage: wigner_3j(2, 6, 4, 1, 0, -1.1)
Traceback (most recent call last):
...
ValueError: m values must be integer or half integer
NOTES:
The Wigner 3j symbol obeys the following symmetry rules:
• invariant under any permutation of the columns (with the exception of a sign change
where J := j1 + j2 + j3 ):
Wigner3j(j1 , j2 , j3 , m1 , m2 , m3 ) = Wigner3j(j3 , j1 , j2 , m3 , m1 , m2 )
= Wigner3j(j2 , j3 , j1 , m2 , m3 , m1 )
= (−1)J Wigner3j(j3 , j2 , j1 , m3 , m2 , m1 )
= (−1)J Wigner3j(j1 , j3 , j2 , m1 , m3 , m2 )
= (−1)J Wigner3j(j2 , j1 , j3 , m2 , m1 , m3 )
ALGORITHM:
This function uses the algorithm of [Edmonds74] to calculate the value of the 3j symbol
exactly. Note that the formula contains alternating sums over large factorials and is
therefore unsuitable for finite precision arithmetic and only useful for a computer alge-
bra system [Rasch03].
REFERENCES:
AUTHORS:
• Jens Rasch (2009-03-24): initial version
sympy.physics.wigner.wigner_6j(j_1, j_2, j_3, j_4, j_5, j_6, prec=None)
Calculate the Wigner 6j symbol Wigner6j(j1 , j2 , j3 , j4 , j5 , j6 ).
INPUT:
• j_1, …, j_6 - integer or half integer
• prec - precision, default: None. Providing a precision can drastically speed up the
calculation.
OUTPUT:
Rational number times the square root of a rational number (if prec=None), or real num-
ber if a precision is given.
Examples
It is an error to have arguments that are not integer or half integer values or do not fulfill
the triangle relation:
sage: wigner_6j(2.5,2.5,2.5,2.5,2.5,2.5)
Traceback (most recent call last):
...
ValueError: j values must be integer or half integer and fulfill the triangle␣
,→relation
sage: wigner_6j(0.5,0.5,1.1,0.5,0.5,1.1)
Traceback (most recent call last):
...
ValueError: j values must be integer or half integer and fulfill the triangle␣
,→relation
NOTES:
The Wigner 6j symbol is related to the Racah symbol but exhibits more symmetries as
detailed below.
• Wigner 6j symbols are left invariant under any permutation of the columns:
Wigner6j(j1 , j2 , j3 , j4 , j5 , j6 ) = Wigner6j(j3 , j1 , j2 , j6 , j4 , j5 )
= Wigner6j(j2 , j3 , j1 , j5 , j6 , j4 )
= Wigner6j(j3 , j2 , j1 , j6 , j5 , j4 )
= Wigner6j(j1 , j3 , j2 , j4 , j6 , j5 )
= Wigner6j(j2 , j1 , j3 , j5 , j4 , j6 )
• They are invariant under the exchange of the upper and lower arguments in each of
any two columns, i.e.
Examples
It is an error to have arguments that are not integer or half integer values or do not fulfill
the triangle relation:
ALGORITHM:
This function uses the algorithm of [Edmonds74] to calculate the value of the 3j symbol
exactly. Note that the formula contains alternating sums over large factorials and is
therefore unsuitable for finite precision arithmetic and only useful for a computer alge-
bra system [Rasch03].
Unit systems
This module integrates unit systems into SymPy, allowing a user choose which system to use
when doing their computations and providing utilities to display and convert units.
Unit systems are composed of units and constants, which are themselves described from
dimensions and numbers, and possibly a prefix. Quantities are defined by their unit and their
numerical value, with respect to the current system.
The main advantage of this implementation over the old unit module is that it divides the
units in unit systems, so that the user can decide which units to use, instead of having all in
the name space (for example astrophysicists can only use units with ua, Earth or Sun masses,
the theoricists will use natural system, etc.). Moreover it allows a better control over the
dimensions and conversions.
Ideas about future developments can be found on the Github wiki.
Dimensions
Introduction
At the root of unit systems are dimension systems, whose structure mainly determines the
one of unit systems. Our definition could seem rough but they are largely sufficient for our
purposes.
A dimension will be defined as a property which is measurable and assigned to a specific
phenomenon. In this sense dimensions are different from pure numbers because they carry
some extra-sense, and for this reason two different dimensions cannot be added. For example
time or length are dimensions, but also any other things which has some sense for us, like
angle, number of particles (moles…) or information (bits…).
From this point of view the only truly dimensionless quantity are pure numbers. The idea
of being dimensionless is very system-dependent, as can be seen from the (c, ℏ, G), in which
all units appears to be dimensionless in the usual common sense. This is unavoidable for
computability of generic unit systems (but at the end we can tell the program what is dimen-
sionless).
Dimensions can be composed together by taking their product or their ratio (to be defined
below). For example the velocity is defined as length divided by time, or we can see the length
as velocity multiplied by time, depending of what we see as the more fundamental: in general
we can select a set of base dimensions from which we can describe all the others.
Group structure
After this short introduction whose aim was to introduce the dimensions from an intuitive
perspective, we describe the mathematical structure. A dimension system with n independent
dimensions {di }i=1,...,n is described by a multiplicative group G:
• there an identity element 1 corresponding to pure numbers;
• the product D3 = D1 D2 of two elements D1 , D2 ∈ G is also in G;
• any element D ∈ G has an inverse D−1 ∈ G.
We denote
Dn = D × · · · × D,
| {z }
n times
and by definition D0 = 1. The {di }i=1,...,n are called generators of the group since any element
D ∈ G can be expressed as the product of powers of the generators:
∏
n
D= dai i , ai ∈ Z.
i=1
The identity is given for ai = 0, ∀i, while we recover the generator di for ai = 1, aj = 0, ∀j ̸= i.
This group has the following properties:
1. abelian, since the generator commutes, [di , dj ] = 0;
2. countable (infinite but discrete) since the elements are indexed by the powers of the
generators1 .
One can change the dimension basis {d′i }i=1,...,n by taking some combination of the old gener-
ators:
∏
n
d′i =
P
dj ij .
j=1
It is possible to use the linear space Zn as a representation of the group since the power
coefficients ai carry all the information one needs (we do not distinguish between the element
of the group and its representation):
a1
(di )j = δij , D = ... .
an
1 In general we will consider only dimensions with a maximum coefficient, so we can only a truncation of the
group; but this is not useful for the algorithm.
The change of basis to d′i follows the usual rule of change of basis for linear space, the matrix
being given by the coefficients Pij , which are simply the coefficients of the new vectors in
term of the old basis:
d′i = Pij dj .
An example
In order to illustrate all this formalism, we end this section with a specific example, the MKS
system (m, kg, s) with dimensions (L: length, M: mass, T: time). They are represented as (we
will always sort the vectors in alphabetic order)
1 0 0
L = 0 , M = 1 , T = 0 .
0 0 1
V = LT −1 , A = M L2 T −2 ,
1 2
V = 0 , A = 1 .
−1 −2
We can change the basis to go to the natural system (m, c, ℏ) with dimension (L: length, V:
velocity, A: action)2 . In this basis the generators are
1 0 0
A = 0 , L = 1 , V = 0 ,
0 0 1
T = LV −1 , M = AV −2 ,
0 1
T = 1 , M = 0 .
−1 −2
Finally the inverse change of basis matrix P −1 is obtained by gluing the vectors expressed in
the old basis:
2 1 1
P −1 = 1 0 0 .
−2 0 −1
To find the change of basis matrix we just have to take the inverse
0 1 0
P = 1 0 1 .
0 −2 −1
2 We anticipate a little by considering c and ℏ as units and not as physical constants.
Quantities
A quantity is defined by its name, dimension and factor to a canonical quantity of the same
dimension. The canonical quantities are an internal reference of the units module and should
not be relevant for end-users. Both units and physical constants are quantities.
Units
Units, such as meters, seconds and kilograms, are usually reference quantities chosen by men
to refer to other quantities.
After defining several units of different dimensions we can form a unit system, which is basi-
cally a dimension system with a notion of scale.
Constants
Physical constants are just quantities. They indicate that we used not to understand that two
dimensions are in fact the same. For example, we see a velocity for the light different from
1 because we do not think that time is the same as space (which is normal because of our
sense; but it is different at the fundamental level). For example, once there was the “heat
constant” which allowed to convert between joules and calories since people did not know
that heat was energy. As soon as they understood it they fixed this constant to 1 (this is a very
schematic story).
We can interpret the fact that now we fix the value of fundamental constants in the SI as
showing that they are units (and we use them to define the other usual units).
It is not possible to define from scratch units and unit systems: one needs to define some
references, and then build the rest over them. Said in another way, we need an origin for the
scales of our units (i.e. a unit with factor 1), and to be sure that all units of a given dimension
are defined consistently we need to use the same origin for all of them. This can happen if
we want to use a derived unit as a base units in another system: we should not define it as
having a scale 1, because, even if it is inconsistent inside the system, we could not convert
to the first system since we have two different units (from our point of view) of same scale
(which means they are equal for the computer).
We will say that the dimensions and scales defined outside systems are canonical, because
we use them for all computations. On the other side the dimensions and scales obtained with
reference to a system are called physical, because they ultimately carry a sense.
Let’s use a concrete (and important) example: the case of the mass units. We would like to
define the gram as the origin. We would like to define the gram as the canonical origin for
the mass, so we assign it a scale 1. Then we can define a system (e.g. in chemistry) that take
it as a base unit. The MKS system prefers to use the kilogram; a naive choice would be to
attribute it a scale if 1 since it is a base, but we see that we could not convert to the chemistry
system because g and kg have both been given the same factor. So we need to define kg as
1000 g, and only then use it as a base in MKS. But as soon as we ask the question “what is
the factor of kg in MKS?”, we get the answer 1, since it is a base unit.
Thus we will define all computations without referring to a system, and it is only at the end
that we can plug the result into a system to give the context we are interested in.
Literature
Examples
In the following sections we give few examples of what can be done with this module.
Dimensional analysis
ma = F
where m, a and F are the mass, the acceleration and the force respectively. Knowing the
dimensions of m (M ) and a (LT −2 ), we will determine the dimension of F ; obviously we will
find that it is a force: M LT −2 .
From there we will use the expression of the gravitational force between the particle of mass
m and the body of mass M , at a distance r
GmM
F =
r2
to determine the dimension of the Newton’s constant G. The result should be L3 M −1 T −2 .
>>> from sympy import symbols
>>> from sympy.physics.units import length, mass, acceleration, force
>>> from sympy.physics.units import gravitational_constant as G
>>> from sympy.physics.units.dimensions import dimsys_SI
>>> F = mass*acceleration
>>> F
Dimension(acceleration*mass)
>>> dimsys_SI.get_dimensional_dependencies(F)
{'length': 1, 'mass': 1, 'time': -2}
>>> dimsys_SI.get_dimensional_dependencies(force)
{'length': 1, 'mass': 1, 'time': -2}
Dimensions cannot compared directly, even if in the SI convention they are the
same:
>>> F == force
False
Note that one should first solve the equation, and then substitute with the dimensions.
T2 4π 2
=
a3 GM
we can find the Venus orbital period using the known values for the other variables (taken
from Wikipedia). The result should be 224.701 days.
>>> M = Quantity(”solar_mass”)
>>> M.set_dimension(mass, ”SI”)
>>> M.set_scale_factor(1.9891e30*kilogram, ”SI”)
To convert to days, use the convert_to function (and possibly approximate the outcoming
result):
We could also have the solar mass and the day as units coming from the astrophysical system,
but we wanted to show how to create a unit that one needs.
We can see in this example that intermediate dimensions can be ill-defined, such as sqrt(G),
but one should check that the final result - when all dimensions are combined - is well defined.
Most of the examples in the doc use MKS system and are presented from the computer point
of view: from a human point, adding length to time is not legal in MKS but it is in natural
system; for a computer in natural system there is no time dimension (but a velocity dimension
instead) - in the basis - so the question of adding time to length has no meaning.
class sympy.physics.units.dimensions.Dimension
This class represent the dimension of a physical quantities.
The Dimension constructor takes as parameters a name and an optional symbol.
For example, in classical mechanics we know that time is different from temperature and
dimensions make this difference (but they do not provide any measure of these quantites.
has_integer_powers(dim_sys)
Check if the dimension object has only integer powers.
All the dimension powers should be integers, but rational powers may appear in
intermediate steps. This method may be used to check that the final result is well-
defined.
is_dimensionless
Check if the dimension object really has a dimension.
A dimension should have at least one component with non-zero power.
class sympy.physics.units.dimensions.DimensionSystem
DimensionSystem represents a coherent set of dimensions.
The constructor takes three parameters:
• base dimensions;
• derived dimensions: these are defined in terms of the base dimensions (for example
velocity is defined from the division of length by time);
• dependency of dimensions: how the derived dimensions depend on the base dimen-
sions.
Optionally either the derived_dims or the dimensional_dependencies may be omitted.
can_transf_matrix
Useless method, kept for compatibility with previous versions.
DO NOT USE.
Return the canonical transformation matrix from the canonical to the base dimen-
sion basis.
It is the inverse of the matrix computed with inv_can_transf_matrix().
dim
Useless method, kept for compatibility with previous versions.
DO NOT USE.
Give the dimension of the system.
That is return the number of dimensions forming the basis.
dim_can_vector(dim)
Useless method, kept for compatibility with previous versions.
DO NOT USE.
Dimensional representation in terms of the canonical base dimensions.
dim_vector(dim)
Useless method, kept for compatibility with previous versions.
DO NOT USE.
Vector representation in terms of the base dimensions.
inv_can_transf_matrix
Useless method, kept for compatibility with previous versions.
DO NOT USE.
Compute the inverse transformation matrix from the base to the canonical dimension
basis.
It corresponds to the matrix where columns are the vector of base dimensions in
canonical basis.
This matrix will almost never be used because dimensions are always defined with
respect to the canonical basis, so no work has to be done to get them in this basis.
Nonetheless if this matrix is not square (or not invertible) it means that we have
chosen a bad basis.
is_consistent
Useless method, kept for compatibility with previous versions.
DO NOT USE.
Check if the system is well defined.
is_dimensionless(dimension)
Check if the dimension object really has a dimension.
A dimension should have at least one component with non-zero power.
list_can_dims
Useless method, kept for compatibility with previous versions.
DO NOT USE.
List all canonical dimension names.
print_dim_base(dim)
Give the string expression of a dimension in term of the basis symbols.
static sort_dims(dims)
Useless method, kept for compatibility with previous versions.
DO NOT USE.
Sort dimensions given in argument using their str function.
This function will ensure that we get always the same tuple for a given set of dimen-
sions.
Unit prefixes
dim
Give the dimension of the system.
That is return the number of units forming the basis.
extend(base, units=(), name=”, description=”)
Extend the current system into a new one.
Take the base and normal units of the current system to merge them to the base
and normal units given in argument. If not provided, name and description are
overridden by empty strings.
is_consistent
Check if the underlying dimension system is consistent.
print_unit_base(unit)
Useless method.
DO NOT USE, use instead convert_to.
Give the string expression of a unit in term of the basis.
Units are displayed by decreasing power.
Physical quantities
Physical quantities.
class sympy.physics.units.quantities.Quantity
Physical quantity: can be a unit of measure, a constant or a generic quantity.
abbrev
Symbol representing the unit name.
Prepend the abbreviation with the prefix symbol if it is defines.
convert_to(other)
Convert the quantity to another quantity of same dimensions.
Examples
free_symbols
Return free symbols from quantity.
scale_factor
Overall magnitude of the quantity as compared to the canonical units.
Examples
7.62950196312651e-20*gravitational_constant**(-0.5)*hbar**0.5*speed_of_light**0.5
Abstract
Gamma matrices
Examples
>>> LorentzIndex.metric
metric(LorentzIndex,LorentzIndex)
sympy.physics.hep.gamma_matrices.extract_type_tens(expression, component)
Extract from a TensExpr all tensors with component.
Returns two tensor expressions:
• the first contains all Tensor of having component.
• the second contains all remaining.
sympy.physics.hep.gamma_matrices.gamma_trace(t)
trace of a single line of gamma matrices
Examples
sympy.physics.hep.gamma_matrices.kahane_simplify(expression)
This function cancels contracted elements in a product of four dimensional gamma ma-
trices, resulting in an expression equal to the given one, without the contracted gamma
matrices.
Parameters ‘expression‘ the tensor expression containing the gamma
matrices to simplify.
Notes
If spinor indices are given, the matrices must be given in the order given in the product.
References
[1] Algorithm for Reducing Contracted Products of gamma Matrices, Joseph Kahane,
Journal of Mathematical Physics, Vol. 9, No. 10, October 1968.
Examples
When using, always remember that the original expression coefficient has to be handled
separately
>>> tc = G(i0)*G(i1)
>>> kahane_simplify(tc)
GammaMatrix(i0)*GammaMatrix(i1)
Algorithm
The idea behind the algorithm is to use some well-known identities, i.e., for contractions
enclosing an even number of γ matrices
γ µ γa1 · · · γa2N γµ = 2(γa2N γa1 · · · γa2N −1 + γa2N −1 · · · γa1 γa2N )
for an odd number of γ matrices
γ µ γa1 · · · γa2N +1 γµ = −2γa2N +1 γa2N · · · γa1
Instead of repeatedly applying these identities to cancel out all contracted indices, it is
possible to recognize the links that would result from such an operation, the problem is
thus reduced to a simple rearrangement of free gamma matrices.
sympy.physics.hep.gamma_matrices.simplify_gpgp(ex, sort=True)
simplify products G(i)*p(-i)*G(j)*p(-j) -> p(i)*p(-i)
Examples
Abstract
In this documentation the components of the sympy.physics.vector module have been dis-
cussed. vector has been written to facilitate the operations pertaining to 3-dimensional
vectors, as functions of time or otherwise, in sympy.physics.
Guide to Vector
In vector, vectors and reference frames are the “building blocks” of dynamic systems. This
document will describe these mathematically and describe how to use them with this module’s
code.
Vector
A vector is a geometric object that has a magnitude (or length) and a direction. Vectors in
3-space are often represented on paper as:
Vector Algebra
Vector Operations
Multiple algebraic operations can be done with vectors: addition between vectors, scalar
multiplication, and vector multiplication.
Vector addition as based on the parallelogram law.
b
a
a+b a
b
Scalar multiplication is the product of a vector and a scalar; the result is a vector with the
same orientation but whose magnitude is scaled by the scalar. Note that multiplication by
-1 is equivalent to rotating the vector by 180 degrees about an arbitrary axis in the plane
perpendicular to the vector.
a
2a
-a
A unit vector is simply a vector whose magnitude is equal to 1. Given any vector v we can
define a unit vector as:
v
^
nv =
∥v∥
Note that every vector can be written as the product of a scalar and unit vector.
Three vector products are implemented in vector: the dot product, the cross product, and
the outer product.
The dot product operation maps two vectors to a scalar. It is defined as:
a · b = ∥a∥∥b∥ cos(θ)
b
a c
a a 45° a
a·b=b·a
The cross product vector multiplication operation of two vectors returns a vector:
a×b=c
The vector c has the following properties: it’s orientation is perpendicular to both a and b,
it’s magnitude is defined as ∥c∥ = ∥a∥∥b∥ sin(θ) (where θ is the angle between a and b), and
has a sense defined by using the right hand rule between ∥a∥∥b∥. The figure below shows
this:
c / sqrt(2)
c
b a 45°
d
a axa=0
(a × b) × c ̸= a × (b × c)
inertia section (that is where it is used). Other useful vector properties and relationships are:
α(a + b) = αa + αb
a · (b + c) = a · b + a · c
a × (b + c) = a × b + a × b
(a × b) · c gives the scalar triple product.
a × (b · c) does not work, as you cannot cross a vector and a scalar.
(a × b) · c = a · (b × c)
(a × b) · c = (b × c) · a = (c × a) · b
(a × b) × c = b(a · c) − a(b · c)
a × (b × c) = b(a · c) − c(a · b)
Alternative Representation
a = ax^
nx + ay^ ny + az^ nz
b = bx^
n x + by ^
n y + bz ^
nz
a + b = (ax + bx )^
nx + (ay + by )^
ny + (az + bz )^
nz
αb = αbx^
nx + αby^
ny + αbz^
nz
a · b = ax bx + ay by + az bz
^
nx ^ ny ^ nz
a × b = det ax ay az
bx by bz
ax ay az
(a × b) · c = det bx by bz
cx cy cz
a = (a · ^ nx + (a · ^
nx )^ ny + (a · ^
ny )^ nz )^
nz
Examples
a=^
nx + 5^
ny
b=^
ny + α^
nz
a+b=^
nx + 6^
ny + α^
nz
a·b=5
a ·^
ny = 5
a ·^
nz = 0
a × b = 5α^
nx − α^
ny + ^
nz
b × a = −5α^ ny − ^
nx + α^ nz
Vector Calculus
To deal with the calculus of vectors with moving object, we have to introduce the concept of
a reference frame. A classic example is a train moving along its tracks, with you and a friend
inside. If both you and your friend are sitting, the relative velocity between the two of you is
zero. From an observer outside the train, you will both have velocity though.
We will now apply more rigor to this definition. A reference frame is a virtual “platform”
which we choose to observe vector quantities from. If we have a reference frame N, vector
a is said to be fixed in the frame N if none of its properties ever change when observed from
N. We will typically assign a fixed orthonormal basis vector set with each reference frame; N
will have ^nx , ^
ny , ^
nz as its basis vectors.
Derivatives of Vectors
A vector which is not fixed in a reference frame therefore has changing properties when
observed from that frame. Calculus is the study of change, and in order to deal with the
peculiarities of vectors fixed and not fixed in different reference frames, we need to be more
explicit in our definitions.
f
B
e Fixed in:
A B
A cx
d dx
e x
c f x
In the above figure, we have vectors c, d, e, f. If one were to take the derivative of e with
respect to θ:
de
dθ
it is not clear what the derivative is. If you are observing from frame A, it is clearly non-zero.
If you are observing from frame B, the derivative is zero. We will therefore introduce the
frame as part of the derivative notation:
A
de
̸= 0, the derivative of e with respect to θ in the reference frame A
dθ
B
de
= 0, the derivative of e with respect to θ in the reference frame B
dθ
A
dc
= 0, the derivative of c with respect to θ in the reference frame A
dθ
B
dc
̸= 0, the derivative of c with respect to θ in the reference frame B
dθ
Here are some additional properties of derivatives of vectors in specific frames:
A A
d da A db
(a + b) = +
dt dt dt
A A
d dγ da
γa = a+γ
dt dt dt
A A A
d da db
(a × b) = ×b+a×
dt dt dt
We need to now define the relationship between two different reference frames; or how to
relate the basis vectors of one frame to another. We can do this using a direction cosine matrix
(DCM). The direction cosine matrix relates the basis vectors of one frame to another, in the
following fashion:
^
^
ax [A B ] bx
^
ay = C ^ by
^
az ^
bz
When two frames (say, A & B) are initially aligned, then one frame has all of its basis vectors
rotated around an axis which is aligned with a basis vector, we say the frames are related by
a simple rotation. The figure below shows this:
az
bz
B
by
A θ
ay
θ
ax bx
The above rotation is a simple rotation about the Z axis by an angle θ. Note that after the
az and ^
rotation, the basis vectors ^ bz are still aligned.
This rotation can be characterized by the following direction cosine matrix:
cos(θ) − sin(θ) 0
C = sin(θ) cos(θ) 0
A B
0 0 1
Simple rotations about the X and Y axes are defined by:
1 0 0
DCM for x-axis rotation: 0 cos(θ) − sin(θ)
0 sin(θ) cos(θ)
cos(θ) 0 sin(θ)
DCM for y-axis rotation: 0 1 0
− sin(θ) 0 cos(θ)
Rotation in the positive direction here will be defined by using the right-hand rule.
The direction cosine matrix is also involved with the definition of the dot product between
sets of basis vectors. If we have two reference frames with associated basis vectors, their
direction cosine matrix can be defined as:
Cxx Cxy Cxz ^ax · ^
bx ^ax · ^ ax · ^
by ^ bz
Cyx Cyy Cyz = ^ay · ^
bx ^ay · ^ bz
ay · ^
by ^
Czx Czy Czz ^ ^
az · b x ^ ^
az · by ^ ^
az · b z
Additionally, the direction cosine matrix is orthogonal, in that:
A
CB = (B CA )−1
= (B CA )T
If we have reference frames A and B, which in this example have undergone a simple z-axis
rotation by an amount θ, we will have two sets of basis vectors. We can then define two
vectors: a = ^
ax + ^ az and b = ^
ay + ^ bx + ^
by + ^
bz . If we wish to express b in the A frame, we do
the following:
b =^
bx + ^ by + ^
bz
[ ] [ ] [ ]
b= ^ ax · (bx + ^
^ by + ^
bz ) ^ax + ^ay · (^
bx + ^
by + ^
bz ) ^ay + ^az · (^
bx + ^
by + ^
bz ) ^az
b = (cos(θ) − sin(θ))^ ay + ^
ax + (sin(θ) + cos(θ))^ az
a =^ax + ^ay + ^
az
[ ] [ ] [ ]
a= ^ bx · (^
ax + ^ az ) ^
ay + ^ bx + ^by · (^
ax + ^ az ) ^
ay + ^ by + ^
bz · (^
ax + ^ az ) ^
ay + ^ bz
a = (cos(θ) + sin(θ))^bx + (− sin(θ) + cos(θ))^ by + ^ bz
If we have reference frames A and B we will have two sets of basis vectors. We can then
define two vectors: a = ax^ ax + ay^ az and b = bx^
ay + az^ b x + by ^
b y + bz ^
bz . If we want to take
the derivative of b in the reference frame A, we must first express it in A, and the take the
derivatives of the measure numbers:
A
db d(b · ^
ax ) d(b · ^
ay ) d(b · ^
az )
= ^
ax + ^
ay + ^
az +
dx dx dx dx
Examples
ay by
ax bx
A B
In this example we have two bodies, each with an attached reference frame. We will say that
θ and x are functions of time. We wish to know the time derivative of vector c in both the A
and B frames.
First, we need to define c; c = x^
bx + l^
by . This provides a definition in the B frame. We can
now do the following:
B
dc dx ^ dl
= bx + ^ by
dt dt dt
= ẋ^
bx
To take the derivative in the A frame, we have to first relate the two frames:
cos(θ) 0 sin(θ)
C =
A B
0 1 0
− sin(θ) 0 cos(θ)
Note that this is the time derivative of c in A, and is expressed in the A frame. We can express
it in the B frame however, and the expression will still be valid:
A
dc
= (−θ̇ sin(θ)x + cos(θ)ẋ)^
ax + (θ̇ cos(θ)x + sin(θ)ẋ)^
az
dt
= ẋ^
bx − θx^bz
Note the difference in expression complexity between the two forms. They are equivalent, but
one is much simpler. This is an extremely important concept, as defining vectors in the more
complex forms can vastly slow down formulation of the equations of motion and increase their
length, sometimes to a point where they cannot be shown on screen.
We have waited until after all of the relevant mathematical relationships have been defined
for vectors and reference frames to introduce code. This is due to how vectors are formed.
When starting any problem in vector, one of the first steps is defining a reference frame
(remember to import sympy.physics.vector first):
>>> from sympy.physics.vector import *
>>> N = ReferenceFrame('N')
Now we have created a reference frame, N. To have access to any basis vectors, first a
reference frame needs to be created. Now that we have made and object representing N, we
can access its basis vectors:
>>> N.x
N.x
>>> N.y
N.y
>>> N.z
N.z
Remember, don’t add a scalar quantity to a vector (N.x + 5); this will raise an error. At this
point, we’ll use SymPy’s Symbol in our vectors. Remember to refer to SymPy’s Gotchas and
Pitfalls when dealing with symbols.:
>>> from sympy import Symbol, symbols
>>> x = Symbol('x')
>>> x * N.x
(continues on next page)
In vector multiple interfaces to vector multiplication have been implemented, at the operator
level, method level, and function level. The vector dot product can work as follows:
The “official” interface is the function interface; this is what will be used in all examples. This
is to avoid confusion with the attribute and methods being next to each other, and in the case
of the operator operation priority. The operators used in vector for vector multiplication do
not posses the correct order of operations; this can lead to errors. Care with parentheses is
needed when using operators to represent vector multiplication.
The cross product is the other vector multiplication which will be discussed here. It offers
similar interfaces to the dot product, and comes with the same warnings.
Two additional operations can be done with vectors: normalizing the vector to length 1, and
getting its magnitude. These are done as follows:
Vectors are often expressed in a matrix form, especially for numerical purposes. Since the
matrix form does not contain any information about the reference frame the vector is defined
in, you must provide a reference frame to extract the measure numbers from the vector. There
is a convenience function to do this:
We have already introduced our first reference frame. We can take the derivative in that
frame right now, if we desire:
>>> (x * N.x + N.y).diff(x, N)
N.x
SymPy has a diff function, but it does not currently work with vector Vectors, so please use
Vector’s diff method. The reason for this is that when differentiating a Vector, the frame
of reference must be specified in addition to what you are taking the derivative with respect
to; SymPy’s diff function doesn’t fit this mold.
The more interesting case arise with multiple reference frames. If we introduce a second
reference frame, A, we now have two frames. Note that at this point we can add components
of N and A together, but cannot perform vector multiplication, as no relationship between the
two frames has been defined.
>>> A = ReferenceFrame('A')
>>> A.x + N.x
N.x + A.x
If we want to do vector multiplication, first we have to define and orientation. The orient
method of ReferenceFrame provides that functionality.
>>> A.orient(N, 'Axis', [x, N.y])
If we desire, we can view the DCM between these two frames at any time. This can be
calculated with the dcm method. This code: N.dcm(A) gives the dcm N CA .
This orients the A frame relative to the N frame by a simple rotation around the Y axis, by an
amount x. Other, more complicated rotation types include Body rotations, Space rotations,
quaternions, and arbitrary axis rotations. Body and space rotations are equivalent to doing 3
simple rotations in a row, each about a basis vector in the new frame. An example follows:
>>> N = ReferenceFrame('N')
>>> Bp = ReferenceFrame('Bp')
>>> Bpp = ReferenceFrame('Bpp')
>>> B = ReferenceFrame('B')
>>> q1,q2,q3 = symbols('q1 q2 q3')
>>> Bpp.orient(N,'Axis', [q1, N.x])
>>> Bp.orient(Bpp,'Axis', [q2, Bpp.y])
>>> B.orient(Bp,'Axis', [q3, Bp.z])
>>> N.dcm(B)
Matrix([
[ cos(q2)*cos(q3), -
,→sin(q3)*cos(q2), sin(q2)],
[sin(q1)*sin(q2)*cos(q3) + sin(q3)*cos(q1), -sin(q1)*sin(q2)*sin(q3) +␣
,→cos(q1)*cos(q3), -sin(q1)*cos(q2)],
Space orientations are similar to body orientation, but applied from the frame to body. Body
and space rotations can involve either two or three axes: ‘XYZ’ works, as does ‘YZX’, ‘ZXZ’,
‘YXY’, etc. What is key is that each simple rotation is about a different axis than the previous
one; ‘ZZX’ does not completely orient a set of basis vectors in 3 space.
Sometimes it will be more convenient to create a new reference frame and orient relative to an
existing one in one step. The orientnew method allows for this functionality, and essentially
wraps the orient method. All of the things you can do in orient, you can do in orientnew.
Quaternions (or Euler Parameters) use 4 value to characterize the orientation of the frame.
This and arbitrary axis rotations are described in the orient and orientnew method help, or
in the references [Kane1983].
Finally, before starting multiframe calculus operations, we will introduce another vector tool:
dynamicsymbols. dynamicsymbols is a shortcut function to create undefined functions of time
within SymPy. The derivative of such a ‘dynamicsymbol’ is shown below.
The ‘dynamicsymbol’ printing is not very clear above; we will also introduce a few other tools
here. We can use vprint instead of print for non-interactive sessions.
>>> q1
q1(t)
>>> q1d = diff(q1, Symbol('t'))
>>> vprint(q1)
q1
>>> vprint(q1d)
q1'
For interactive sessions use init_vprinting. There also exist analogs for SymPy’s vprint,
vpprint, and latex, vlatex.
A ‘dynamicsymbol’ should be used to represent any time varying quantity in vector, whether
it is a coordinate, varying position, or force. The primary use of a ‘dynamicsymbol’ is for
speeds and coordinates (of which there will be more discussion in the Kinematics Section of
the documentation).
Now we will define the orientation of our new frames with a ‘dynamicsymbol’, and can take
derivatives and time derivatives with ease. Some examples follow.
>>> N = ReferenceFrame('N')
>>> B = N.orientnew('B', 'Axis', [q1, N.x])
>>> (B.y*q2 + B.z).diff(q2, N)
B.y
>>> (B.y*q2 + B.z).dt(N)
(-q1' + q2')*B.y + q2*q1'*B.z
Note that the output vectors are kept in the same frames that they were provided in. This
remains true for vectors with components made of basis vectors from multiple frames:
What follows is a short description of how vectors are defined by the code in vector. It is
provided for those who want to learn more about how this part of sympy.physics.vector
works, and does not need to be read to use this module; don’t read it unless you want to learn
how this module was implemented.
Every Vector’s main information is stored in the args attribute, which stores the three mea-
sure numbers for each basis vector in a frame, for every relevant frame. A vector does not
exist in code until a ReferenceFrame is created. At this point, the x, y, and z attributes of the
reference frame are immutable Vector’s which have measure numbers of [1,0,0], [0,1,0], and
[0,0,1] associated with that ReferenceFrame. Once these vectors are accessible, new vectors
can be created by doing algebraic operations with the basis vectors. A vector can have com-
ponents from multiple frames though. That is why args is a list; it has as many elements in
the list as there are unique ReferenceFrames in its components, i.e. if there are A and B frame
basis vectors in our new vector, args is of length 2; if it has A, B, and C frame basis vector,
args is of length three.
Each element in the args list is a 2-tuple; the first element is a SymPy Matrix (this is where
the measure numbers for each set of basis vectors are stored) and the second element is a
ReferenceFrame to associate those measure numbers with.
ReferenceFrame stores a few things. First, it stores the name you supply it on creation (name
attribute). It also stores the direction cosine matrices, defined upon creation with the ori-
entnew method, or calling the orient method after creation. The direction cosine matrices
are represented by SymPy’s Matrix, and are part of a dictionary where the keys are the Ref-
erenceFrame and the value the Matrix; these are set bi-directionally; in that when you orient
A to N you are setting A’s orientation dictionary to include N and its Matrix, but you are also
setting N’s orientation dictionary to include A and its Matrix (that DCM being the transpose
of the other).
Vector: Kinematics
This document will give some mathematical background to describing a system’s kinematics
as well as how to represent the kinematics in physics.vector.
Introduction to Kinematics
The first topic is rigid motion kinematics. A rigid body is an idealized representation of a
physical object which has mass and rotational inertia. Rigid bodies are obviously not flexible.
We can break down rigid body motion into translational motion, and rotational motion (when
dealing with particles, we only have translational motion). Rotational motion can further be
broken down into simple rotations and general rotations.
Translation of a rigid body is defined as a motion where the orientation of the body does not
change during the motion; or during the motion any line segment would be parallel to itself
at the start of the motion.
Simple rotations are rotations in which the orientation of the body may change, but there is
always one line which remains parallel to itself at the start of the motion.
General rotations are rotations which there is not always one line parallel to itself at the start
of the motion.
Angular Velocity
The angular velocity of a rigid body refers to the rate of change of its orientation. The angular
velocity of a body is written down as: N ω B , or the angular velocity of B in N, which is a vector.
Note that here, the term rigid body was used, but reference frames can also have angular
velocities. Further discussion of the distinction between a rigid body and a reference frame
will occur later when describing the code representation.
Angular velocity is defined as being positive in the direction which causes the orientation
angles to increase (for simple rotations, or series of simple rotations).
nx
nx
N
nz
B
ny
N
wB=q nx
The angular velocity vector represents the time derivative of the orientation. As a time deriva-
tive vector quantity, like those covered in the Vector & ReferenceFrame documentation, this
quantity (angular velocity) needs to be defined in a reference frame. That is what the N is in
the above definition of angular velocity; the frame in which the angular velocity is defined in.
The angular velocity of B in N can also be defined by:
N
d^
by ^ ^ N ^
db z ^ ^ N ^
dbx ^ ^
N
ωB = ( · bz )bx + ( · bx )by + ( · by )bz
dt dt dt
It is also common for a body’s angular velocity to be written as:
N
ω B = wx^
b x + wy ^
b y + wz ^
bz
There are a few additional important points relating to angular velocity. The first is the addi-
tion theorem for angular velocities, a way of relating the angular velocities of multiple bodies
and frames. The theorem follows:
N
ω D =N ω A +A ω B +B ω C +C ω D
nz
A
N
ny ax
q2
nx
q1 cy
bz
q3
B C
D
N
ωA = 0
A
ω B = q˙1^
ax
ω = −q˙2^
B C
bz
C
ω D = q˙3^
cy
N
ax − q˙2^
ω D = q˙1^ bz + q˙3^
cy
Note the signs used in the angular velocity definitions, which are related to how the displace-
ment angle is defined in this case.
This theorem makes defining angular velocities of multibody systems much easier, as the
angular velocity of a body in a chain needs to only be defined to the previous body in order to
be fully defined (and the first body needs to be defined in the desired reference frame). The
following figure shows an example of when using this theorem can make things easier.
p2 w3
w1 w2
p3
p1
Here we can easily write the angular velocity of the body D in the reference frame of the first
body A:
A
ω D = w1 p^1 + w2 p^2 + w3 p^3
It is very important to remember to only use this with angular velocities; you cannot use this
theorem with the velocities of points.
There is another theorem commonly used: the derivative theorem. It provides an alternative
method (which can be easier) to calculate the time derivative of a vector in a reference frame:
N B
dv dv N B
= + ω ×v
dt dt
The vector v can be any vector quantity: a position vector, a velocity vector, angular velocity
vector, etc. Instead of taking the time derivative of the vector in N, we take it in B, where B
can be any reference frame or body, usually one in which it is easy to take the derivative on v
in (v is usually composed only of the basis vector set belonging to B). Then we add the cross
product of the angular velocity of our newer frame, N ω B and our vector quantity v. Again, you
can choose any alternative frame for this. Examples follow:
Angular Acceleration
Angular acceleration refers to the time rate of change of the angular velocity vector. Just as
the angular velocity vector is for a body and is specified in a frame, the angular acceleration
vector is for a body and is specified in a frame: N αB , or the angular acceleration of B in N,
which is a vector.
Calculating the angular acceleration is relatively straight forward:
N N
N B d ωB
α =
dt
Note that this can be calculated with the derivative theorem, and when the angular velocity
is defined in a body fixed frame, becomes quite simple:
N N
N B d ωB
α =
dt
B N
d ωB N B N B
N B
α = + ω × ω
dt
N
if ω B = wx^
bx + wy^
by + wz^
bz
B N
N B d ωB
then α = + N
ω B ×N ω B
dt | {z }
this is 0 by definition
N B
α = w˙x^
bx + w˙y^
by + w˙z^
bz
Again, this is only for the case in which the angular velocity of the body is defined in body
fixed components.
Consider a point, P : we can define some characteristics of the point. First, we can define
a position vector from some other point to P . Second, we can define the velocity vector of
P in a reference frame of our choice. Third, we can define the acceleration vector of P in a
reference frame of our choice.
These three quantities are read as:
Note that the position vector does not have a frame associated with it; this is because there
is no time derivative involved, unlike the velocity and acceleration vectors.
We can find these quantities for a simple example easily:
rOP P
O
It is critical to understand in the above example that the point O is fixed in the reference frame
N. There is no addition theorem for translational velocities; alternatives will be discussed
later though. Also note that the position of every point might not always need to be defined
to form the dynamic equations of motion. When you don’t want to define the position vector
of a point, you can start by just defining the velocity vector. For the above example:
N P
Let us instead define the velocity vector as: v = ux^
nx + uy^
ny
N P
then acceleration can be written as: a = u̇x^
nx + u̇y^
ny
There will often be cases when the velocity of a point is desired and a related point’s velocity
is known. For the cases in which we have two points fixed on a rigid body, we use the 2-Point
Theorem:
N
P
fixed in B
B
S
fixed in B
O
fixed in N
Let’s say we know the velocity of the point S and the angular velocity of the body B, both
defined in the reference frame N. We can calculate the velocity and acceleration of the point
P in N as follows:
v =N vS +N ω B × rSP
N P
a =N aS +N αB × rSP +N ω B × (N ω B × rSP )
N P
When only one of the two points is fixed on a body, the 1 point theorem is used instead.
N
P
not fixed in B
B
S
fixed in B
O
fixed in N
Here, the velocity of point S is known in the frame N, the angular velocity of B is known in
N, and the velocity of the point P is known in the frame associated with body B. We can then
write the velocity and acceleration of P in N as:
v =B vP +N vS +N ω B × rSP
N P
a =B aS +N aO +N αB × rSP +N ω B × (N ω B × rSP ) + 2N ω B ×B vP
N P
bz
B
nz O by
bx
N P
ny rOP
nx
This example has a disc translating and rotating in a plane. We can easily define the angular
velocity of the body B and velocity of the point O:
N
nz = u3^
ω B = u3^ bz
N O
v = u1^
nx + u2^
ny
We can use the 2 point theorem to calculate the velocity and acceleration of point P now.
rOP = R^
bx
v =N vO +N ω B × rOP
N P
N P
v = u1^ ny + u3^
nx + u2^ bz × R^
bx = u1^ ny + u3 R^
nx + u2^ by
a =N aO +N αB × rOP +N ω B × (N ω B × rOP )
N P
N P
a = u˙1^ ny + u˙3^
nx + u˙2^ bz × R^
bx + u3^
bz × (u3^
bz × R^
bx )
N P
a = u˙1^ ny + Ru˙3^
nx + u˙2^ by − Ru23^
bx
N ny
by
O
bx cy
nx B cx
Q
C P
1580 Chapter 5. SymPy Modules Reference
SymPy Documentation, Release 1.3
In this example we have a double pendulum. We can use the two point theorem twice here in
order to find the velocity of points Q and P ; point O’s velocity is zero in N.
rOQ = l^
bx
rQP = l^
cx
ω = u1^
N B
bz
N
ω C = u2^
cz
v =N vO +N ω B × rOQ
N Q
v = u1 l^
N Q
by
v =N vQ +N ω C × rQP
N P
v = u1 l^
N Q
by + u2^cz × l^
cx
v = u1 l^
N Q
by + u2 l^
cy
nz
N
O
ny
nx
q1
Q C
q2 P
cz
cy
cx
In this example we have a particle moving on a ring; the ring is supported by a rod which can
rotate about the ^
nx axis. First we use the two point theorem to find the velocity of the center
point of the ring, Q, then use the 1 point theorem to find the velocity of the particle on the
ring.
N
ω C = u1^
nx
rOQ = −l^
cz
N Q
v = u1 l^
cy
QP
r cx + sin(q2 )^
= R(cos(q2 )^ cy )
C P
v = Ru2 (−sin(q2 )^
cx + cos(q2 )^
cy )
N P
v =C vP +N vQ +N ω C × rQP
N P
v = Ru2 (−sin(q2 )^
cx + cos(q2 )^
cy ) + u1 l^ cx × R(cos(q2 )^
cy + u1^ cx + sin(q2 )^
cy
N P
v = −Ru2 sin(q2 )^
cx + (Ru2 cos(q2 ) + u1 l)^
cy + Ru1 sin(q2 )^
cz
A final topic in the description of velocities of points is that of rolling, or rather, rolling without
slip. Two bodies are said to be rolling without slip if and only if the point of contact on each
body has the same velocity in another frame. See the following figure:
N A
Pa is the contact point on body A
Kinematics in physics.vector
It should be clear by now that the topic of kinematics here has been mostly describing the
correct way to manipulate vectors into representing the velocities of points. Within vector
there are convenient methods for storing these velocities associated with frames and points.
We’ll now revisit the above examples and show how to represent them in sympy (page 801).
The topic of reference frame creation has already been covered. When a ReferenceFrame is
created though, it automatically calculates the angular velocity of the frame using the time
derivative of the DCM and the angular velocity definition.
>>> from sympy import Symbol, sin, cos
>>> from sympy.physics.vector import *
>>> N = ReferenceFrame('N')
>>> q1 = dynamicsymbols('q1')
>>> A = N.orientnew('A', 'Axis', [q1, N.x])
>>> A.ang_vel_in(N)
q1'*N.x
>>> B = ReferenceFrame('B')
>>> u1 = dynamicsymbols('u1')
>>> B.set_ang_vel(N, u1 * B.y)
>>> B.ang_vel_in(N)
u1*B.y
>>> N.ang_vel_in(B)
- u1*B.y
Both upon frame creation during orientnew and when calling set_ang_vel, the angular ve-
locity is set in both frames involved, as seen above.
nz
A
N
ny ax
q2
nx
q1 cy
bz
q3
B C
D
Here we have multiple bodies with angular velocities defined relative to each other. This is
coded as:
>>> N = ReferenceFrame('N')
>>> A = ReferenceFrame('A')
>>> B = ReferenceFrame('B')
>>> C = ReferenceFrame('C')
>>> D = ReferenceFrame('D')
>>> u1, u2, u3 = dynamicsymbols('u1 u2 u3')
>>> A.set_ang_vel(N, 0)
>>> B.set_ang_vel(A, u1 * A.x)
>>> C.set_ang_vel(B, -u2 * B.z)
>>> D.set_ang_vel(C, u3 * C.y)
>>> D.ang_vel_in(N)
u1*A.x - u2*B.z + u3*C.y
In vector the shortest path between two frames is used when finding the angular velocity.
That would mean if we went back and set:
>>> D.set_ang_vel(N, 0)
>>> D.ang_vel_in(N)
0
The path that was just defined is what is used. This can cause problems though, as now the
angular velocity definitions are inconsistent. It is recommended that you avoid doing this.
Points are a translational analog to the rotational ReferenceFrame. Creating a Point can be
done in two ways, like ReferenceFrame:
>>> O = Point('O')
>>> P = O.locatenew('P', 3 * N.x + N.y)
>>> P.pos_from(O)
3*N.x + N.y
>>> Q = Point('Q')
>>> Q.set_pos(P, N.z)
>>> Q.pos_from(P)
N.z
>>> Q.pos_from(O)
3*N.x + N.y + N.z
Similar to ReferenceFrame, the position vector between two points is found by the shortest
path (number of intermediate points) between them. Unlike rotational motion, there is no
addition theorem for the velocity of points. In order to have the velocity of a Point in a
ReferenceFrame, you have to set the value.
>>> O = Point('O')
>>> O.set_vel(N, u1*N.x)
>>> O.vel(N)
u1*N.x
For both translational and rotational accelerations, the value is computed by taking the time
derivative of the appropriate velocity, unless the user sets it otherwise.
>>> O.acc(N)
u1'*N.x
>>> O.set_acc(N, u2*u1*N.y)
>>> O.acc(N)
u1*u2*N.y
bz
B
nz O by
bx
N P
ny rOP
nx
First is the translating, rotating disc.
>>> N = ReferenceFrame('N')
>>> u1, u2, u3 = dynamicsymbols('u1 u2 u3')
>>> R = Symbol('R')
>>> B = ReferenceFrame('B')
>>> O = Point('O')
>>> O.set_vel(N, u1 * N.x + u2 * N.y)
>>> P = O.locatenew('P', R * B.x)
>>> B.set_ang_vel(N, u3 * B.z)
>>> P.v2pt_theory(O, N, B)
u1*N.x + u2*N.y + R*u3*B.y
>>> P.a2pt_theory(O, N, B)
u1'*N.x + u2'*N.y - R*u3**2*B.x + R*u3'*B.y
nz
N
O
ny
nx
q1
Q C
q2 P
cz
cy
cx
This is the particle moving on a ring, again.
>>> N = ReferenceFrame('N')
>>> u1, u2 = dynamicsymbols('u1 u2')
>>> q1, q2 = dynamicsymbols('q1 q2')
>>> l = Symbol('l')
>>> R = Symbol('R')
>>> C = N.orientnew('C', 'Axis', [q1, N.x])
>>> C.set_ang_vel(N, u1 * N.x)
>>> O = Point('O')
>>> O.set_vel(N, 0)
>>> Q = O.locatenew('Q', -l * C.z)
>>> P = Q.locatenew('P', R * (cos(q2) * C.x + sin(q2) * C.y))
>>> P.set_vel(C, R * u2 * (-sin(q2) * C.x + cos(q2) * C.y))
(continues on next page)
This document will describe some of the more advanced functionality that this module offers
but which is not part of the “official” interface. Here, some of the features that will be im-
plemented in the future will also be covered, along with unanswered questions about proper
functionality. Also, common problems will be discussed, along with some solutions.
Inertia (Dyadics)
A dyadic tensor is a second order tensor formed by the juxtaposition of a pair of vectors.
There are various operations defined with respect to dyadics, which have been implemented
in vector in the form of class Dyadic. To know more, refer to the Dyadic and Vector class
APIs. Dyadics are used to define the inertia of bodies within mechanics. Inertia dyadics can
be defined explicitly but the inertia function is typically much more convenient for the user:
>>> inertia(N, 1, 2, 3)
(N.x|N.x) + 2*(N.y|N.y) + 3*(N.z|N.z)
Supply a reference frame along with the products and moments of inertia
for a general object:
>>> inertia(N, 1, 2, 3, 4, 5, 6)
(N.x|N.x) + 4*(N.x|N.y) + 6*(N.x|N.z) + 4*(N.y|N.x) + 2*(N.y|N.y) + 5*(N.y|N.z) +␣
,→6*(N.z|N.x) + 5*(N.z|N.y) + 3*(N.z|N.z)
Notice that the inertia function returns a dyadic with each component represented as two
unit vectors separated by a |. Refer to the Dyadic (page 1628) section for more information
about dyadics.
Inertia is often expressed in a matrix, or tensor, form, especially for numerical purposes.
Since the matrix form does not contain any information about the reference frame(s) the
inertia dyadic is defined in, you must provide one or two reference frames to extract the
measure numbers from the dyadic. There is a convenience function to do this:
Common Issues
Here issues with numerically integrating code, choice of dynamicsymbols for coordinate and
speed representation, printing, differentiating, and substitution will occur.
Printing
The default printing options are to use sorting for Vector and Dyadic measure numbers, and
have unsorted output from the vprint, vpprint, and vlatex functions. If you are printing
something large, please use one of those functions, as the sorting can increase printing time
from seconds to minutes.
Substitution
Substitution into large expressions can be slow, and take a few minutes.
Acceleration of Points
At a minimum, points need to have their velocities defined, as the acceleration can be calcu-
lated by taking the time derivative of the velocity in the same frame. If the 1 point or 2 point
theorems were used to compute the velocity, the time derivative of the velocity expression
will most likely be more complex than if you were to use the acceleration level 1 point and 2
point theorems. Using the acceleration level methods can result in shorted expressions at this
point, which will result in shorter expressions later (such as when forming Kane’s equations).
Advanced Interfaces
Here we will cover advanced options in: ReferenceFrame, dynamicsymbols, and some asso-
ciated functionality.
ReferenceFrame
ReferenceFrame is shown as having a .name attribute and .x, .y, and .z attributes for ac-
cessing the basis vectors, as well as a fairly rigidly defined print output. If you wish to have
a different set of indices defined, there is an option for this. This will also require a different
interface for accessing the basis vectors.
>>> from sympy.physics.vector import ReferenceFrame, vprint, vpprint, vlatex
>>> N = ReferenceFrame('N', indices=['i', 'j', 'k'])
>>> N['i']
N['i']
>>> N.x
N['i']
>>> vlatex(N.x)
'\\mathbf{\\hat{n}_{i}}'
Also, the latex output can have custom strings; rather than just indices though, the entirety
of each basis vector can be specified. The custom latex strings can occur without custom
indices, and also overwrites the latex string that would be used if there were custom indices.
dynamicsymbols
The dynamicsymbols function also has ‘hidden’ functionality; the variable which is associated
with time can be changed, as well as the notation for printing derivatives.
Note that only dynamic symbols created after the change are different. The same is not true
for the .s tr attribute; this affects the printing output only, so dynamic symbols created before
or after will print the same way.
Also note that Vector’s .dt method uses the ._t attribute of dynamicsymbols, along with a
number of other important functions and methods. Don’t mix and match symbols representing
time.
Introduction
A vector, on the other hand, is an entity that is characterized by a magnitude and a direction.
Examples of vector quantities are displacement, velocity, magnetic field, etc.
A scalar can be depicted just by a number, for e.g. a temperature of 300 K. On the other
hand, vectorial quantities like acceleration are usually denoted by a vector. Given a vector
V, the magnitude of the corresponding quantity can be calculated as the magnitude of the
vector itself ∥V∥, while the direction would be specified by a unit vector in the direction of the
original vector, ^ V
V = ∥V∥ .
Fields
In general, a f ield is a vector or scalar quantity that can be specified everywhere in space as
a function of position (Note that in general a field may also be dependent on time and other
custom variables). In this module, we deal with 3-dimensional spaces only. Hence, a field is
defined as a function of the x, y and z coordinates corresponding to a location in 3D space.
For example, temperate in 3 dimensional space (a temperature field) can be written as T (x, y, z)
– a scalar function of the position. An example of a scalar field in electromagnetism is the
electric potential.
In a similar manner, a vector field can be defined as a vectorial function of the location (x, y, z)
of any point in space.
For instance, every point on the earth may be considered to be in the gravitational force field
of the earth. We may specify the field by the magnitude and the direction of acceleration due
to gravity (i.e. force per unit mass ) g(x, y, z) at every point in space.
To give an example from electromagnetism, consider an electric potential of form 2x2 y, a
scalar field in 3D space. The corresponding conservative electric field can be computed as
the gradient of the electric potential function, and expressed as 4xy^i + 2x 2^
√ j. The magnitude of
this electric field can in turn be expressed as a scalar field of the form 4x4 + 16x2 y 2 .
Vector math and basic calculus operations with respect to vectors have already been elabo-
rated upon in other sections of this module’s documentation.
On the other hand, base scalars (or coordinate variables) are implemented as special SymPy
Symbol s assigned to every frame, one for each direction from X, Y and Z. For a frame R, the
X, Y and Z base scalar Symbol s can be accessed using the R[0], R[1] and R[2] expressions
respectively.
Therefore, to generate the expression for the aforementioned electric potential field 2x2 y, you
would have to do
In string representation, R_x denotes the X base scalar assigned to ReferenceFrame R. Es-
sentially, R_x is the string representation of R[0].
Scalar fields can be treated just as any other SymPy expression, for any math/calculus func-
tionality. Hence, to differentiate the above electric potential with respect to x (i.e. R[0]), you
would have to use the diff method.
Like vectors (and vector fields), scalar fields can also be re-expressed in other frames of
reference, apart from the one they were defined in – assuming that an orientation relationship
exists between the concerned frames. This can be done using the express method, in a way
similar to vectors - but with the variables parameter set to True.
Moreover, considering scalars can also be functions of time just as vectors, differentiation
with respect to time is also possible. Depending on the Symbol s present in the expression
and the frame with respect to which the time differentiation is being done, the output will
change/remain the same.
Curl
Divergence
Divergence is a vector operator that measures the magnitude of a vector field’s source or sink
at a given point, in terms of a signed scalar.
The divergence operator always returns a scalar after operating on a vector.
In the 3D Cartesian system, the divergence of a 3D vector F, denoted by ∇ · F is given by -
∇·F= ∂U
∂x + ∂V
∂y + ∂W
∂z
Gradient
Consider a scalar field f (x, y, z) in 3D space. The gradient of this field is defined as the vector of
the 3 partial derivatives of f with respect to x, y and z in the X, Y and Z directions respectively.
In the 3D Cartesian system, the gradient of a scalar field f , denoted by ∇f is given by -
∂f^ ∂f^ ∂f ^
∇f = ∂x i + ∂y j + ∂z k
In vector calculus, a conservative field is a field that is the gradient of some scalar field.
Conservative fields have the property that their line integral over any path depends only on
the end-points, and is independent of the path between them. A conservative vector field is
also said to be ‘irrotational’, since the curl of a conservative field is always zero.
In physics, conservative fields represent forces in physical systems where energy is con-
served.
To check if a vector field is conservative in physics.vector, use the is_conservative func-
tion.
A solenoidal field, on the other hand, is a vector field whose divergence is zero at all points
in space.
To check if a vector field is solenoidal in physics.vector, use the is_solenoidal function.
We have previously mentioned that every conservative field can be defined as the gradient of
some scalar field. This scalar field is also called the ‘scalar potential field’ corresponding to
the aforementioned conservative field.
The scalar_potential function in physics.vector calculates the scalar potential field cor-
responding to a given conservative vector field in 3D space - minus the extra constant of
integration, of course.
Example of usage -
>>> scalar_potential(conservative_field, R)
2*R_x**2*R_y*R_z
Vector API
Essential Classes
CoordinateSym
class sympy.physics.vector.frame.CoordinateSym
A coordinate symbol/base scalar associated wrt a Reference Frame.
Ideally, users should not instantiate this class. Instances of this class must only be ac-
cessed through the corresponding frame as ‘frame[index]’.
CoordinateSyms having the same frame and index parameters are equal (even though
they may be instantiated separately).
Parameters name : string
The display name of the CoordinateSym
frame : ReferenceFrame
The reference frame this base scalar belongs to
index : 0, 1 or 2
The index of the dimension denoted by this coordinate variable
Examples
ReferenceFrame
Examples
ang_vel_in(otherframe)
Returns the angular velocity Vector of the ReferenceFrame.
Effectively returns the Vector: ^N omega ^B which represent the angular velocity
of B in N, where B is self, and N is otherframe.
Parameters otherframe : ReferenceFrame
The ReferenceFrame which the angular velocity is returned in.
Examples
dcm(otherframe)
The direction cosine matrix between frames.
This gives the DCM between this frame and the otherframe. The format is N.xyz =
N.dcm(B) * B.xyz A SymPy Matrix is returned.
Parameters otherframe : ReferenceFrame
The otherframe which the DCM is generated to.
Examples
Examples
Now we have a choice of how to implement the orientation. First is Body. Body
orientation takes this reference frame through three successive simple rotations.
Acceptable rotation orders are of length 3, expressed in XYZ or 123, and cannot
have a rotation about about an axis twice in a row.
Next is Space. Space is like Body, but the rotations are applied in the opposite order.
Next is Quaternion. This orients the new ReferenceFrame with Quaternions, defined
as a finite rotation about lambda, a unit vector, by some amount theta. This orienta-
tion is described by four parameters: q0 = cos(theta/2) q1 = lambda_x sin(theta/2)
q2 = lambda_y sin(theta/2) q3 = lambda_z sin(theta/2) Quaternion does not take in
a rotation order.
Last is DCM (Direction Cosine Matrix). This is a rotation matrix given manually.
Examples
Now we have a choice of how to implement the orientation. First is Body. Body
orientation takes this reference frame through three successive simple rotations.
Acceptable rotation orders are of length 3, expressed in XYZ or 123, and cannot
have a rotation about about an axis twice in a row.
Next is Space. Space is like Body, but the rotations are applied in the opposite order.
Next is Quaternion. This orients the new ReferenceFrame with Quaternions, defined
as a finite rotation about lambda, a unit vector, by some amount theta. This orienta-
tion is described by four parameters: q0 = cos(theta/2) q1 = lambda_x sin(theta/2)
q2 = lambda_y sin(theta/2) q3 = lambda_z sin(theta/2) Quaternion does not take in
a rotation order.
partial_velocity(frame, *gen_speeds)
Returns the partial angular velocities of this frame in the given frame with respect
to one or more provided generalized speeds.
Parameters frame : ReferenceFrame
The frame with which the angular velocity is defined in.
gen_speeds : functions of time
The generalized speeds.
Returns partial_velocities : tuple of Vector
The partial angular velocity vectors corresponding to the provided
generalized speeds.
Examples
set_ang_acc(otherframe, value)
Define the angular acceleration Vector in a ReferenceFrame.
Defines the angular acceleration of this ReferenceFrame, in another. Angular ac-
celeration can be defined with respect to multiple different ReferenceFrames. Care
must be taken to not create loops which are inconsistent.
Parameters otherframe : ReferenceFrame
A ReferenceFrame to define the angular acceleration in
value : Vector
The Vector representing angular acceleration
Examples
set_ang_vel(otherframe, value)
Define the angular velocity vector in a ReferenceFrame.
Defines the angular velocity of this ReferenceFrame, in another. Angular velocity
can be defined with respect to multiple different ReferenceFrames. Care must be
taken to not create loops which are inconsistent.
Parameters otherframe : ReferenceFrame
A ReferenceFrame to define the angular velocity in
value : Vector
The Vector representing angular velocity
Examples
variable_map(otherframe)
Returns a dictionary which expresses the coordinate variables of this frame in terms
of the variables of otherframe.
If Vector.simp is True, returns a simplified version of the mapped values. Else, re-
turns them without simplification.
Simplification of the expressions may take time.
Parameters otherframe : ReferenceFrame
The other frame to map the variables to
Examples
x
The basis Vector for the ReferenceFrame, in the x direction.
y
The basis Vector for the ReferenceFrame, in the y direction.
z
The basis Vector for the ReferenceFrame, in the z direction.
Vector
class sympy.physics.vector.vector.Vector(inlist)
The class used to define vectors.
It along with ReferenceFrame are the building blocks of describing a classical mechanics
system in PyDy and sympy.physics.vector.
Attributes
applyfunc(f)
Apply a function to each component of a vector.
cross(other)
The cross product operator for two Vectors.
Returns a Vector, expressed in the same ReferenceFrames as self.
Parameters other : Vector
The Vector which we are crossing with
Examples
Examples
doit(**hints)
Calls .doit() on each term in the Vector
dot(other)
Dot product of two vectors.
Returns a scalar, the dot product of the two Vectors
Parameters other : Vector
The Vector which we are dotting with
Examples
dt(otherframe)
Returns a Vector which is the time derivative of the self Vector, taken in frame oth-
erframe.
Calls the global time_derivative method
Parameters otherframe : ReferenceFrame
The frame to calculate the time derivative in
express(otherframe, variables=False)
Returns a Vector equivalent to this one, expressed in otherframe. Uses the global
express method.
Parameters otherframe : ReferenceFrame
The frame for this Vector to be described in
variables : boolean
If True, the coordinate symbols(if present) in this Vector are re-
expressed in terms otherframe
Examples
free_symbols(reference_frame)
Returns the free symbols in the measure numbers of the vector expressed in the
given reference frame.
Parameter
reference_frame [ReferenceFrame] The frame with respect to which the free sym-
bols of the given vector is to be determined.
magnitude()
Returns the magnitude (Euclidean norm) of self.
normalize()
Returns a Vector of magnitude 1, codirectional with self.
outer(other)
Outer product between two Vectors.
A rank increasing operation, which returns a Dyadic from two Vectors
Parameters other : Vector
The Vector to take the outer product with
Examples
separate()
The constituents of this vector in different reference frames, as per its definition.
Returns a dict mapping each ReferenceFrame to the corresponding constituent Vec-
tor.
Examples
simplify()
Returns a simplified Vector.
subs(*args, **kwargs)
Substitution on the Vector.
Examples
to_matrix(reference_frame)
Returns the matrix form of the vector with respect to the given frame.
Parameters reference_frame : ReferenceFrame
The reference frame that the rows of the matrix correspond to.
Returns matrix : ImmutableMatrix, shape(3,1)
The matrix that gives the 1D vector.
Examples
Dyadic
class sympy.physics.vector.dyadic.Dyadic(inlist)
A Dyadic object.
See: http://en.wikipedia.org/wiki/Dyadic_tensor Kane, T., Levinson, D. Dynamics Theory
and Applications. 1985 McGraw-Hill
A more powerful way to represent a rigid body’s inertia. While it is more complex, by
choosing Dyadic components to be in body fixed basis vectors, the resulting matrix is
equivalent to the inertia tensor.
applyfunc(f)
Apply a function to each component of a Dyadic.
cross(other)
For a cross product in the form: Dyadic x Vector.
Parameters other : Vector
The Vector that we are crossing this Dyadic with
Examples
doit(**hints)
Calls .doit() on each term in the Dyadic
dot(other)
The inner product operator for a Dyadic and a Dyadic or Vector.
Parameters other : Dyadic or Vector
The other Dyadic or Vector to take the inner product with
Examples
dt(frame)
Take the time derivative of this Dyadic in a frame.
This function calls the global time_derivative method
Parameters frame : ReferenceFrame
The frame to take the time derivative in
Examples
express(frame1, frame2=None)
Expresses this Dyadic in alternate frame(s)
The first frame is the list side expression, the second frame is the right side; if Dyadic
is in form A.x|B.y, you can express it in two different frames. If no second frame is
given, the Dyadic is expressed in only one frame.
Calls the global express function
Parameters frame1 : ReferenceFrame
The frame to express the left side of the Dyadic in
frame2 : ReferenceFrame
If provided, the frame to express the right side of the Dyadic in
Examples
simplify()
Returns a simplified Dyadic.
subs(*args, **kwargs)
Substitution on the Dyadic.
Examples
to_matrix(reference_frame, second_reference_frame=None)
Returns the matrix form of the dyadic with respect to one or two reference frames.
Parameters reference_frame : ReferenceFrame
The reference frame that the rows and columns of the matrix corre-
spond to. If a second reference frame is provided, this only corre-
sponds to the rows of the matrix.
second_reference_frame : ReferenceFrame, optional, default=None
The reference frame that the columns of the matrix correspond to.
Returns matrix : ImmutableMatrix, shape(3,3)
The matrix that gives the 2D tensor form.
Examples
Kinematics (Docstrings)
Point
class sympy.physics.vector.point.Point(name)
This object represents a point in a dynamic system.
It stores the: position, velocity, and acceleration of a point. The position is a vector
defined as the vector distance from a parent point to this point.
a1pt_theory(otherpoint, outframe, interframe)
Sets the acceleration of this point with the 1-point theory.
The 1-point theory for point acceleration looks like this:
^N a^P = ^B a^P + ^N a^O + ^N alpha^B x r^OP + ^N omega^B x (^N
omega^B x r^OP) + 2 ^N omega^B x ^B v^P
Examples
Examples
acc(frame)
The acceleration Vector of this Point in a ReferenceFrame.
Parameters frame : ReferenceFrame
The frame in which the returned acceleration vector will be defined
in
Examples
locatenew(name, value)
Creates a new point with a position defined from this point.
Parameters name : str
The name for the new point
value : Vector
The position of the new point relative to this point
Examples
partial_velocity(frame, *gen_speeds)
Returns the partial velocities of the linear velocity vector of this point in the given
frame with respect to one or more provided generalized speeds.
Parameters frame : ReferenceFrame
The frame with which the velocity is defined in.
gen_speeds : functions of time
The generalized speeds.
Returns partial_velocities : tuple of Vector
Examples
pos_from(otherpoint)
Returns a Vector distance between this Point and the other Point.
Parameters otherpoint : Point
The otherpoint we are locating this one relative to
Examples
set_acc(frame, value)
Used to set the acceleration of this Point in a ReferenceFrame.
Parameters frame : ReferenceFrame
The frame in which this point’s acceleration is defined
value : Vector
The vector value of this point’s acceleration in the frame
Examples
set_pos(otherpoint, value)
Used to set the position of this point w.r.t. another point.
Parameters otherpoint : Point
The other point which this point’s location is defined relative to
value : Vector
The vector which defines the location of this point
Examples
set_vel(frame, value)
Sets the velocity Vector of this Point in a ReferenceFrame.
Parameters frame : ReferenceFrame
The frame in which this point’s velocity is defined
value : Vector
The vector value of this point’s velocity in the frame
Examples
Examples
Examples
vel(frame)
The velocity Vector of this Point in the ReferenceFrame.
Parameters frame : ReferenceFrame
The frame in which the returned velocity vector will be defined in
Examples
kinematic_equations
Examples
gen_speeds : iterable
An iterable of generalized speeds.
frame : ReferenceFrame
The reference frame that the partial derivatives are going to be taken
in.
Examples
sympy.physics.vector.functions.get_motion_params(frame, **kwargs)
Returns the three motion parameters - (acceleration, velocity, and position) as vectorial
functions of time in the given frame.
If a higher order differential function is provided, the lower order functions are used
as boundary conditions. For example, given the acceleration, the velocity and position
parameters are taken as boundary conditions.
The values of time at which the boundary conditions are specified are taken from
timevalue1(for position boundary condition) and timevalue2(for velocity boundary con-
dition).
If any of the boundary conditions are not provided, they are taken to be zero by default
(zero vectors, in case of vectorial inputs). If the boundary conditions are also functions
of time, they are converted to constants by substituting the time values in the dynamic-
symbols._t time Symbol.
This function can also be used for calculating rotational motion parameters. Have a look
at the Parameters and Examples for more clarity.
Parameters frame : ReferenceFrame
The frame to express the motion parameters in
acceleration : Vector
Acceleration of the object/frame as a function of time
velocity : Vector
Velocity as function of time or as boundary condition of velocity at
time = timevalue1
position : Vector
Velocity as function of time or as boundary condition of velocity at
time = timevalue1
timevalue1 : sympyfiable
Value of time for position boundary condition
timevalue2 : sympyfiable
Value of time for velocity boundary condition
Examples
Printing (Docstrings)
init_printing
sympy.physics.vector.printing.init_printing(pretty_print=True, or-
der=None, use_unicode=None,
use_latex=None, wrap_line=None,
num_columns=None,
no_global=False, ip=None, eu-
ler=False, forecolor=’Black’,
backcolor=’Transparent’,
fontsize=’10pt’, la-
tex_mode=’equation*’,
print_builtin=True,
str_printer=None,
pretty_printer=None, la-
tex_printer=None, **settings)
Initializes pretty-printer depending on the environment.
Parameters pretty_print: boolean
If True, use pretty_print to stringify or the provided pretty printer; if
False, use sstrrepr to stringify or the provided string printer.
order: string or None
There are a few different settings for this parameter: lex (default),
which is lexographic order; grlex, which is graded lexographic order;
grevlex, which is reversed graded lexographic order; old, which is
used for compatibility reasons and for long expressions; None, which
sets it to lex.
use_unicode: boolean or None
If True, use unicode characters; if False, do not use unicode charac-
ters.
use_latex: string, boolean, or None
If True, use default latex rendering in GUI interfaces (png and math-
jax); if False, do not use latex rendering; if ‘png’, enable latex ren-
dering with an external latex compiler, falling back to matplotlib if
external compilation fails; if ‘matplotlib’, enable latex rendering with
matplotlib; if ‘mathjax’, enable latex text generation, for example
MathJax rendering in IPython notebook or text rendering in LaTeX
documents
wrap_line: boolean
If True, lines will wrap at the end; if False, they will not wrap but
continue as one line. This is only relevant if prettyp rint is True.
num_columns: int or None
If int, number of columns before wrapping is set to num_columns; if
None, number of columns before wrapping is set to terminal width.
This is only relevant if prettyp rint is True.
no_global: boolean
If True, the settings become system wide; if False, use just for this
console/session.
ip: An interactive console
This can either be an instance of IPython, or a class that derives from
code.InteractiveConsole.
euler: boolean, optional, default=False
Loads the euler package in the LaTeX preamble for handwritten style
fonts (http://www.ctan.org/pkg/euler).
forecolor: string, optional, default=’Black’
DVI setting for foreground color.
backcolor: string, optional, default=’Transparent’
DVI setting for background color.
fontsize: string, optional, default=‘10pt’
A font size to pass to the LaTeX documentclass function in the pream-
ble.
latex_mode: string, optional, default=’equation*’
The mode used in the LaTeX printer. Can be one of: {‘in-
line’|’plain’|’equation’|’equation*’}.
Examples
vprint
sympy.physics.vector.printing.vprint(expr, **settings)
Function for printing of expressions generated in the sympy.physics vector package.
Extends SymPy’s StrPrinter, takes the same setting accepted by SymPy’s sstr(), and is
equivalent to print(sstr(f oo)).
Parameters expr : valid SymPy object
SymPy expression to print.
settings : args
Same as the settings accepted by SymPy’s sstr().
Examples
vpprint
sympy.physics.vector.printing.vpprint(expr, **settings)
Function for pretty printing of expressions generated in the sympy.physics vector pack-
age.
Mainly used for expressions not inside a vector; the output of running scripts and gen-
erating equations of motion. Takes the same options as SymPy’s pretty_print(); see that
function for more information.
Parameters expr : valid SymPy object
SymPy expression to pretty print
settings : args
Same as those accepted by SymPy’s pretty_print.
vlatex
sympy.physics.vector.printing.vlatex(expr, **settings)
Function for printing latex representation of sympy.physics.vector objects.
For latex representation of Vectors, Dyadics, and dynamicsymbols. Takes the same op-
tions as SymPy’s latex(); see that function for more information;
Parameters expr : valid SymPy object
SymPy expression to represent in LaTeX form
settings : args
Same as latex()
Examples
dynamicsymbols
sympy.physics.vector.dynamicsymbols(names, level=0)
Uses symbols and Function for functions of time.
Creates a SymPy UndefinedFunction, which is then initialized as a function of a variable,
the default being Symbol(‘t’).
Parameters names : str
Names of the dynamic symbols you want to create; works the same
way as inputs to symbols
level : int
Level of differentiation of the returned function; d/dt once of t, twice
of t, etc.
Examples
dot
sympy.physics.vector.functions.dot(vec1, vec2)
Dot product convenience wrapper for Vector.dot(): Dot product of two vectors.
Examples
cross
sympy.physics.vector.functions.cross(vec1, vec2)
Cross product convenience wrapper for Vector.cross(): The cross product operator for
two Vectors.
Returns a Vector, expressed in the same ReferenceFrames as self.
Examples
outer
sympy.physics.vector.functions.outer(vec1, vec2)
Outer product convenience wrapper for Vector.outer(): Outer product between two Vec-
tors.
Examples
express
Examples
time_derivative
References
http://en.wikipedia.org/wiki/Rotating_reference_frame#Time_derivatives_in_the_two_
frames
Examples
curl
sympy.physics.vector.fieldfunctions.curl(vect, frame)
Returns the curl of a vector field computed wrt the coordinate symbols of the given
frame.
Parameters vect : Vector
The vector operand
frame : ReferenceFrame
The reference frame to calculate the curl in
Examples
divergence
sympy.physics.vector.fieldfunctions.divergence(vect, frame)
Returns the divergence of a vector field computed wrt the coordinate symbols of the
given frame.
Parameters vect : Vector
The vector operand
frame : ReferenceFrame
The reference frame to calculate the divergence in
Examples
gradient
sympy.physics.vector.fieldfunctions.gradient(scalar, frame)
Returns the vector gradient of a scalar field computed wrt the coordinate symbols of the
given frame.
Parameters scalar : sympifiable
The scalar field to take the gradient of
frame : ReferenceFrame
The frame to calculate the gradient in
Examples
scalar_potential
sympy.physics.vector.fieldfunctions.scalar_potential(field, frame)
Returns the scalar potential function of a field in a given frame (without the added inte-
gration constant).
Parameters field : Vector
The vector field whose scalar potential function is to be calculated
frame : ReferenceFrame
The frame to do the calculation in
Examples
scalar_potential_difference
sympy.physics.vector.fieldfunctions.scalar_potential_difference(field,
frame,
point1,
point2,
origin)
Returns the scalar potential difference between two points in a certain frame, wrt a given
field.
If a scalar field is provided, its values at the two points are considered. If a conservative
vector field is provided, the values of its scalar potential function at the two points are
used.
Returns (potential at position 2) - (potential at position 1)
Parameters field : Vector/sympyfiable
The field to calculate wrt
frame : ReferenceFrame
The frame to do the calculations in
point1 : Point
The initial Point in given frame
position2 : Point
The second Point in the given frame
origin : Point
The Point to use as reference point for position vector calculation
Examples
is_conservative
sympy.physics.vector.fieldfunctions.is_conservative(field)
Checks if a field is conservative.
Parameters field : Vector
The field to check for conservative property
Examples
is_solenoidal
sympy.physics.vector.fieldfunctions.is_solenoidal(field)
Checks if a field is solenoidal.
Parameters field : Vector
The field to check for solenoidal property
Examples
Classical Mechanics
Abstract
Vector
This module derives the vector-related abilities and related functionalities from physics.
vector. Please have a look at the documentation of physics.vector and its necessary API
to understand the vector capabilities of mechanics.
Mechanics
In physics, mechanics describes conditions of rest (statics) or motion (dynamics). There are a
few common steps to all mechanics problems. First, an idealized representation of a system is
described. Next, we use physical laws to generate equations that define the system’s behavior.
Then, we solve these equations, sometimes analytically but usually numerically. Finally, we
extract information from these equations and solutions. The current scope of the module is
multi-body dynamics: the motion of systems of multiple particles and/or rigid bodies. For
example, this module could be used to understand the motion of a double pendulum, planets,
robotic manipulators, bicycles, and any other system of rigid bodies that may fascinate us.
Often, the objective in multi-body dynamics is to obtain the trajectory of a system of rigid
bodies through time. The challenge for this task is to first formulate the equations of motion
of the system. Once they are formulated, they must be solved, that is, integrated forward in
time. When digital computers came around, solving became the easy part of the problem.
Now, we can tackle more complicated problems, which leaves the challenge of formulating
the equations.
The term “equations of motion” is used to describe the application of Newton’s second law to
multi-body systems. The form of the equations of motion depends on the method used to gen-
erate them. This package implements two of these methods: Kane’s method and Lagrange’s
method. This module facilitates the formulation of equations of motion, which can then be
solved (integrated) using generic ordinary differential equation (ODE) solvers.
The approach to a particular class of dynamics problems, that of forward dynamics, has the
following steps:
1. describing the system’s geometry and configuration,
2. specifying the way the system can move, including constraints on its motion
3. describing the external forces and moments on the system,
4. combining the above information according to Newton’s second law (F = ma), and
5. organizing the resulting equations so that they can be integrated to obtain the system’s
trajectory through time.
Together with the rest of SymPy, this module performs steps 4 and 5, provided that the user
can perform 1 through 3 for the module. That is to say, the user must provide a complete
representation of the free body diagrams that themselves represent the system, with which
this code can provide equations of motion in a form amenable to numerical integration. Step
5 above amounts to arduous algebra for even fairly simple multi-body systems. Thus, it is
desirable to use a symbolic math package, such as Sympy, to perform this step. It is for
this reason that this module is a part of Sympy. Step 4 amounts to this specific module,
sympy.physics.mechanics.
Guide to Mechanics
This document will describe how to represent masses and inertias in mechanics and use of
the RigidBody and Particle classes.
It is assumed that the reader is familiar with the basics of these topics, such as finding the
center of mass for a system of particles, how to manipulate an inertia tensor, and the definition
of a particle and rigid body. Any advanced dynamics text can provide a reference for these
details.
Mass
The only requirement for a mass is that it needs to be a sympify-able expression. Keep in
mind that masses can be time varying.
Particle
Particles are created with the class Particle in mechanics. A Particle object has an asso-
ciated point and an associated mass which are the only two attributes of the object.:
The associated point contains the position, velocity and acceleration of the particle. mechan-
ics allows one to perform kinematic analysis of points separate from their association with
masses.
Inertia
See the Inertia (Dyadics) section in ‘Advanced Topics’ part of physics/vector docs.
Rigid Body
Rigid bodies are created in a similar fashion as particles. The RigidBody class generates
objects with four attributes: mass, center of mass, a reference frame, and an inertia tuple:
The mass is specified exactly as is in a particle. Similar to the Particle’s .point, the Rigid-
Body’s center of mass, .masscenter must be specified. The reference frame is stored in an
analogous fashion and holds information about the body’s orientation and angular velocity.
Finally, the inertia for a rigid body needs to be specified about a point. In mechanics, you are
allowed to specify any point for this. The most common is the center of mass, as shown in the
above code. If a point is selected which is not the center of mass, ensure that the position
between the point and the center of mass has been defined. The inertia is specified as a tuple
of length two with the first entry being a Dyadic and the second entry being a Point of which
the inertia dyadic is defined about.
Dyadic
ax ⊗ ^
^ ax = ^
ax^
ax
ax ⊗ ^
^ ay = ^
ax^
ay
Where ^ax^
ax and ^ax^
ay are the outer products obtained by multiplying the left side as a column
vector by the right side as a row vector. Note that the order is significant.
Some additional properties of a dyadic are:
or in dyadic form:
a11^
ax^
ax + a12^
ax^
ay + a13^
ax^
az + a21^
ay^
ax + a22^
ay^
ay + a23^
ay^
az + a31^
az^
ax + a32^
az^
ay + a33^
az^
az
Just as with vectors, the later representation makes it possible to keep track of which frames
the dyadic is defined with respect to. Also, the two components of each term in the dyadic
need not be in the same frame. The following is valid:
ax ⊗ ^
^ ax^
by = ^ by
Dyadics can also be crossed and dotted with vectors; again, order matters:
^ ax · ^
ax^ ax = ^
ax
^ ax · ^
ay^ ax = ^
ay
^ ay · ^
ax^ ax = 0
ax · ^
^ ax^
ax = ^
ax
ax · ^
^ ax^
ay = ^
ay
ax · ^
^ ay^
ax = 0
ax × ^
^ ay^
ax = ^
az^ax
ax × ^
^ ax^
ax = 0
^ ax × ^
ay^ az = −^ay^
ay
One can also take the time derivative of dyadics or express them in different frames, just like
with vectors.
Linear Momentum
Angular Momentum
where the angular momentum of the body about it’s mass center is:
∗
N
HB/B = I∗ · ω
and the angular momentum of the mass center about O is:
∗
N
HB /O
= r∗ × mv∗
where I∗ is the central inertia dyadic of rigid body B, ω is the inertial angular velocity of B,
r∗ is a position vector from point O to the mass center of B, m is the mass of B and v∗ is the
velocity of the mass center in the inertial frame.
The following example shows how to use the momenta functions in mechanics.
One begins by creating the requisite symbols to describe the system. Then the reference
frame is created and the kinematics are done.
Finally, the bodies that make up the system are created. In this case the system consists of a
particle Pa and a RigidBody A.
>> Pa = Particle('Pa', P, m)
>> I = outer(N.z, N.z)
>> A = RigidBody('A', Ac, a, M, (I, Ac))
Then one can either choose to evaluate the momenta of individual components of the system
or of the entire system itself.
>> linear_momentum(N,A)
M*l1*q1d*N.y
>> angular_momentum(O, N, Pa)
4*l1**2*m*q1d*N.z
>> linear_momentum(N, A, Pa)
(M*l1*q1d + 2*l1*m*q1d)*N.y
>> angular_momentum(O, N, A, Pa)
(4*l1**2*m*q1d + q1d)*N.z
It should be noted that the user can determine either momenta in any frame in mechanics as
the user is allowed to specify the reference frame when calling the function. In other words
the user is not limited to determining just inertial linear and angular momenta. Please refer
to the docstrings on each function to learn more about how each function works precisely.
Kinetic Energy
TB = Tt + Tr
Potential Energy
Potential energy is defined as the energy possessed by a body or system by virtue of its position
or arrangement.
Since there are a variety of definitions for potential energy, this is not discussed further here.
One can learn more about this in any elementary text book on dynamics.
Lagrangian
L=T −V
The following example shows how to use the energy functions in mechanics.
As was discussed above in the momenta functions, one first creates the system by going
through an identical procedure.
>> from sympy import symbols
>> from sympy.physics.mechanics import dynamicsymbols, ReferenceFrame, outer
>> from sympy.physics.mechanics import RigidBody, Particle, mechanics_printing
>> from symp.physics.mechanics import kinetic_energy, potential_energy, Point
>> mechanics_printing()
>> m, M, l1, g, h, H = symbols('m M l1 g h H')
>> omega = dynamicsymbols('omega')
>> N = ReferenceFrame('N')
>> O = Point('O')
>> O.set_vel(N, 0 * N.x)
>> Ac = O.locatenew('Ac', l1 * N.x)
>> P = Ac.locatenew('P', l1 * N.x)
>> a = ReferenceFrame('a')
>> a.set_ang_vel(N, omega * N.z)
(continues on next page)
The user can then determine the kinetic energy of any number of entities of the system:
>> kinetic_energy(N, Pa)
2*l1**2*m*q1d**2
>> kinetic_energy(N, Pa, A)
M*l1**2*q1d**2/2 + 2*l1**2*m*q1d**2 + q1d**2/2
It should be noted that the user can determine either kinetic energy relative to any frame in
mechanics as the user is allowed to specify the reference frame when calling the function. In
other words the user is not limited to determining just inertial kinetic energy.
For potential energies, the user must first specify the potential energy of every entity of the
system using the potential_energy property. The potential energy of any number of entities
comprising the system can then be determined:
>> Pa.potential_energy = m * g * h
>> A.potential_energy = M * g * H
>> potential_energy(A, Pa)
H*M*g + g*h*m
mechanics provides functionality for deriving equations of motion using Kane’s method
[Kane1985]. This document will describe Kane’s method as used in this module, but not
how the equations are actually derived.
Structure of Equations
In mechanics we are assuming there are 5 basic sets of equations needed to describe a system.
They are: holonomic constraints, non-holonomic constraints, kinematic differential equations,
dynamic equations, and differentiated non-holonomic equations.
fh (q, t) = 0
knh (q, t)u + fnh (q, t) = 0
kkq_ (q, t)q̇ + kku (q, t)u + fk (q, t) = 0
kd (q, t)u̇ + fd (q, q̇, u, t) = 0
kdnh (q, t)u̇ + fdnh (q, q̇, u, t) = 0
In mechanics holonomic constraints are only used for the linearization process; it is assumed
that they will be too complicated to solve for the dependent coordinate(s). If you are able to
easily solve a holonomic constraint, you should consider redefining your problem in terms of
a smaller set of coordinates. Alternatively, the time-differentiated holonomic constraints can
be supplied.
Kane’s method forms two expressions, Fr and Fr∗ , whose sum is zero. In this module, these
expressions are rearranged into the following form:
M(q, t)u̇ = f(q, q̇, u, t)
For a non-holonomic system with o total speeds and m motion constraints, we will get o - m
equations. The mass-matrix/forcing equations are then augmented in the following fashion:
[ ]
kd (q, t)
M(q, t) =
kdnh (q, t)
[ ]
−fd (q, q̇, u, t)
(q, q̇, u, t) =
(forcing)
−fdnh (q, q̇, u, t)
The formulation of the equations of motion in mechanics starts with creation of a KanesMethod
object. Upon initialization of the KanesMethod object, an inertial reference frame needs to be
supplied. along with some basic system information, such as coordinates and speeds
It is also important to supply the order of coordinates and speeds properly if there are de-
pendent coordinates and speeds. They must be supplied after independent coordinates and
speeds or as a keyword argument; this is shown later.
Additionally, if there are auxiliary speeds, they need to be identified here. See the examples
for more information on this. In this example u4 is the auxiliary speed.
Kinematic differential equations must also be supplied; there are to be provided as a list of
expressions which are each equal to zero. A trivial example follows:
>>> N = ReferenceFrame('N')
>>> q1, q2, q3, q4 = dynamicsymbols('q1 q2 q3 q4')
>>> q1d, q2d, q3d, q4d = dynamicsymbols('q1 q2 q3 q4', 1)
>>> u1, u2, u3, u4 = dynamicsymbols('u1 u2 u3 u4')
>>> #Here we will assume q2 is dependent, and u2 and u3 are dependent
>>> speed_cons = [u2 - u1, u3 - u1 - u4]
>>> coord_cons = [q2 - q1]
>>> q_ind = [q1, q3, q4]
>>> q_dep = [q2]
>>> u_ind = [u1, u4]
>>> u_dep = [u2, u3]
>>> kd = [q1d - u1, q2d - u2, q3d - u3, q4d - u4]
>>> KM = KanesMethod(N, q_ind, u_ind, kd,
... q_dependent=q_dep,
... configuration_constraints=coord_cons,
... u_dependent=u_dep,
... velocity_constraints=speed_cons)
The final step in forming the equations of motion is supplying a list of bodies and particles, and
a list of 2-tuples of the form (Point, Vector) or (ReferenceFrame, Vector) to represent
applied forces and torques.
>>> N = ReferenceFrame('N')
>>> q, u = dynamicsymbols('q u')
>>> qd, ud = dynamicsymbols('q u', 1)
>>> P = Point('P')
>>> P.set_vel(N, u * N.x)
>>> Pa = Particle('Pa', P, 5)
>>> BL = [Pa]
>>> FL = [(P, 7 * N.x)]
>>> KM = KanesMethod(N, [q], [u], [qd - u])
>>> (fr, frstar) = KM.kanes_equations(BL, FL)
>>> KM.mass_matrix
Matrix([[5]])
>>> KM.forcing
Matrix([[7]])
When there are motion constraints, the mass matrix is augmented by the kdnh (q, t) matrix, and
the forcing vector by the fdnh (q, q̇, u, t) vector.
There are also the “full” mass matrix and “full” forcing vector terms, these include the kine-
matic differential equations; the mass matrix is of size (n + o) x (n + o), or square and the
size of all coordinates and speeds.
>>> KM.mass_matrix_full
Matrix([
[1, 0],
[0, 5]])
>>> KM.forcing_full
Matrix([
[u],
[7]])
mechanics provides functionality for deriving equations of motion using Lagrange’s method.
This document will describe Lagrange’s method as used in this module, but not how the
equations are actually derived.
Structure of Equations
In mechanics we are assuming there are 3 basic sets of equations needed to describe a sys-
tem; the constraint equations, the time differentiated constraint equations and the dynamic
equations.
In this module, the expressions formed by using Lagrange’s equations of the second kind are
rearranged into the following form:
M(q, t)x = f(q, q̇, t)
where in the case of a system without constraints:
x = q̈
For a constrained system with n generalized speeds and m constraints, we will get n - m
equations. The mass-matrix/forcing equations are then augmented in the following fashion:
[ ]
q̈
x=
λ
[ ]
M(q, t) = md (q, t) c (q, t)
[ ]
F(q̇, q, t) = fd (q, q̇, t)
The formulation of the equations of motion in mechanics using Lagrange’s Method starts
with the creation of generalized coordinates and a Lagrangian. The Lagrangian can either
be created with the Lagrangian function or can be a user supplied function. In this case we
will supply the Lagrangian.
>>> mechanics_printing(pretty_print=False)
>>> LM.form_lagranges_equations()
Matrix([
[2*q1''],
[2*q2'']])
>>> LM.mass_matrix
Matrix([
[2, 0],
[0, 2]])
>>> LM.forcing
Matrix([
[0],
[0]])
If there are any holonomic or non-holonomic constraints, they must be supplied as keyword
arguments (hol_coneqs and nonhol_coneqs respectively) in a list of expressions which are
equal to zero. Modifying the example above, the equations of motion can then be generated:
When the equations of motion are generated in this case, the Lagrange multipliers are intro-
duced; they are represented by lam1 in this case. In general, there will be as many multipliers
as there are constraint equations.
>>> LM.form_lagranges_equations()
Matrix([
[ lam1 + 2*q1''],
[-lam1 + 2*q2'']])
Also in the case of systems with constraints, the ‘full’ mass matrix is augmented by the kdc (q, t)
matrix, and the forcing vector by the fdc (q, q̇, t) vector. The ‘full’ mass matrix is of size (2n +
o) x (2n + o), i.e. it’s a square matrix.
>>> LM.mass_matrix_full
Matrix([
[1, 0, 0, 0, 0],
[0, 1, 0, 0, 0],
[0, 0, 2, 0, -1],
[0, 0, 0, 2, 1],
[0, 0, 1, -1, 0]])
>>> LM.forcing_full
Matrix([
[q1'],
[q2'],
[ 0],
[ 0],
[ 0]])
If there are any non-conservative forces or moments acting on the system, they must also
This code will go over the manual input of the equations of motion for the simple pen-
dulum that uses the Cartesian location of the mass as the generalized coordinates into
SymbolicSystem.
The equations of motion are formed in the physics/mechanics/examples. In that spot the
variables q1 and q2 are used in place of x and y and the reference frame is rotated 90 degrees.
>>> from sympy import atan, symbols, Matrix
>>> from sympy.physics.mechanics import (dynamicsymbols, ReferenceFrame,
... Particle, Point)
>>> import sympy.physics.mechanics.system as system
The first step will be to initialize all of the dynamic and constant symbols.
>>> x, y, u, v, lam = dynamicsymbols('x y u v lambda')
>>> m, l, g = symbols('m l g')
where
x : states, e.g. [q, u] t : time r : specified (exogenous) inputs p : constants q :
generalized coordinates u : generalized speeds F_1 : right hand side of the com-
bined equations in explicit form F_2 : right hand side of the combined equations
in implicit form F_3 : right hand side of the dynamical equations in implicit form
M_2 : mass matrix of the combined equations in implicit form M_3 : mass matrix
of the dynamical equations in implicit form G : right hand side of the kinematical
differential equations
>>> dyn_implicit_mat = Matrix([[1, 0, -x/m],
... [0, 1, -y/m],
... [0, 0, l**2/m]])
>>> dyn_implicit_rhs = Matrix([0, 0, u**2 + v**2 - g*y])
>>> comb_implicit_mat = Matrix([[1, 0, 0, 0, 0],
... [0, 1, 0, 0, 0],
... [0, 0, 1, 0, -x/m],
... [0, 0, 0, 1, -y/m],
... [0, 0, 0, 0, l**2/m]])
>>> comb_implicit_rhs = Matrix([u, v, 0, 0, u**2 + v**2 - g*y])
>>> kin_explicit_rhs = Matrix([u, v])
>>> comb_explicit_rhs = comb_implicit_mat.LUsolve(comb_implicit_rhs)
Now the reference frames, points and particles will be set up so this information can be passed
into system.SymbolicSystem in the form of a bodies and loads iterable.
>>> theta = atan(x/y)
>>> omega = dynamicsymbols('omega')
>>> N = ReferenceFrame('N')
>>> A = N.orientnew('A', 'Axis', [theta, N.z])
>>> A.set_ang_vel(N, omega * N.z)
>>> O = Point('O')
>>> O.set_vel(N, 0)
>>> P = O.locatenew('P', l * A.x)
>>> P.v2pt_theory(O, N, A)
l*omega*A.y
>>> Pa = Particle('Pa', P, m)
The equations of motion are in the form of a differential algebraic equation (DAE) and DAE
solvers need to know which of the equations are the algebraic expressions. This information
is passed into SymbolicSystem as a list specifying which rows are the algebraic equations. In
this example it is a different row based on the chosen equations of motion format. The row
index should always correspond to the mass matrix that is being input to the SymbolicSystem
class but will always correspond to the row index of the combined dynamics and kinematics
when being accessed from the SymbolicSystem class.
>>> alg_con = [2]
>>> alg_con_full = [4]
An iterable containing the states now needs to be created for the system. The SymbolicSystem
class can determine which of the states are considered coordinates or speeds by passing in
the indexes of the coordinates and speeds. If these indexes are not passed in the object will
not be able to differentiate between coordinates and speeds.
Now the equations of motion instances can be created using the above mentioned equations
of motion formats.
Like coordinates and speeds, the bodies and loads attributes can only be
accessed if they are specified during initialization of the `SymbolicSystem`
class. Lastly here are some attributes accessible from the `SymbolicSystem`
class. ::
>>> symsystem1.states
Matrix([
[ x(t)],
[ y(t)],
[ u(t)],
[ v(t)],
[lambda(t)]])
>>> symsystem2.coordinates
Matrix([
[x(t)],
[y(t)]])
>>> symsystem3.speeds
Matrix([
[u(t)],
[v(t)]])
>>> symsystem1.comb_explicit_rhs
Matrix([
[ u(t)],
[ v(t)],
[(-g*y(t) + u(t)**2 + v(t)**2)*x(t)/l**2],
[(-g*y(t) + u(t)**2 + v(t)**2)*y(t)/l**2],
[ m*(-g*y(t) + u(t)**2 + v(t)**2)/l**2]])
>>> symsystem2.comb_implicit_rhs
Matrix([
[ u(t)],
[ v(t)],
[ 0],
[ 0],
[-g*y(t) + u(t)**2 + v(t)**2]])
>>> symsystem2.comb_implicit_mat
Matrix([
(continues on next page)
Linearization in Physics/Mechanics
mechanics includes methods for linearizing the generated equations of motion (EOM) about
an operating point (also known as the trim condition). Note that this operating point doesn’t
have to be an equilibrium position, it just needs to satisfy the equations of motion.
Linearization is accomplished by taking the first order Taylor expansion of the EOM about
the operating point. When there are no dependent coordinates or speeds this is simply the
jacobian of the right hand side about q and u. However, in the presence of constraints more
care needs to be taken. The linearization methods provided here handle these constraints
correctly.
Background
In mechanics we assume all systems can be represented in the following general form:
fc (q, t) = 0l×1
fv (q, u, t) = 0m×1
fa (q, q̇, u, u̇, t) = 0m×1
f0 (q, q̇, t) + f1 (q, u, t) = 0n×1
f2 (q, u, u̇, t) + f3 (q, q̇, u, r, t) + f4 (q, λ, t) = 0(o−m+k)×1
where
q, q̇ ∈ Rn
u, u̇ ∈ Ro
r ∈ Rs
λ ∈ Rk
In this form,
• fc represents the configuration constraint equations
• fv represents the velocity constraint equations
• fa represents the acceleration constraint equations
• f0 and f1 form the kinematic differential equations
• f2 , f3 , and f4 form the dynamic differential equations
• q and q̇ are the generalized coordinates and their derivatives
• u and u̇ are the generalized speeds and their derivatives
• r is the system inputs
• λ is the Lagrange multipliers
This generalized form is held inside the Linearizer class, which performs the actual lin-
earization. Both KanesMethod and LagrangesMethod objects have methods for forming the
linearizer using the to_linearizer class method.
If the system being linearized contains constraint equations, this results in not all gener-
alized coordinates being independent (i.e. q1 may depend on q2 ). With l configuration con-
straints, and m velocity constraints, there are l dependent coordinates and m dependent
speeds.
In general, you may pick any of the coordinates and speeds to be dependent, but in practice
some choices may result in undesirable singularites. Methods for deciding which coordi-
nates/speeds to make dependent is behind the scope of this guide. For more information,
please see [Blajer1994].
Once the system is coerced into the generalized form, the linearized EOM can be solved for.
The methods provided in mechanics allow for two different forms of the linearized EOM:
M , A, and B In this form, the forcing matrix is linearized into two separate matrices A and
B. This is the default form of the linearized EOM. The resulting equations are:
δ q̇ [ ]
δqi [ ]
M δ u̇ = A + B δr
δui
δλ
where
M ∈ R(n+o+k)×(n+o+k)
A ∈ R(n+o+k)×(n−l+o−m)
B ∈ R(n+o+k)×s
Note that qi and ui are just the independent coordinates and speeds, while q and u con-
tains both the independent and dependent coordinates and speeds.
A and B In this form, the linearized EOM are brought into explicit first order form, in terms of
just the independent coordinates and speeds. This form is often used in stability analysis
or control theory. The resulting equations are:
[ ] [ ]
δ q˙i δqi [ ]
=A + B δr
δ u̇i δui
where
A ∈ R(n−l+o−m)×(n−l+o−m)
B ∈ R(n−l+o−m)×s
After initializing the KanesMethod object and forming Fr and Fr∗ using the kanes_equations
class method, linearization can be accomplished in a couple ways. The different methods will
be demonstrated with a simple pendulum system:
A linearizer object can be created using the to_linearizer class method. This coerces the
representation found in the KanesMethod object into the generalized form described above.
As the independent and dependent coordinates and speeds are specified upon creation of the
KanesMethod object, there is no need to specify them here.
>>> linearizer = KM.to_linearizer()
The linearized EOM can then be formed with the linearize method of the Linearizer object:
>>> M, A, B = linearizer.linearize()
>>> M
Matrix([
[1, 0],
[0, -L**2*m]])
>>> A
Matrix([
[ 0, 1],
[L*g*m*cos(q1(t)), 0]])
>>> B
Matrix(0, 0, [])
Note that the same effect can be had by applying msubs to the matrices generated without
the op_point kwarg:
>>> assert msubs(A, op_point) == A_op
Sometimes the returned matrices may not be in the most simplified form. Simplification can
be performed after the fact, or the Linearizer object can be made to perform simplification
internally by setting the simplify kwarg to True.
The linearize method of the KanesMethod class is provided as a nice wrapper that calls
to_linearizer internally, performs the linearization, and returns the result. Note that all
the kwargs available in the linearize method described above are also available here:
The additional output inp_vec is a vector containing all found dynamicsymbols not included
in the generalized coordinate or speed vectors. These are assumed to be inputs to the system,
forming the r vector described in the background above. In this example there are no inputs,
so the vector is empty:
>>> inp_vec
Matrix(0, 0, [])
Previous releases of SymPy contained a linearization method for KanesM ethod objects. This
method is deprecated, and will be removed from future releases. Until then, you must set
new_method=True in all calls to KanesMethod.linearize. After the old method is removed,
this kwarg will no longer be needed.
Linearization of Lagrange’s equations proceeds much the same as that of Kane’s equations.
As before, the process will be demonstrated with a simple pendulum system:
A Linearizer object can be formed from a LagrangesMethod object using the to_linearizer
class method. The only difference between this process and that of the KanesMethod class is
that the LagrangesMethod object doesn’t already have its independent and dependent coordi-
nates and speeds specified internally. These must be specified in the call to to_linearizer.
In this example there are no dependent coordinates and speeds, but if there were they would
be included in the q_dep and qd_dep kwargs:
Once in this form, everything is the same as it was before with the KanesMethod example:
>>> A
Matrix([
[ 0, 1],
[-g/L, 0]])
Potential Issues
While the Linearizer class should be able to linearize all systems, there are some potential
issues that could occur. These are discussed below, along with some troubleshooting tips for
solving them.
This could be due to a number of things, but the most likely one is that solving a large linear
system symbolically is an expensive operation. Specifying an operating point will reduce
the expression size and speed this up. If a purely symbolic solution is desired though (for
application of many operating points at a later period, for example) a way to get around
this is to evaluate with A_and_B=False, and then solve manually after applying the operating
point:
>>> M, A, B = linearizer.linearize()
>>> M_op = msubs(M, op_point)
>>> A_op = msubs(A, op_point)
>>> perm_mat = linearizer.perm_mat
>>> A_lin = perm_mat.T * M_op.LUsolve(A_op)
>>> A_lin
Matrix([
[ 0, 1],
[-g/L, 0]])
The fewer symbols in A and M before solving, the faster this solution will be. Thus, for large
expressions, it may be to your benefit to delay conversion to the A and B form until most
symbols are subbed in for their numeric values.
There are two potential causes for this. The first (and the one you should check first) is that
some choices of dependent coordinates will result in singularities at certain operating points.
Coordinate partitioning in a systemic manner to avoid this is beyond the scope of this guide;
see [Blajer1994] for more information.
The other potential cause for this is that the matrices may not have been in the most reduced
form before the operating point was substituted in. A simple example of this behavior is:
Note that if this expression was simplified before substitution, the correct value results:
>>> expr.simplify().subs(op_point)
1
A good way of avoiding this hasn’t been found yet. For expressions of reasonable size, using
msubs with smart=True will apply an algorithm that tries to avoid these conditions. For large
expressions though this is extremely time consuming.
Further Examples
The pendulum example used above was simple, but didn’t include any dependent coordinates
or speeds. For a more thorough example, the same pendulum was linearized with dependent
coordinates using both Kane’s and Lagrange’s methods:
In this example we demonstrate the use of the functionality provided in mechanics for deriving
the equations of motion (EOM) for a pendulum with a nonminimal set of coordinates. As the
pendulum is a one degree of freedom system, it can be described using one coordinate and
one speed (the pendulum angle, and the angular velocity respectively). Choosing instead to
describe the system using the x and y coordinates of the mass results in a need for constraints.
The system is shown below:
Ay
Ny
Ax
Nx g
q1 L
q2 u2
m
u1
The system will be modeled using both Kane’s and Lagrange’s methods, and the resulting
EOM linearized. While this is a simple problem, it should illustrate the use of the linearization
methods in the presence of constraints.
Kane’s Method
First we need to create the dynamicsymbols needed to describe the system as shown in the
above diagram. In this case, the generalized coordinates q1 and q2 represent the mass x and
y coordinates in the inertial N frame. Likewise, the generalized speeds u1 and u2 represent
the velocities in these directions. We also create some symbols to represent the length and
mass of the pendulum, as well as gravity and time.
Next, we create a world coordinate frame N , and its origin point N ∗ . The velocity of the origin
is set to 0. A second coordinate frame A is oriented such that its x-axis is along the pendulum
Locating the pendulum mass is then as easy as specifying its location with in terms of its x
and y coordinates in the world frame. A Particle object is then created to represent the
mass at this location.
The kinematic differential equations (KDEs) relate the derivatives of the generalized coor-
dinates to the generalized speeds. In this case the speeds are the derivatives, so these are
simple. A dictionary is also created to map q̇ to u:
The velocity of the mass is then the time derivative of the position from the origin N ∗ :
As this system has more coordinates than degrees of freedom, constraints are needed. The
configuration constraints relate the coordinates to each other. In this case the constraint is
that the distance from the origin to the mass is always the length L (the pendulum doesn’t
get longer). Likewise, the velocity constraint is that the mass velocity in the A.x direction is
always 0 (no radial velocity).
With the problem setup, the equations of motion can be generated using the KanesMethod
class. As there are constraints, dependent and independent coordinates need to be provided
to the class. In this case we’ll use q2 and u2 as the independent coordinates and speeds:
For linearization, operating points can be specified on the call, or be substituted in after-
wards. In this case we’ll provide them in the call, supplied in a list. The A_and_B=True kwarg
indicates to solve invert the M matrix and solve for just the explicit linearized A and B ma-
trices. The simplify=True kwarg indicates to simplify inside the linearize call, and return
the presimplified matrices. The cost of doing this is small for simple systems, but for larger
systems this can be a costly operation, and should be avoided.
The resulting A matrix has dimensions 2 x 2, while the number of total states is len(q) +
len(u) = 2 + 2 = 4. This is because for constrained systems the resulting A_and_B form has
a partitioned state vector only containing the independent coordinates and speeds. Written
out mathematically, the system linearized about this point would be written as:
[ ] [ ][ ]
q˙2 0 1 q2
= −g
u˙2 L 0 u2
Lagrange’s Method
The derivation using Lagrange’s method is very similar to the approach using Kane’s method
described above. As before, we first create the dynamicsymbols needed to describe the sys-
tem. In this case, the generalized coordinates q1 and q2 represent the mass x and y coordinates
in the inertial N frame. This results in the time derivatives q˙1 and q˙2 representing the veloc-
ities in these directions. We also create some symbols to represent the length and mass of
the pendulum, as well as gravity and time.
Next, we create a world coordinate frame N , and its origin point N ∗ . The velocity of the origin
is set to 0. A second coordinate frame A is oriented such that its x-axis is along the pendulum
(as shown in the diagram above).
Locating the pendulum mass is then as easy as specifying its location with in terms of its x
and y coordinates in the world frame. A Particle object is then created to represent the
mass at this location.
As this system has more coordinates than degrees of freedom, constraints are needed. In this
case only a single holonomic constraints is needed: the distance from the origin to the mass
is always the length L (the pendulum doesn’t get longer).
With the problem setup, the Lagrangian can be calculated, and the equations of motion
formed. Note that the call to LagrangesMethod includes the Lagrangian, the generalized
coordinates, the constraints (specified by hol_coneqs or nonhol_coneqs), the list of (body,
force) pairs, and the inertial frame. In contrast to the KanesMethod initializer, independent
and dependent coordinates are not partitioned inside the LagrangesMethod object. Such a
partition is supplied later.
Next, we compose the operating point dictionary, set in the hanging at rest position:
As there are constraints in the formulation, there will be corresponding Lagrange Multipliers.
These may appear inside the linearized form as well, and thus should also be included inside
the operating point dictionary. Fortunately, the LagrangesMethod class provides an easy way
of solving for the multipliers at a given operating point using the solve_multipliers method.
With this solution, linearization can be completed. Note that in contrast to the KanesMethod
approach, the LagrangesMethod.linearize method also requires the partitioning of the gen-
eralized coordinates and their time derivatives into independent and dependent vectors. This
is the same as what was passed into the KanesMethod constructor above:
>>> op_point.update(lam_op)
>>> # Perform the Linearization
>>> A, B, inp_vec = LM.linearize([q2], [q2d], [q1], [q1d],
... op_point=op_point, A_and_B=True)
>>> A
(continues on next page)
The resulting A matrix has dimensions 2 x 2, while the number of total states is 2*len(q)
= 4. This is because for constrained systems the resulting A_and_B form has a partitioned
state vector only containing the independent coordinates and their derivatives. Written out
mathematically, the system linearized about this point would be written as:
[ ] [ ][ ]
q˙2 0 1 q2
= −g
q¨2 L 0 q˙2
Here are some examples that illustrate how one typically uses this module. We have ordered
the examples roughly according to increasing difficulty. If you have used this module to do
something others might find useful or interesting, consider adding it here!
A rolling disc
The disc is assumed to be infinitely thin, in contact with the ground at only 1 point, and it is
rolling without slip on the ground. See the image below.
nz
N ny
R
rz
ry
rx
nx
We model the rolling disc in three different ways, to show more of the functionality of this
module.
Here the definition of the rolling disc’s kinematics is formed from the contact point up, remov-
ing the need to introduce generalized speeds. Only 3 configuration and three speed variables
are need to describe this system, along with the disc’s mass and radius, and the local gravity
(note that mass will drop out).
The kinematics are formed by a series of simple rotations. Each simple rotation creates a new
frame, and the next rotation is defined by the new frame’s basis vectors. This example uses a
3-1-2 series of rotations, or Z, X, Y series of rotations. Angular velocity for this is defined using
the second frame’s basis (the lean frame); it is for this reason that we defined intermediate
frames, rather than using a body-three orientation.
>>> N = ReferenceFrame('N')
>>> Y = N.orientnew('Y', 'Axis', [q1, N.z])
>>> L = Y.orientnew('L', 'Axis', [q2, Y.x])
>>> R = L.orientnew('R', 'Axis', [q3, L.y])
>>> w_R_N_qd = R.ang_vel_in(N)
>>> R.set_ang_vel(N, u1 * L.x + u2 * L.y + u3 * L.z)
This is the translational kinematics. We create a point with no velocity in N; this is the contact
point between the disc and ground. Next we form the position vector from the contact point
to the disc’s center of mass. Finally we form the velocity and acceleration of the disc.
>>> C = Point('C')
>>> C.set_vel(N, 0)
>>> Dmc = C.locatenew('Dmc', r * L.z)
>>> Dmc.v2pt_theory(C, N, R)
r*u2*L.x - r*u1*L.y
This is a simple way to form the inertia dyadic. The inertia of the disc does not change within
the lean frame as the disc rolls; this will make for simpler equations in the end.
Kinematic differential equations; how the generalized coordinate time derivatives relate to
generalized speeds.
Creation of the force list; it is the gravitational force at the center of mass of the disc. Then
we create the disc by assigning a Point to the center of mass attribute, a ReferenceFrame to
the frame attribute, and mass and inertia. Then we form the body list.
Finally we form the equations of motion, using the same steps we did before. Specify inertial
frame, supply generalized coordinates and speeds, supply kinematic differential equation dic-
tionary, compute Fr from the force list and Fr* from the body list, compute the mass matrix
and forcing terms, then solve for the u dots (time derivatives of the generalized speeds).
>>> KM = KanesMethod(N, q_ind=[q1, q2, q3], u_ind=[u1, u2, u3], kd_eqs=kd)
>>> (fr, frstar) = KM.kanes_equations(BodyList, ForceList)
>>> MM = KM.mass_matrix
>>> forcing = KM.forcing
>>> rhs = MM.inv() * forcing
>>> kdd = KM.kindiffdict()
>>> rhs = rhs.subs(kdd)
>>> rhs.simplify()
>>> mprint(rhs)
Matrix([
[(4*g*sin(q2) + 6*r*u2*u3 - r*u3**2*tan(q2))/(5*r)],
[ -2*u1*u3/3],
[ (-2*u2 + u3*tan(q2))*u1]])
We will now revisit the rolling disc example, except this time we are bringing the non-
contributing (constraint) forces into evidence. See [Kane1985] for a more thorough expla-
nation of this. Here, we will turn on the automatic simplifcation done when doing vector
operations. It makes the outputs nicer for small problems, but can cause larger vector oper-
ations to hang.
>>> from sympy import symbols, sin, cos, tan
>>> from sympy.physics.mechanics import *
>>> mechanics_printing(pretty_print=False)
>>> q1, q2, q3, u1, u2, u3 = dynamicsymbols('q1 q2 q3 u1 u2 u3')
>>> q1d, q2d, q3d, u1d, u2d, u3d = dynamicsymbols('q1 q2 q3 u1 u2 u3', 1)
>>> r, m, g = symbols('r m g')
These two lines introduce the extra quantities needed to find the constraint forces.
>>> u4, u5, u6, f1, f2, f3 = dynamicsymbols('u4 u5 u6 f1 f2 f3')
The definition of rolling without slip necessitates that the velocity of the contact point is zero;
as part of bringing the constraint forces into evidence, we have to introduce speeds at this
point, which will by definition always be zero. They are normal to the ground, along the path
which the disc is rolling, and along the ground in a perpendicular direction.
>>> C = Point('C')
>>> C.set_vel(N, u4 * L.x + u5 * (Y.z ^ L.x) + u6 * Y.z)
>>> Dmc = C.locatenew('Dmc', r * L.z)
(continues on next page)
Just as we previously introduced three speeds as part of this process, we also introduce three
forces; they are in the same direction as the speeds, and represent the constraint forces in
those directions.
Here the rolling disc is formed from the contact point up, removing the need to introduce
generalized speeds. Only 3 configuration and 3 speed variables are needed to describe this
system, along with the disc’s mass and radius, and the local gravity.
The kinematics are formed by a series of simple rotations. Each simple rotation creates a new
frame, and the next rotation is defined by the new frame’s basis vectors. This example uses
a 3-1-2 series of rotations, or Z, X, Y series of rotations. Angular velocity for this is defined
using the second frame’s basis (the lean frame).
>>> N = ReferenceFrame('N')
>>> Y = N.orientnew('Y', 'Axis', [q1, N.z])
>>> L = Y.orientnew('L', 'Axis', [q2, Y.x])
>>> R = L.orientnew('R', 'Axis', [q3, L.y])
This is the translational kinematics. We create a point with no velocity in N; this is the contact
point between the disc and ground. Next we form the position vector from the contact point
to the disc’s center of mass. Finally we form the velocity and acceleration of the disc.
>>> C = Point('C')
>>> C.set_vel(N, 0)
>>> Dmc = C.locatenew('Dmc', r * L.z)
>>> Dmc.v2pt_theory(C, N, R)
r*(sin(q2)*q1' + q3')*L.x - r*q2'*L.y
We then set the potential energy and determine the Lagrangian of the rolling disc.
>>> BodyD.potential_energy = - m * g * r * cos(q2)
>>> Lag = Lagrangian(N, BodyD)
Then the equations of motion are generated by initializing the LagrangesMethod object. Fi-
nally we solve for the generalized accelerations(q double dots) with the rhs method.
>>> q = [q1, q2, q3]
>>> l = LagrangesMethod(Lag, q)
>>> le = l.form_lagranges_equations()
>>> le = le.simplify()
>>> le
Matrix([
[m*r**2*(6*sin(q2)*q3'' + 5*sin(2*q2)*q1'*q2' + 6*cos(q2)*q2'*q3' - 5*cos(2*q2)*q1''/
,→2 + 7*q1''/2)/4],
[ 3*m*r**2*(sin(q2)*q1'' + cos(q2)*q1
,→'*q2' + q3'')/2]])
A bicycle
The bicycle is an interesting system in that it has multiple rigid bodies, non-holonomic con-
straints, and a holonomic constraint. The linearized equations of motion are presented in
Note that this code has been crudely ported from Autolev, which is the reason for some of the
unusual naming conventions. It was purposefully as similar as possible in order to aid initial
porting & debugging. We set Vector.simp to False (in case it has been set True elsewhere),
since it slows down the computations:
>>> Vector.simp = False
>>> mechanics_printing(pretty_print=False)
Declaration of Coordinates & Speeds: A simple definition for qdots, qd = u,is used in this
code. Speeds are: yaw frame ang. rate, roll frame ang. rate, rear wheel frame ang. rate
(spinning motion), frame ang. rate (pitching motion), steering frame ang. rate, and front
wheel ang. rate (spinning motion). Wheel positions are ignorable coordinates, so they are
not introduced.
>>> q1, q2, q3, q4, q5 = dynamicsymbols('q1 q2 q3 q4 q5')
>>> q1d, q2d, q4d, q5d = dynamicsymbols('q1 q2 q4 q5', 1)
>>> u1, u2, u3, u4, u5, u6 = dynamicsymbols('u1 u2 u3 u4 u5 u6')
>>> u1d, u2d, u3d, u4d, u5d, u6d = dynamicsymbols('u1 u2 u3 u4 u5 u6', 1)
Set up reference frames for the system: N - inertial Y - yaw R - roll WR - rear wheel, rotation
angle is ignorable coordinate so not oriented Frame - bicycle frame TempFrame - statically
rotated frame for easier reference inertia definition Fork - bicycle fork TempFork - statically
rotated frame for easier reference inertia definition WF - front wheel, again posses an ignor-
able coordinate
>>> N = ReferenceFrame('N')
>>> Y = N.orientnew('Y', 'Axis', [q1, N.z])
>>> R = Y.orientnew('R', 'Axis', [q2, Y.x])
>>> Frame = R.orientnew('Frame', 'Axis', [q4 + htangle, R.y])
>>> WR = ReferenceFrame('WR')
>>> TempFrame = Frame.orientnew('TempFrame', 'Axis', [-htangle, Frame.y])
>>> Fork = Frame.orientnew('Fork', 'Axis', [q5, Frame.x])
>>> TempFork = Fork.orientnew('TempFork', 'Axis', [-htangle, Fork.y])
>>> WF = ReferenceFrame('WF')
Kinematics of the Bicycle: First block of code is forming the positions of the relevant points
rear wheel contact -> rear wheel’s center of mass -> frame’s center of mass + frame/fork
connection -> fork’s center of mass + front wheel’s center of mass -> front wheel contact
point.
Set the angular velocity of each frame: Angular accelerations end up being calculated auto-
matically by differentiating the angular velocities when first needed. :: u1 is yaw rate u2 is
roll rate u3 is rear wheel rate u4 is frame pitch rate u5 is fork steer rate u6 is front wheel
rate
Form the velocities of the points, using the 2-point theorem. Accelerations again are calcu-
lated automatically when first needed.
>>> WR_cont.set_vel(N, 0)
>>> WR_mc.v2pt_theory(WR_cont, N, WR)
WRrad*(u1*sin(q2) + u3 + u4)*R.x - WRrad*u2*R.y
>>> Steer.v2pt_theory(WR_mc, N, Frame)
WRrad*(u1*sin(q2) + u3 + u4)*R.x - WRrad*u2*R.y + framelength*(u1*sin(q2) + u4)*Frame.
,→x - framelength*(-u1*sin(htangle + q4)*cos(q2) + u2*cos(htangle + q4))*Frame.y
,→framecg1*(u1*sin(q2) + u4)*Frame.z
Sets the inertias of each body. Uses the inertia frame to construct the inertia dyadics. Wheel
inertias are only defined by principal moments of inertia, and are in fact constant in the frame
and fork reference frames; it is for this reason that the orientations of the wheels does not
need to be defined. The frame and fork inertias are defined in the ‘Temp’ frames which are
fixed to the appropriate body frames; this is to allow easier input of the reference values of
the benchmark paper. Note that due to slightly different orientations, the products of inertia
need to have their signs flipped; this is done later when entering the numerical value.
The kinematic differential equations; they are defined quite simply. Each entry in this list is
equal to zero.
The nonholonomic constraints are the velocity of the front wheel contact point dotted into the
X, Y, and Z directions; the yaw frame is used as it is “closer” to the front wheel (1 less DCM
connecting them). These constraints force the velocity of the front wheel contact point to be
0 in the inertial frame; the X and Y direction constraints enforce a “no-slip” condition, and the
Z direction constraint forces the front wheel contact point to not move away from the ground
frame, essentially replicating the holonomic constraint which does not allow the frame pitch
to change in an invalid fashion.
>>> conlist_speed = [WF_cont.vel(N) & Y.x,
... WF_cont.vel(N) & Y.y,
... WF_cont.vel(N) & Y.z]
The holonomic constraint is that the position from the rear wheel contact point to the front
wheel contact point when dotted into the normal-to-ground plane direction must be zero;
effectively that the front and rear wheel contact points are always touching the ground plane.
This is actually not part of the dynamic equations, but instead is necessary for the linearization
process.
>>> conlist_coord = [WF_cont.pos_from(WR_cont) & Y.z]
The force list; each body has the appropriate gravitational force applied at its center of mass.
>>> FL = [(Frame_mc, -mframe * g * Y.z), (Fork_mc, -mfork * g * Y.z),
... (WF_mc, -mwf * g * Y.z), (WR_mc, -mwr * g * Y.z)]
>>> BL = [BodyFrame, BodyFork, BodyWR, BodyWF]
The N frame is the inertial frame, coordinates are supplied in the order of independent, de-
pendent coordinates. The kinematic differential equations are also entered here. Here the
independent speeds are specified, followed by the dependent speeds, along with the non-
holonomic constraints. The dependent coordinate is also provided, with the holonomic con-
straint. Again, this is only comes into play in the linearization process, but is necessary for
the linearization to correctly work.
>>> KM = KanesMethod(N, q_ind=[q1, q2, q5],
... q_dependent=[q4], configuration_constraints=conlist_coord,
... u_ind=[u2, u3, u5],
... u_dependent=[u1, u4, u6], velocity_constraints=conlist_speed,
... kd_eqs=kd)
>>> print('Before Forming Generalized Active and Inertia Forces, Fr and Fr*')
Before Forming Generalized Active and Inertia Forces, Fr and Fr*
>>> (fr, frstar) = KM.kanes_equations(BL, FL)
>>> print('Base Equations of Motion Computed')
Base Equations of Motion Computed
This is the start of entering in the numerical values from the benchmark paper to validate the
eigenvalues of the linearized equations from this model to the reference eigenvalues. Look
at the aforementioned paper for more information. Some of these are intermediate values,
used to transform values from the paper into the coordinate systems used in this model.
>>> PaperRadRear = 0.3
>>> PaperRadFront = 0.35
>>> HTA = evalf.N(pi/2-pi/10)
>>> TrailPaper = 0.08
>>> rake = evalf.N(-(TrailPaper*sin(HTA)-(PaperRadFront*cos(HTA))))
>>> PaperWb = 1.02
>>> PaperFrameCgX = 0.3
>>> PaperFrameCgZ = 0.9
>>> PaperForkCgX = 0.9
>>> PaperForkCgZ = 0.7
>>> FrameLength = evalf.N(PaperWb*sin(HTA) - (rake - \
... (PaperRadFront - PaperRadRear)*cos(HTA)))
>>> FrameCGNorm = evalf.N((PaperFrameCgZ - PaperRadRear - \
(continues on next page)
Here is the final assembly of the numerical values. The symbol ‘v’ is the forward speed of the
bicycle (a concept which only makes sense in the upright, static equilibrium case?). These
are in a dictionary which will later be substituted in. Again the sign on the product of inertia
values is flipped here, due to different orientations of coordinate systems.
>>> v = Symbol('v')
>>> val_dict = {
... WFrad: PaperRadFront,
... WRrad: PaperRadRear,
... htangle: HTA,
... forkoffset: rake,
... forklength: PaperForkL,
... framelength: FrameLength,
... forkcg1: ForkCGPar,
... forkcg3: ForkCGNorm,
... framecg1: FrameCGNorm,
... framecg3: FrameCGPar,
... Iwr11: 0.0603,
... Iwr22: 0.12,
... Iwf11: 0.1405,
... Iwf22: 0.28,
... Ifork11: 0.05892,
... Ifork22: 0.06,
... Ifork33: 0.00708,
... Ifork31: 0.00756,
... Iframe11: 9.2,
... Iframe22: 11,
... Iframe33: 2.8,
... Iframe31: -2.4,
... mfork: 4,
... mframe: 85,
... mwf: 3,
... mwr: 2,
... g: 9.81,
... q1: 0,
... q2: 0,
... q4: 0,
... q5: 0,
... u1: 0,
... u2: 0,
... u3: v/PaperRadRear,
... u4: 0,
(continues on next page)
Linearizes the forcing vector; the equations are set up as MM udot = forcing, where MM is the
mass matrix, udot is the vector representing the time derivatives of the generalized speeds,
and forcing is a vector which contains both external forcing terms and internal forcing terms,
such as centripetal or Coriolis forces. This actually returns a matrix with as many rows as total
coordinates and speeds, but only as many columns as independent coordinates and speeds.
(Note that below this is commented out, as it takes a few minutes to run, which is not good
when performing the doctests)
As mentioned above, the size of the linearized forcing terms is expanded to include both q’s
and u’s, so the mass matrix must have this done as well. This will likely be changed to be part
of the linearized process, for future reference.
I think this is pretty self explanatory. It takes a really long time though. I’ve experimented with
using evalf with substitution, this failed due to maximum recursion depth being exceeded; I
also tried lambdifying this, and it is also not successful. (again commented out due to speed)
Finally, we construct an “A” matrix for the form xdot = A x (x being the state vector, although
in this case, the sizes are a little off). The following line extracts only the minimum entries
required for eigenvalue analysis, which correspond to rows and columns for lean, steer, lean
rate, and steer rate. (this is all commented out due to being dependent on the above code,
which is also commented out):
Upon running the above code yourself, enabling the commented out lines, compare the com-
puted eigenvalues to those is the referenced paper. This concludes the bicycle example.
This document will describe some of the more advanced functionality that this module offers
but which is not part of the “official” interface. Here, some of the features that will be im-
plemented in the future will also be covered, along with unanswered questions about proper
functionality. Also, common problems will be discussed, along with some solutions.
Common Issues
Here issues with numerically integrating code, choice of dynamicsymbols for coordinate and
speed representation, printing, differentiating, and substitution will occur.
Differentiating
Differentiation of very large expressions can take some time in SymPy; it is possible for large
expressions to take minutes for the derivative to be evaluated. This will most commonly come
up in linearization.
The Kane object is set up with the assumption that the generalized speeds are not the same
symbol as the time derivatives of the generalized coordinates. This isn’t to say that they can’t
be the same, just that they have to have a different symbol. If you did this:
Your code would not work. Currently, kinematic differential equations are required to be
provided. It is at this point that we hope the user will discover they should not attempt the
behavior shown in the code above.
This behavior might not be true for other methods of forming the equations of motion though.
Printing
The default printing options are to use sorting for Vector and Dyad measure numbers, and
have unsorted output from the mprint, mpprint, and mlatex functions. If you are printing
something large, please use one of those functions, as the sorting can increase printing time
from seconds to minutes.
Substitution
In this case, x was replaced with 1 inside the Derivative as well, which is undesired.
• Substitution into large expressions can be slow.
If your substitution is simple (direct replacement of expressions with other expressions, such
as when evaluating at an operating point) it is recommended to use the provided msubs func-
tion, as it is significantly faster, and handles the derivative issue appropriately:
Linearization
Currently, the linearization methods don’t support cases where there are non-coordinate, non-
speed dynamic symbols outside of the “dynamic equations”. It also does not support cases
where time derivatives of these types of dynamic symbols show up. This means if you have
kinematic differential equations which have a non-coordinate, non-speed dynamic symbol, it
will not work. It also means if you have defined a system parameter (say a length or distance
or mass) as a dynamic symbol, its time derivative is likely to show up in the dynamic equations,
and this will prevent linearization.
Acceleration of Points
At a minimum, points need to have their velocities defined, as the acceleration can be calcu-
lated by taking the time derivative of the velocity in the same frame. If the 1 point or 2 point
theorems were used to compute the velocity, the time derivative of the velocity expression
will most likely be more complex than if you were to use the acceleration level 1 point and 2
point theorems. Using the acceleration level methods can result in shorted expressions at this
point, which will result in shorter expressions later (such as when forming Kane’s equations).
Advanced Interfaces
Advanced Functionality
Remember that the Kane object supports bodies which have time-varying masses and inertias,
although this functionality isn’t completely compatible with the linearization method.
Future Features
Code Output
A function for generating code output for numerical integration is the highest priority feature
to implement next. There are a number of considerations here.
Code output for C (using the GSL libraries), Fortran 90 (using LSODA), MATLAB, and SciPy
is the goal. Things to be considered include: use of cse on large expressions for MATLAB and
SciPy, which are interpretive. It is currently unclear whether compiled languages will benefit
from common subexpression elimination, especially considering that it is a common part of
compiler optimization, and there can be a significant time penalty when calling cse.
Care needs to be taken when constructing the strings for these expressions, as well as han-
dling of input parameters, and other dynamic symbols. How to deal with output quantities
when integrating also needs to be decided, with the potential for multiple options being con-
sidered.
Autolev Parser
Introduction
Autolev (now superseded by MotionGenesis) is a domain specific language used for symbolic
multibody dynamics. The SymPy mechanics module now has enough power and functionality
to be a fully featured symbolic dynamics module. This parser parses Autolev (version 4.1)
code to SymPy code by making use of SymPy’s math libraries and the mechanics module.
The parser has been built using the ANTLR framework and its main purpose is to help former
users of Autolev to get familiarized with multibody dynamics in SymPy.
The sections below shall discuss details of the parser like usage, gotchas, issues and future
improvements. For a detailed comparison of Autolev and SymPy Mechanics you might want
to look at the SymPy Mechanics for Autolev Users guide (page 1673).
Usage
% double_pendulum.al
%-------------------
MOTIONVARIABLES' Q{2}', U{2}'
CONSTANTS L,M,G
NEWTONIAN N
FRAMES A,B
SIMPROT(N, A, 3, Q1)
% -> N_A = [COS(Q1), -SIN(Q1), 0; SIN(Q1), COS(Q1), 0; 0, 0, 1]
SIMPROT(N, B, 3, Q2)
% -> N_B = [COS(Q2), -SIN(Q2), 0; SIN(Q2), COS(Q2), 0; 0, 0, 1]
W_A_N>=U1*N3>
% -> W_A_N> = U1*N3>
W_B_N>=U2*N3>
% -> W_B_N> = U2*N3>
POINT O
PARTICLES P,R
P_O_P> = L*A1>
% -> P_O_P> = L*A1>
P_P_R> = L*B1>
% -> P_P_R> = L*B1>
V_O_N> = 0>
% -> V_O_N> = 0>
V2PTS(N, A, O, P)
% -> V_P_N> = L*U1*A2>
V2PTS(N, B, P, R)
% -> V_R_N> = L*U1*A2> + L*U2*B2>
MASS P=M, R=M
Q1' = U1
Q2' = U2
GRAVITY(G*N1>)
% -> FORCE_P> = G*M*N1>
% -> FORCE_R> = G*M*N1>
ZERO = FR() + FRSTAR()
% -> ZERO[1] = -L*M*(2*G*SIN(Q1)+L*(U2^2*SIN(Q1-Q2)+2*U1'+COS(Q1-Q2)*U2'))
% -> ZERO[2] = -L*M*(G*SIN(Q2)-L*(U1^2*SIN(Q1-Q2)-U2'-COS(Q1-Q2)*U1'))
KANE()
INPUT M=1,G=9.81,L=1
INPUT Q1=.1,Q2=.2,U1=0,U2=0
INPUT TFINAL=10, INTEGSTP=.01
CODE DYNAMICS() some_filename.c
>>> print(sympy_code)
import sympy.physics.mechanics as me
import sympy as sm
import math as m
import numpy as np
y=sys.integrate()
The commented code is not part of the output code. The print statements demonstrate how
to get responses similar to the ones in the Autolev file. Note that we need to use SymPy func-
tions like .ang_vel_in(), .dcm() etc in many cases unlike directly printing out the variables
like zero. If you are completely new to SymPy mechanics, the SymPy Mechanics for Autolev
Users guide (page 1673) guide should help. You might also have to use basic SymPy simplifi-
cations and manipulations like trigsimp(), expand(), evalf() etc for getting outputs similar
to Autolev. Refer to the SymPy Tutorial to know more about these.
Gotchas
• Don’t use variable names that conflict with Python’s reserved words. This is one example
where this is violated:
%Autolev Code
%------------
LAMBDA = EIG(M)
#SymPy Code
#----------
lambda = sm.Matrix([i.evalf() for i in (m).eigenvals().keys()])
• Make sure that the names of vectors and scalars are different. Autolev treats these
differently but these will get overwritten in Python. The parser currently allows the
names of bodies and scalars/vectors to coincide but doesn’t do this between scalars and
vectors. This should probably be changed in the future.
%Autolev Code
%------------
VARIABLES X,Y
FRAMES A
A> = X*A1> + Y*A2>
A = X+Y
#SymPy Code
#----------
x, y = me.dynamicsymbols('x y')
frame_a = me.ReferenceFrame('a')
a = x*frame_a.x + y*frame_a.y
a = x + y
# Note how frame_a is named differently so it doesn't cause a problem.
# On the other hand, 'a' gets rewritten from a scalar to a vector.
# This should be changed in the future.
• When dealing with Matrices returned by functions, one must check the order of the val-
ues as they may not be the same as in Autolev. This is especially the case for eigenvalues
and eigenvectors.
%Autolev Code
%------------
EIG(M, E1, E2)
% -> [5; 14; 13]
E2ROW = ROWS(E2, 1)
EIGVEC> = VECTOR(A, E2ROW)
#SymPy Code
#----------
e1 = sm.Matrix([i.evalf() for i in m.eigenvals().keys()])
# sm.Matrix([5;13;14]) different order
e2 = sm.Matrix([i[2][0].evalf() for i in m.eigenvects()]).reshape(m.shape[0], m.
,→shape[1])
• When using EVALUATE, use something like 90*UNITS(deg,rad) for angle substitutions
as radians are the default in SymPy. You could also add np.deg2rad() directly in the
SymPy code.
This need not be done for the output code (generated on parsing the CODE commands)
as the parser takes care of this when deg units are given in the INPUT declarations.
The DEGREES setting, on the other hand, works only in some cases like in SIMPROT where
an angle is expected.
%Autolev Code
%------------
A> = Q1*A1> + Q2*A2>
B> = EVALUATE(A>, Q1:30*UNITS(DEG,RAD))
#SymPy Code
#----------
a = q1*a.frame_a.x + q2*frame_a.y
b = a.subs({q1:30*0.0174533})
# b = a.subs({q1:np.deg2rad(30)}
• Most of the Autolev settings have not been parsed and have no effect on the parser.
The only ones that work somewhat are COMPLEX and DEGREES. It is advised to look into
alternatives to these in SymPy and Python.
• The REPRESENT command is not supported. Use the MATRIX, VECTOR or DYADIC commands
instead. Autolev 4.1 suggests these over REPRESENT as well while still allowing it but the
parser doesn’t parse it.
• Do not use variables declarations of the type WO{3}RD{2,4}. The parser can only handle
one variable name followed by one pair of curly braces and any number of ' s. You would
have to declare all the cases manually if you want to achieve something like WO{3}RD{2,
4}.
• The parser can handle normal versions of most commands but it may not parse functions
with Matrix arguments properly in most cases. Eg:
M=COEF([E1;E2],[U1,U2,U3])
This would compute the coefficients of U1, U2 and U3 in E1 and E2. It is preferable to
manually construct a Matrix using the regular versions of these commands.
%Autolev Code
%------------
% COEF([E1;E2],[U1,U2,U3])
M = [COEF(E1,U1),COEF(E1,U2),COEF(E1,U3) &
;COEF(E2,U1),COEF(E2,U2),COEF(E2,U3)]
• MOTIONVARIABLE declarations must be used for the generalized coordinates and speeds
and all other variables must be declared in regular VARIABLE declarations. The parser
requires this to distinguish between them to pass the correct parameters to the Kane’s
method object.
It is also preferred to always declare the speeds corresponding to the coordinates and
to pass in the kinematic differential equations. The parser is able to handle some cases
where this isn’t the case by introducing some dummy variables of its own but SymPy on
its own does require them.
Also note that older Autolev declarations like VARIABLES U{3}' are not supported either.
%Autolev Code
%------------
MOTIONVARIABLES' Q{2}', U{2}'
% ----- OTHER LINES ----
Q1' = U1
Q2' = U2
----- OTHER LINES ----
ZERO = FR() + FRSTAR()
#SymPy Code
#----------
q1, q2, u1, u2 = me.dynamicsymbols('q1 q2 u1 u2')
q1d, q2d, u1d, u2d = me.dynamicsymbols('q1 q2 u1 u2', 1)
• The parser creates SymPy symbols and dynamicsymbols by parsing variable declarations
in the Autolev Code.
For intermediate expressions which are directly initialized the parser does not create
SymPy symbols. It just assigns them to the expression.
On the other hand, when a declared variable is assigned to an expression, the parser
stores the expression against the variable in a dictionary so as to not reassign it to a
completely different entity. This constraint is due to the inherent nature of Python and
how it differs from a language like Autolev.
Also, Autolev seems to be able to assume whether to use a variable or the rhs expression
that variable has been assigned to in equations even without an explicit RHS() call in
some cases. For the parser to work correctly however, it is better to use RHS() wherever
a variable’s rhs expression is meant to be used.
%Autolev Code
%------------
VARIABLES X, Y
E = X + Y
X = 2*Y
RHS_X = RHS(X)
I1 = X
I2 = Y
I3 = X + Y
INERTIA B,I1,I2,I3
% -> I_B_BO>> = I1*B1>*B1> + I2*B2>*B2> + I3*B3>*B3>
#SymPy Code
#----------
x,y = me.dynamicsymbols('x y')
e = x + y # No symbol is made out of 'e'
rhs_x = 2*y
%Autolev Code
%------------
SOLVE(ZERO,X,Y)
A = RHS(X)*2 + RHS(Y)
#SymPy Code
#----------
print(sm.solve(zero,x,y))
# Behind the scenes the rhs of x
# is set to sm.solve(zero,x,y)[x].
a = sm.solve(zero,x,y)[x]*2 + sm.solve(zero,x,y)[y]
The indexing like [x] and [y] doesn’t always work so you might want to look at the
underlying dictionary that solve returns and index it correctly.
• Inertia declarations and Inertia functions work somewhat differently in the context of
the parser. This might be hard to understand at first but this had to be done to bridge
the gap due to the differences in SymPy and Autolev. Here are some points about them:
1. Inertia declarations (INERTIA B,I1,I2,I3) set the inertias of rigid bodies.
2. Inertia setters of the form I_C_D>> = expr however, set the inertias only when C is a
body. If C is a particle then I_C_D>> = expr simply parses to i_c_d = expr and i_c_d
acts like a regular variable.
3. When it comes to inertia getters (I_C_D>> used in an expression or INERTIA com-
mands), these MUST be used with the EXPRESS command to specify the frame as SymPy
needs this information to compute the inertia dyadic.
%Autolev Code
%------------
INERTIA B,I1,I2,I3
I_B_BO>> = X*A1>*A1> + Y*A2>*A2> % Parser will set the inertia of B
I_P_Q>> = X*A1>*A1> + Y^2*A2>*A2> % Parser just parses it as i_p_q = expr
E1 = 2*EXPRESS(I_B_O>>,A)
E2 = I_P_Q>>
E3 = EXPRESS(I_P_O>>,A)
E4 = EXPRESS(INERTIA(O),A)
% In E2 we are just using the dyadic object I_P_Q>> (as I_P_Q>> = expr
% doesn't act as a setter) defined above and not asking the parser
% or SymPy to compute anything.
% E4 asks the parser to compute the inertias of all bodies wrt about O.
• In an inertia declaration of a body, if the inertia is being set about a point other than the
center of mass, one needs to make sure that the position vector setter for that point and
the center of mass appears before the inertia declaration as SymPy will throw an error
otherwise.
%Autolev Code
%------------
P_SO_O> = X*A1>
INERTIA S_(O) I1,I2,I3
• Note that all Autolev commands have not been implemented. The parser now covers the
important ones in their basic forms. If you are doubtful whether a command is included
or not, please have a look at this file in the source code. Search for “<command>” to
verify this. Looking at the code for the specific command will also give an idea about
what form it is expected to work in.
• A lot of the issues have already been discussed in the Gotchas section. Some of these
are:
– Vector names coinciding with scalar names are overwritten in Python.
– Some convenient variable declarations aren’t parsed.
– Some convenient forms of functions to return matrices aren’t parsed.
– Settings aren’t parsed.
– symbols and rhs expressions work very differently in Python which might cause un-
desirable results.
– Dictionary indexing for the parsed code of the SOLVE command is not proper in many
cases.
– Need to change dynamicsymbols._t to dynamicsymbols('t') for the PyDy simula-
tion code to work properly.
Here are some other ones:
• Eigenvectors do not seem to work as expected. The values in Autolev and SymPy are not
the same in many cases.
• Block matrices aren’t parsed by the parser. It would actually be easier to make a change
in SymPy to allow matrices to accept other matrices for arguments.
• The SymPy equivalent of the TAYLOR command .series() does not work with dynamic-
symbols().
• Only DEPENDENT constraints are currently parsed. Need to parse AUXILIARY constraints
as well. This should be done soon as it isn’t very difficult.
• None of the energy and momentum functions are parsed right now. It would be nice
to get these working as well. Some changes should probably be made to SymPy. For
instance, SymPy doesn’t have a function equivalent to NICHECK().
• The numerical integration parts work properly only in the case of the KANE command
with no arguments. Things like KANE(F1,F2) do not currently work.
• Also, the PyDy numerical simulation code works only for cases where the matrix say ZERO
= FR() + FRSTAR() is solved for. It doesn’t work well when the matrix has some other
equations plugged in as well. One hurdle faced in achieving this was that PyDy’s System
class automatically takes in the forcing_full and mass_matrix_full and solves them
without giving the user the flexibility to specify the equations. It would be nice to add
this functionality to the System class.
Future Improvements
The parser has been built by referring to and parsing codes from the Autolev Tutorial and
the book Dynamics Online: Theory and Implementation Using Autolev. Basically, the process
involved going through each of these codes, validating the parser results and improving the
rules if required to make sure the codes parsed well.
The parsed codes of these are available on GitLab here. The repo is private so access needs
to be requested. As of now, most codes till Chapter 4 of Dynamics Online have been parsed.
Completing all the remaining codes of the book (namely, 2-10, 2-11, rest of Ch4, Ch5 and Ch6
(less important) ) would make the parser more complete.
2. Fixing Issues
The second thing to do would be to go about fixing the problems described above in the
Gotchas (page 1667) and Limitations and Issues (page 1672) sections in order of priority and
ease. Many of these require changes in the parser code while some of these are better fixed
by adding some functionality to SymPy.
3. Switching to an AST
The parser is currently built using a kind of Concrete Syntax Tree (CST) using the ANTLR
framework. It would be ideal to switch from a CST to an Abstract Syntax Tree (AST). This
way, the parser code will be independent of the ANTLR grammar which makes it a lot more
flexible. It would also be easier to make changes to the grammar and the rules of the parser.
Introduction
Autolev uses 1 (one) based indexing. The Python uses 0 (zero) based indexing. The
initial element of a sequence is found using initial element of a sequence is found using
a[1]. a[0].
One can define their own commands in SymPy code is Python code, so one can
Autolev by making .R and .A files which can define functions in their code. This is a lot
be used in their programs. more convenient.
The tables below give rough equivalents for some common Autolev expressions. These are
not exact equivalents, but rather should be taken as hints to get you going in the right di-
rection. For more detail read the built-in documentation on SymPy vectors, SymPy mechanics
and PyDy .
In the tables below, it is assumed that you have executed the following commands in Python:
import sympy.physics.mechanics as me
import sympy as sm
Mathematical Equivalents
Constants D- d = sm.symbols(‘d’,
real=True,
nonpositive=True)
Variables q, s q, s =
me.dynamicsymbols(q, s)
Physical Equivalents
Autolev’s CODE Option() command allows one to generate Matlab, C, or Fortran code for nu-
merical evaluation and visualization. Option can be Dynamics, ODE, Nonlinear or Algebraic.
Numerical evaluation for dynamics can be achieved using PyDy. One can pass in the Kanes-
Method object to the System class along with the values for the constants, specifieds, initial
conditions and time steps. The equations of motion can then be integrated. The plotting is
achieved using matlplotlib. Here is an example from the PyDy Documentation on how it is
done:
sys = System(kane,
constants = {mass: 1.0, stiffness: 1.0,
damping: 0.2, gravity: 9.8},
specifieds = {force: lambda x, t: sin(t)},
initial_conditions = {position: 0.1, speed:-1.0},
times = linspace(0.0, 10.0, 1000))
y = sys.integrate()
For information on all the things PyDy can accomplish refer to the PyDy Documentation.
The tools in the PyDy workflow are :
• SymPy: SymPy is a Python library for symbolic computation. It provides computer
algebra capabilities either as a standalone application, as a library to other applica-
tions, or live on the web as SymPy Live or SymPy Gamma.
• NumPy: NumPy is a library for the Python programming language, adding support
for large, multi-dimensional arrays and matrices, along with a large collection of
high-level mathematical functions to operate on these arrays.
• SciPy: SciPy is an open source Python library used for scientific computing and tech-
nical computing. SciPy contains modules for optimization, linear algebra, integra-
tion, interpolation, special functions, FFT, signal and image processing, ODE solvers
and other tasks common in science and engineering.
• IPython: IPython is a command shell for interactive computing in multiple pro-
gramming languages, originally developed for the Python programming language,
that offers introspection, rich media, shell syntax, tab completion, and history.
• Theano: Theano is a numerical computation library for Python. In Theano, computa-
tions are expressed using a NumPy-esque syntax and compiled to run efficiently on
either CPU or GPU architectures
• Cython: Cython is a superset of the Python programming language, designed to
give C-like performance with code that is mostly written in Python. Cython is a
compiled language that generates CPython extension modules.
• matplotlib: matplotlib is a plotting library for the Python programming language and
its numerical mathematics extension NumPy.
One will be able to write code equivalent to the Matlab, C or Fortran code generated by
Autolev using these scientific computing tools. It is recommended to go over these modules
to gain an understanding of scientific computing with Python.
Links
SymPy Tutorial
SymPy Documentation
SymPy Physics Vector Documentation
SymPy Mechanics Documentation
PyDy Documentation
MultiBody Dynamics with Python
Mechanics API
Particle
Examples
angular_momentum(point, frame)
Angular momentum of the particle about the point.
The angular momentum H, about some point O of a particle, P, is given by:
H=rxm*v
where r is the position vector from point O to the particle P, m is the mass of the
particle, and v is the velocity of the particle in the inertial frame, N.
Parameters point : Point
The point about which angular momentum of the particle is desired.
frame : ReferenceFrame
The frame in which angular momentum is desired.
Examples
kinetic_energy(frame)
Kinetic energy of the particle
The kinetic energy, T, of a particle, P, is given by
‘T = 1/2 m v^2’
where m is the mass of particle P, and v is the velocity of the particle in the supplied
ReferenceFrame.
Examples
linear_momentum(frame)
Linear momentum of the particle.
The linear momentum L, of a particle P, with respect to frame N is given by
L=m*v
where m is the mass of the particle, and v is the velocity of the particle in the frame
N.
Parameters frame : ReferenceFrame
The frame in which linear momentum is desired.
Examples
mass
Mass of the particle.
point
Point of the particle.
potential_energy
The potential energy of the Particle.
Examples
RigidBody
Examples
Attributes
angular_momentum(point, frame)
Returns the angular momentum of the rigid body about a point in the given frame.
The angular momentum H of a rigid body B about some point O in a frame N is given
by:
H = I·w + r×Mv
where I is the central inertia dyadic of B, w is the angular velocity of body B in the
frame, N, r is the position vector from point O to the mass center of B, and v is the
velocity of the mass center in the frame, N.
Parameters point : Point
The point about which angular momentum is desired.
frame : ReferenceFrame
The frame in which angular momentum is desired.
Examples
central_inertia
The body’s central inertia dyadic.
kinetic_energy(frame)
Kinetic energy of the rigid body
The kinetic energy, T, of a rigid body, B, is given by
‘T = 1/2 (I omega^2 + m v^2)’
where I and m are the central inertia dyadic and mass of rigid body B, respectively,
omega is the body’s angular velocity and v is the velocity of the body’s mass center
in the supplied ReferenceFrame.
Parameters frame : ReferenceFrame
The RigidBody’s angular velocity and the velocity of it’s mass center
are typically defined with respect to an inertial frame but any relevant
frame in which the velocities are known can be supplied.
Examples
linear_momentum(frame)
Linear momentum of the rigid body.
The linear momentum L, of a rigid body B, with respect to frame N is given by
L = M * v*
where M is the mass of the rigid body and v* is the velocity of the mass center of B
in the frame, N.
Parameters frame : ReferenceFrame
The frame in which linear momentum is desired.
Examples
potential_energy
The potential energy of the RigidBody.
Examples
inertia
Examples
inertia_of_point_mass
sympy.physics.mechanics.functions.inertia_of_point_mass(mass, pos_vec,
frame)
Inertia dyadic of a point mass relative to point O.
Parameters mass : Sympifyable
Mass of the point mass
pos_vec : Vector
Examples
linear_momentum
sympy.physics.mechanics.functions.linear_momentum(frame, *body)
Linear momentum of the system.
This function returns the linear momentum of a system of Particle’s and/or RigidBody’s.
The linear momentum of a system is equal to the vector sum of the linear momentum of
its constituents. Consider a system, S, comprised of a rigid body, A, and a particle, P. The
linear momentum of the system, L, is equal to the vector sum of the linear momentum
of the particle, L1, and the linear momentum of the rigid body, L2, i.e.
L = L1 + L2
Parameters frame : ReferenceFrame
The frame in which linear momentum is desired.
body1, body2, body3… : Particle and/or RigidBody
The body (or bodies) whose linear momentum is required.
Examples
angular_momentum
Examples
kinetic_energy
sympy.physics.mechanics.functions.kinetic_energy(frame, *body)
Kinetic energy of a multibody system.
This function returns the kinetic energy of a system of Particle’s and/or RigidBody’s.
The kinetic energy of such a system is equal to the sum of the kinetic energies of its
constituents. Consider a system, S, comprising a rigid body, A, and a particle, P. The
kinetic energy of the system, T, is equal to the vector sum of the kinetic energy of the
particle, T1, and the kinetic energy of the rigid body, T2, i.e.
T = T1 + T2
Examples
potential_energy
sympy.physics.mechanics.functions.potential_energy(*body)
Potential energy of a multibody system.
This function returns the potential energy of a system of Particle’s and/or RigidBody’s.
The potential energy of such a system is equal to the sum of the potential energy of its
constituents. Consider a system, S, comprising a rigid body, A, and a particle, P. The
potential energy of the system, V, is equal to the vector sum of the potential energy of
the particle, V1, and the potential energy of the rigid body, V2, i.e.
V = V1 + V2
Potential energy is a scalar.
Parameters body1, body2, body3… : Particle and/or RigidBody
The body (or bodies) whose potential energy is required.
Examples
Lagrangian
sympy.physics.mechanics.functions.Lagrangian(frame, *body)
Lagrangian of a multibody system.
This function returns the Lagrangian of a system of Particle’s and/or RigidBody’s. The
Lagrangian of such a system is equal to the difference between the kinetic energies and
potential energies of its constituents. If T and V are the kinetic and potential energies
of a system then it’s Lagrangian, L, is defined as
L=T-V
The Lagrangian is a scalar.
Parameters frame : ReferenceFrame
The frame in which the velocity or angular velocity of the body is
defined to determine the kinetic energy.
body1, body2, body3… : Particle and/or RigidBody
The body (or bodies) whose Lagrangian is required.
Examples
KaneMethod
Examples
Next we need to arrange/store information in the way that KanesMethod requires. The
kinematic differential equations need to be stored in a dict. A list of forces/torques must
be constructed, where each entry in the list is a (Point, Vector) or (ReferenceFrame,
Vector) tuple, where the Vectors represent the Force or Torque. Next a particle needs
to be created, and it needs to have a point and mass assigned to it. Finally, a list of all
bodies and particles needs to be created.
>>> kd = [qd - u]
>>> FL = [(P, (-k * q - c * u) * N.x)]
(continues on next page)
Finally we can generate the equations of motion. First we create the KanesMethod object
and supply an inertial frame, coordinates, generalized speeds, and the kinematic differ-
ential equations. Additional quantities such as configuration and motion constraints,
dependent coordinates and speeds, and auxiliary speeds are also supplied here (see the
online documentation). Next we form FR* and FR to complete: Fr + Fr* = 0. We have
the equations of motion at this point. It makes sense to rearrange them though, so we
calculate the mass matrix and the forcing terms, for E.o.M. in the form: [MM] udot =
forcing, where MM is the mass matrix, udot is a vector of the time derivatives of the
generalized speeds, and forcing is a vector representing “forcing” terms.
Please look at the documentation pages for more information on how to perform lin-
earization and how to deal with dependent coordinates & speeds, and how do deal with
bringing non-contributing forces into evidence.
Attributes
auxiliary_eqs
A matrix containing the auxiliary equations.
forcing
The forcing vector of the system.
forcing_full
The forcing vector of the system, augmented by the kinematic differential equations.
kanes_equations(bodies, loads=None)
Method to form Kane’s equations, Fr + Fr* = 0.
Returns (Fr, Fr*). In the case where auxiliary generalized speeds are present (say, s
auxiliary speeds, o generalized speeds, and m motion constraints) the length of the
returned vectors will be o - m + s in length. The first o - m equations will be the
constrained Kane’s equations, then the s auxiliary Kane’s equations. These auxiliary
equations can be accessed with the auxiliary_eqs().
Parameters bodies : iterable
An iterable of all RigidBody’s and Particle’s in the system. A system
must have at least one body.
loads : iterable
Takes in an iterable of (Particle, Vector) or (ReferenceFrame, Vector)
tuples which represent the force at a point or torque on a frame. Must
be either a non-empty iterable of tuples or None which corresponds
to a system with no constraints.
kindiffdict()
Returns a dictionary mapping q’ to u.
linearize(**kwargs)
Linearize the equations of motion about a symbolic operating point.
If kwarg A_and_B is False (default), returns M, A, B, r for the linearized form, M*[q’,
u’]^T = A*[q_ind, u_ind]^T + B*r.
If kwarg A_and_B is True, returns A, B, r for the linearized form dx = A*x + B*r, where
x = [q_ind, u_ind]^T. Note that this is computationally intensive if there are many
symbolic parameters. For this reason, it may be more desirable to use the default
A_and_B=False, returning M, A, and B. Values may then be substituted in to these
matrices, and the state space form found as A = P.T*M.inv()*A, B = P.T*M.inv()*B,
where P = Linearizer.perm_mat.
In both cases, r is found as all dynamicsymbols in the equations of motion that are
not part of q, u, q’, or u’. They are sorted in canonical form.
The operating points may be also entered using the op_point kwarg. This takes a
dictionary of {symbol: value}, or a an iterable of such dictionaries. The values may
be numeric or symbolic. The more values you can specify beforehand, the faster this
computation will run.
For more documentation, please see the Linearizer class.
mass_matrix
The mass matrix of the system.
mass_matrix_full
The mass matrix of the system, augmented by the kinematic differential equations.
rhs(inv_method=None)
Returns the system’s equations of motion in first order form. The output is the right
hand side of:
The right hand side is what is needed by most numerical ODE integrators.
LagrangesMethod
Examples
>>> Pa = Particle('Pa', P, m)
>>> Pa.potential_energy = k * q**2 / 2.0
>>> L = Lagrangian(N, Pa)
>>> fl = [(P, -b * qd * N.x)]
Finally we can generate the equations of motion. First we create the LagrangesMethod
object. To do this one must supply the Lagrangian, and the generalized coordinates.
The constraint equations, the forcelist, and the inertial frame may also be provided,
if relevant. Next we generate Lagrange’s equations of motion, such that: Lagrange’s
equations of motion = 0. We have the equations of motion at this point.
We can also solve for the states using the ‘rhs’ method.
>>> print(l.rhs())
Matrix([[Derivative(q(t), t)], [(-b*Derivative(q(t), t) - 1.0*k*q(t))/m]])
Attributes
forcing
Returns the forcing vector from ‘lagranges_equations’ method.
forcing_full
Augments qdots to the forcing vector above.
form_lagranges_equations()
Method to form Lagrange’s equations of motion.
Returns a vector of equations of motion using Lagrange’s equations of the second
kind.
linearize(q_ind=None, qd_ind=None, q_dep=None, qd_dep=None, **kwargs)
Linearize the equations of motion about a symbolic operating point.
If kwarg A_and_B is False (default), returns M, A, B, r for the linearized form, M*[q’,
u’]^T = A*[q_ind, u_ind]^T + B*r.
If kwarg A_and_B is True, returns A, B, r for the linearized form dx = A*x + B*r, where
x = [q_ind, u_ind]^T. Note that this is computationally intensive if there are many
symbolic parameters. For this reason, it may be more desirable to use the default
A_and_B=False, returning M, A, and B. Values may then be substituted in to these
matrices, and the state space form found as A = P.T*M.inv()*A, B = P.T*M.inv()*B,
where P = Linearizer.perm_mat.
In both cases, r is found as all dynamicsymbols in the equations of motion that are
not part of q, u, q’, or u’. They are sorted in canonical form.
The operating points may be also entered using the op_point kwarg. This takes a
dictionary of {symbol: value}, or a an iterable of such dictionaries. The values may
be numberic or symbolic. The more values you can specify beforehand, the faster
this computation will run.
For more documentation, please see the Linearizer class.
mass_matrix
Returns the mass matrix, which is augmented by the Lagrange multipliers, if neces-
sary.
If the system is described by ‘n’ generalized coordinates and there are no constraint
equations then an n X n matrix is returned.
If there are ‘n’ generalized coordinates and ‘m’ constraint equations have been sup-
plied during initialization then an n X (n+m) matrix is returned. The (n + m - 1)th
and (n + m)th columns contain the coefficients of the Lagrange multipliers.
mass_matrix_full
Augments the coefficients of qdots to the mass_matrix.
rhs(inv_method=None, **kwargs)
Returns equations that can be solved numerically
Parameters inv_method : str
The specific sympy inverse matrix calculation method to use. For a
list of valid methods, see inv() (page 771)
solve_multipliers(op_point=None, sol_type=’dict’)
Solves for the values of the lagrange multipliers symbolically at the specified oper-
ating point
Parameters op_point : dict or iterable of dicts, optional
Point at which to solve at. The operating point is specified as a dic-
tionary or iterable of dictionaries of {symbol: value}. The value may
be numeric or symbolic itself.
sol_type : str, optional
Solution return type. Valid options are: - ‘dict’: A dict of {symbol :
value} (default) - ‘Matrix’: An ordered column matrix of the solution
to_linearizer(q_ind=None, qd_ind=None, q_dep=None, qd_dep=None)
Returns an instance of the Linearizer class, initiated from the data in the Lagranges-
Method class. This may be more desirable than using the linearize class method,
as the Linearizer object will allow more efficient recalculation (i.e. about varying
operating points).
Parameters q_ind, qd_ind : array_like, optional
The independent generalized coordinates and speeds.
SymbolicSystem (Docstrings)
SymbolicSystem
class sympy.physics.mechanics.system.SymbolicSystem(coord_states,
right_hand_side,
speeds=None,
mass_matrix=None, coor-
dinate_derivatives=None,
alg_con=None, out-
put_eqns={}, co-
ord_idxs=None,
speed_idxs=None, bod-
ies=None, loads=None)
SymbolicSystem is a class that contains all the information about a system in a symbolic
format such as the equations of motions and the bodies and loads in the system.
There are three ways that the equations of motion can be described for Symbolic System:
[1] Explicit form where the kinematics and dynamics are combined x’
= F_1(x, t, r, p)
[2] Implicit form where the kinematics and dynamics are combined
M_2(x, p) x’ = F_2(x, t, r, p)
[3] Implicit form where the kinematics and dynamics are separate
M_3(q, p) u’ = F_3(q, u, t, r, p) q’ = G(q, u, t, r, p)
where
x : states, e.g. [q, u] t : time r : specified (exogenous) inputs p : constants q : generalized
coordinates u : generalized speeds F_1 : right hand side of the combined equations in
explicit form F_2 : right hand side of the combined equations in implicit form F_3 : right
hand side of the dynamical equations in implicit form M_2 : mass matrix of the combined
equations in implicit form M_3 : mass matrix of the dynamical equations in implicit form
G : right hand side of the kinematical differential equations
Parameters coord_states : ordered iterable of functions of time
This input will either be a collection of the coordinates or states
of the system depending on whether or not the speeds are also
given. If speeds are specified this input will be assumed to be the
coordinates otherwise this input will be assumed to be the states.
Notes
Example
As a simple example, the dynamics of a simple pendulum will be input into a Symbol-
icSystem object manually. First some imports will be needed and then symbols will be
set up for the length of the pendulum (l), mass at the end of the pendulum (m), and a
constant for gravity (g).
The system will be defined by an angle of theta from the vertical and a generalized speed
of omega will be used where omega = theta_dot.
Now the equations of motion are ready to be formed and passed to the SymbolicSystem
object.
Attributes
co- (Matrix, shape(n, 1)) This is a matrix containing the generalized coordinates of
or- the system
di-
nates
speeds(Matrix, shape(m, 1)) This is a matrix containing the generalized speeds of the
system
states (Matrix, shape(o, 1)) This is a matrix containing the state variables of the system
alg_con(List) This list contains the indices of the algebraic constraints in the combined
equations of motion. The presence of these constraints requires that a DAE
solver be used instead of an ODE solver. If the system is given in form [3] the
alg_con variable will be adjusted such that it is a representation of the combined
kinematics and dynamics thus make sure it always matches the mass matrix
entered.
dyn_implicit_mat
(Matrix, shape(m, m)) This is the M matrix in form [3] of the equations of motion
(the mass matrix or generalized inertia matrix of the dynamical equations of
motion in implicit form).
dyn_implicit_rhs
(Matrix, shape(m, 1)) This is the F vector in form [3] of the equations of motion
(the right hand side of the dynamical equations of motion in implicit form).
comb_implicit_mat
(Matrix, shape(o, o)) This is the M matrix in form [2] of the equations of motion.
This matrix contains a block diagonal structure where the top left block (the first
rows) represent the matrix in the implicit form of the kinematical equations and
the bottom right block (the last rows) represent the matrix in the implicit form
of the dynamical equations.
comb_implicit_rhs
(Matrix, shape(o, 1)) This is the F vector in form [2] of the equations of motion.
The top part of the vector represents the right hand side of the implicit form
of the kinemaical equations and the bottom of the vector represents the right
hand side of the implicit form of the dynamical equations of motion.
comb_explicit_rhs
(Matrix, shape(o, 1)) This vector represents the right hand side of the combined
equations of motion in explicit form (form [1] from above).
kin_explicit_rhs
(Matrix, shape(m, 1)) This is the right hand side of the explicit form of the kine-
matical equations of motion as can be seen in form [3] (the G matrix).
out- (Dictionary) If output equations were given they are stored in a dictionary where
put_eqns
the key corresponds to the name given for the specific equation and the value
is the equation itself in symbolic form
bod- (Tuple) If the bodies in the system were given they are stored in a tuple for
ies future access
loads (Tuple) If the loads in the system were given they are stored in a tuple for future
access. This includes forces and torques where forces are given by (point of
application, force vector) and torques are given by (reference frame acted upon,
torque vector).
alg_con
Returns a list with the indices of the rows containing algebraic constraints in the
combined form of the equations of motion
bodies
Returns the bodies in the system
comb_explicit_rhs
Returns the right hand side of the equations of motion in explicit form, x’ = F, where
the kinematical equations are included
comb_implicit_mat
Returns the matrix, M, corresponding to the equations of motion in implicit form
(form [2]), M x’ = F, where the kinematical equations are included
comb_implicit_rhs
Returns the column matrix, F, corresponding to the equations of motion in implicit
form (form [2]), M x’ = F, where the kinematical equations are included
compute_explicit_form()
If the explicit right hand side of the combined equations of motion is to provided
upon initialization, this method will calculate it. This calculation can potentially
take awhile to compute.
constant_symbols()
Returns a column matrix containing all of the symbols in the system that do not
depend on time
coordinates
Returns the column matrix of the generalized coordinates
dyn_implicit_mat
Returns the matrix, M, corresponding to the dynamic equations in implicit form, M
x’ = F, where the kinematical equations are not included
dyn_implicit_rhs
Returns the column matrix, F, corresponding to the dynamic equations in implicit
form, M x’ = F, where the kinematical equations are not included
dynamic_symbols()
Returns a column matrix containing all of the symbols in the system that depend on
time
kin_explicit_rhs
Returns the right hand side of the kinematical equations in explicit form, q’ = G
loads
Returns the loads in the system
speeds
Returns the column matrix of generalized speeds
states
Returns the column matrix of the state variables
Linearization (Docstrings)
Linearizer
Attributes
f_0, f_1, f_2, f_3, f_4, (Matrix) Matrices holding the general system form.
f_c, f_v, f_a
q, u, r (Matrix) Matrices holding the generalized coordinates,
speeds, and input vectors.
q_i, u_i (Matrix) Matrices of the independent generalized coordi-
nates and speeds.
q_d, u_d (Matrix) Matrices of the dependent generalized coordinates
and speeds.
perm_mat (Matrix) Permutation matrix such that [q_ind, u_ind]^T =
perm_mat*[q, u]^T
Potential Issues
msubs
If smart=True (default False), also checks for conditions that may result in nan, but if
simplified would yield a valid expression. For example:
>>> from sympy import sin, tan
>>> (sin(x)/tan(x)).subs(x, 0)
nan
>>> msubs(sin(x)/tan(x), {x: 0}, smart=True)
1
It does this by first replacing all tan with sin/cos. Then each node is traversed. If
the node is a fraction, subs is first evaluated on the denominator. If this results in 0,
simplification of the entire fraction is attempted. Using this selective simplification, only
subexpressions that result in 1/0 are targeted, resulting in faster performance.
find_dynamicsymbols
Examples
Printing (Docstrings)
mechanics_printing
mprint
mpprint
mlatex
Body (Docstrings)
Body
Examples
Default behaviour. This results in the creation of a RigidBody object for which the mass,
mass center, frame and inertia attributes are given default values.
This next example demonstrates the code required to specify all of the values of the Body
object. Note this will also create a RigidBody version of the Body object.
The minimal code required to create a Particle version of the Body object involves simply
passing in a name and a mass.
The Particle version of the Body object can also receive a masscenter point and a refer-
ence frame, just not an inertia.
Attributes
apply_force(vec, point=None)
Adds a force to a point (center of mass by default) on the body.
Parameters vec: Vector
Defines the force vector. Can be any vector w.r.t any frame or com-
binations of frames.
point: Point, optional
Defines the point on which the force is applied. Default is the Body’s
center of mass.
Example
The first example applies a gravitational force in the x direction of Body’s frame to
the body’s center of mass.
>>> from sympy import Symbol
>>> from sympy.physics.mechanics import Body
>>> body = Body('body')
>>> g = Symbol('g')
>>> body.apply_force(body.mass * g * body.frame.x)
To apply force to any other point than center of mass, pass that point as well. This
example applies a gravitational force to a point a distance l from the body’s center
of mass in the y direction. The force is again applied in the x direction.
>>> from sympy import Symbol
>>> from sympy.physics.mechanics import Body
>>> body = Body('body')
>>> g = Symbol('g')
>>> l = Symbol('l')
>>> point = body.masscenter.locatenew('force_point', l *
... body.frame.y)
>>> body.apply_force(body.mass * g * body.frame.x, point)
apply_torque(vec)
Adds a torque to the body.
Parameters vec: Vector
Defines the torque vector. Can be any vector w.r.t any frame or com-
binations of frame.
Example
Quantum Mechanics
Abstract
Quantum Functions
Anticommutator
References
[R470]
Examples
>>> ac = AntiCommutator(A,B); ac
{A,B}
>>> ac.doit()
A*B + B*A
>>> ac = AntiCommutator(B,A); ac
{A,B}
>>> AntiCommutator(3*x*A,x*y*B)
3*x**2*y*{A,B}
Adjoint operations applied to the anticommutator are properly applied to the arguments:
>>> Dagger(AntiCommutator(A,B))
{Dagger(A),Dagger(B)}
doit(**hints)
Evaluate anticommutator
Clebsch-Gordan Coefficients
Clebsch-Gordon Coefficients.
class sympy.physics.quantum.cg.CG
Class for Clebsch-Gordan coefficient
Clebsch-Gordan coefficients describe the angular momentum coupling between two sys-
tems. The coefficients give the expansion of a coupled total angular momentum state
and an uncoupled tensor product state. The Clebsch-Gordan coefficients are defined as
[R471]:
Cjj21,m
,m1
2 ,j3 ,m3
= ⟨j1 , m1 ; j2 , m2 |j3 , m3 ⟩
References
[R471]
Examples
class sympy.physics.quantum.cg.Wigner3j
Class for the Wigner-3j symbols
Wigner 3j-symbols are coefficients determined by the coupling of two angular momenta.
When created, they are expressed as symbolic quantities that, for numerical parameters,
can be evaluated using the .doit() method [R472].
Parameters j1, m1, j2, m2, j3, m3 : Number, Symbol
Terms determining the angular momentum of coupled angular mo-
mentum systems.
See also:
References
[R472]
Examples
class sympy.physics.quantum.cg.Wigner6j
Class for the Wigner-6j symbols
See also:
class sympy.physics.quantum.cg.Wigner9j
Class for the Wigner-9j symbols
See also:
sympy.physics.quantum.cg.cg_simp(e)
Simplify and combine CG coefficients
This function uses various symmetry and properties of sums and products of Clebsch-
Gordan coefficients to simplify statements involving these terms [R473].
See also:
References
[R473]
Examples
Commutator
References
[R474]
Examples
Adjoint operations applied to the commutator are properly applied to the arguments:
doit(**hints)
Evaluate commutator
Constants
Dagger
Hermitian conjugation.
class sympy.physics.quantum.dagger.Dagger
General Hermitian conjugate operation.
Take the Hermetian conjugate of an argument [R475]. For matrices this operation is
equivalent to transpose and complex conjugate [R476].
Parameters arg : Expr
The sympy expression that we want to take the dagger of.
References
[R475], [R476]
Examples
>>> A = Operator('A')
>>> B = Operator('B')
>>> Dagger(A*B)
Dagger(B)*Dagger(A)
>>> Dagger(A+B)
Dagger(A) + Dagger(B)
>>> Dagger(A**2)
Dagger(A)**2
Inner Product
References
[R479]
Examples
In simple products of kets and bras inner products will be automatically identified and
created:
>>> b*k
<b|k>
But in more complex expressions, there is ambiguity in whether inner or outer products
should be created:
>>> k*b*k*b
|k><b|*|k>*<b|
A user can force the creation of a inner products in a complex expression by using paren-
theses to group the bra and ket:
>>> k*(b*k)*b
<b|k>*|k>*<b|
Notice how the inner product <b|k> moved to the left of the expression because inner
products are commutative complex numbers.
Tensor Product
Examples
>>> m1 = Matrix([[1,2],[3,4]])
>>> m2 = Matrix([[1,0],[0,1]])
>>> TensorProduct(m1, m2)
Matrix([
[1, 0, 2, 0],
[0, 1, 0, 2],
[3, 0, 4, 0],
[0, 3, 0, 4]])
>>> TensorProduct(m2, m1)
Matrix([
[1, 2, 0, 0],
[3, 4, 0, 0],
[0, 0, 1, 2],
[0, 0, 3, 4]])
We can take the dagger of a tensor product (note the order does NOT reverse like the
dagger of a normal product):
>>> C = Symbol('C',commutative=False)
>>> tp = TensorProduct(A+B,C)
>>> tp
(A + B)xC
>>> tp.expand(tensorproduct=True)
AxC + BxC
sympy.physics.quantum.tensorproduct.tensor_product_simp(e, **hints)
Try to simplify and combine TensorProducts.
In general this will try to pull expressions inside of TensorProducts. It currently only
works for relatively simple cases where the products have only scalars, raw Tensor-
Products, not Add, Pow, Commutators of TensorProducts. It is best to see what it does
by showing examples.
Examples
>>> e = TensorProduct(A,B)*TensorProduct(C,D)
>>> e
AxB*CxD
>>> tensor_product_simp(e)
(A*C)x(B*D)
This is the core logic of this function, and it works inside, powers, sums, commutators
and anticommutators as well:
>>> tensor_product_simp(e**2)
(A*C)x(B*D)**2
Hilbert Space
References
[R477]
Examples
dimension
Return the Hilbert dimension of the space.
class sympy.physics.quantum.hilbert.ComplexSpace
Finite dimensional Hilbert space of complex vectors.
The elements of this Hilbert space are n-dimensional complex valued vectors with the
usual inner product that takes the complex conjugate of the vector on the right.
A classic example of this type of Hilbert space is spin-1/2, which is ComplexSpace(2).
Generalizing to spin-s, the space is ComplexSpace(2*s+1). Quantum computing with N
qubits is done with the direct product space ComplexSpace(2)**N.
Examples
>>> n = symbols('n')
>>> c2 = ComplexSpace(n)
>>> c2
C(n)
>>> c2.dimension
n
class sympy.physics.quantum.hilbert.L2
The Hilbert space of square integrable functions on an interval.
An L2 object takes in a single sympy Interval argument which represents the interval its
functions (vectors) are defined on.
Examples
class sympy.physics.quantum.hilbert.FockSpace
The Hilbert space for second quantization.
Technically, this Hilbert space is a infinite direct sum of direct products of single particle
Hilbert spaces [R478]. This is a mess, so we have a class to represent it directly.
References
[R478]
Examples
Operator
class sympy.physics.quantum.operator.Operator
Base class for non-commuting quantum operators.
An operator maps between quantum states [R480]. In quantum mechanics, observables
(including, but not limited to, measured physical values) are represented as Hermitian
operators [R481].
Parameters args : tuple
The list of numbers or parameters that uniquely specify the operator.
For time-dependent operators, this will include the time.
References
[R480], [R481]
Examples
>>> B = Operator('B')
>>> C = 2*A*A + I*B
>>> C
2*A**2 + I*B
>>> A.is_commutative
False
>>> B.is_commutative
False
>>> A*B == B*A
False
>>> e = (A+B)**3
>>> e.expand()
A*B*A + A*B**2 + A**2*B + A**3 + B*A*B + B*A**2 + B**2*A + B**3
>>> A.inv()
A**(-1)
>>> A*A.inv()
1
class sympy.physics.quantum.operator.HermitianOperator
A Hermitian operator that satisfies H == Dagger(H).
Parameters args : tuple
The list of numbers or parameters that uniquely specify the operator.
For time-dependent operators, this will include the time.
Examples
class sympy.physics.quantum.operator.UnitaryOperator
A unitary operator that satisfies U*Dagger(U) == 1.
Parameters args : tuple
The list of numbers or parameters that uniquely specify the operator.
For time-dependent operators, this will include the time.
Examples
Examples
class sympy.physics.quantum.operator.OuterProduct
An unevaluated outer product between a ket and bra.
This constructs an outer product between any subclass of KetBase and BraBase as
|a><b|. An OuterProduct inherits from Operator as they act as operators in quantum
expressions. For reference see [R482].
Parameters ket : KetBase
The ket on the left side of the outer product.
bar : BraBase
The bra on the right side of the outer product.
References
[R482]
Examples
>>> k = Ket('k')
>>> b = Bra('b')
>>> op = OuterProduct(k, b)
>>> op
|k><b|
>>> op.hilbert_space
H
>>> op.ket
|k>
>>> op.bra
<b|
>>> Dagger(op)
|b><k|
In simple products of kets and bras outer products will be automatically identified and
created:
>>> k*b
|k><b|
But in more complex expressions, outer products are not automatically created:
>>> A = Operator('A')
>>> A*k*b
A*|k>*<b|
A user can force the creation of an outer product in a complex expression by using paren-
theses to group the ket and bra:
>>> A*(k*b)
A*|k><b|
bra
Return the bra on the right side of the outer product.
ket
Return the ket on the left side of the outer product.
class sympy.physics.quantum.operator.DifferentialOperator
An operator for representing the differential operator, i.e. d/dx
It is initialized by passing two arguments. The first is an arbitrary expression that in-
volves a function, such as Derivative(f(x), x). The second is the function (e.g. f(x))
which we are to replace with the Wavefunction that this DifferentialOperator is ap-
plied to.
Parameters expr : Expr
The arbitrary expression which the appropriate Wavefunction is to be
substituted into
func : Expr
A function (e.g. f(x)) which is to be replaced with the appropriate
Wavefunction when this DifferentialOperator is applied
Examples
You can define a completely arbitrary expression and specify where the Wavefunction is
to be substituted
>>> from sympy import Derivative, Function, Symbol
>>> from sympy.physics.quantum.operator import DifferentialOperator
>>> from sympy.physics.quantum.state import Wavefunction
>>> from sympy.physics.quantum.qapply import qapply
>>> f = Function('f')
>>> x = Symbol('x')
>>> d = DifferentialOperator(1/x*Derivative(f(x), x), f(x))
>>> w = Wavefunction(x**2, x)
>>> d.function
f(x)
>>> d.variables
(x,)
>>> qapply(d*w)
Wavefunction(2, x)
expr
Returns the arbitrary expression which is to have the Wavefunction substituted into
it
Examples
free_symbols
Return the free symbols of the expression.
function
Returns the function which is to be replaced with the Wavefunction
Examples
variables
Returns the variables with which the function in the specified arbitrary expression
is evaluated
Examples
A module for mapping operators to their corresponding eigenstates and vice versa
It contains a global dictionary with eigenstate-operator pairings. If a new state-operator pair
is created, this dictionary should be updated as well.
It also contains functions operators_to_state and state_to_operators for mapping between the
two. These can handle both classes and instances of operators and states. See the individual
function descriptions for details.
TODO List: - Update the dictionary with a complete list of state-operator pairs
sympy.physics.quantum.operatorset.operators_to_state(operators, **options)
Returns the eigenstate of the given operator or set of operators
A global function for mapping operator classes to their associated states. It takes either
an Operator or a set of operators and returns the state associated with these.
This function can handle both instances of a given operator or just the class itself (i.e.
both XOp() and XOp)
There are multiple use cases to consider:
1) A class or set of classes is passed: First, we try to instantiate default instances for
these operators. If this fails, then the class is simply returned. If we succeed in instan-
tiating default instances, then we try to call state._operators_to_state on the operator
instances. If this fails, the class is returned. Otherwise, the instance returned by _oper-
ators_to_state is returned.
2) An instance or set of instances is passed: In this case, state._operators_to_state is
called on the instances passed. If this fails, a state class is returned. If the method
returns an instance, that instance is returned.
In both cases, if the operator class or set does not exist in the state_mapping dictionary,
None is returned.
Parameters arg: Operator or set
The class or instance of the operator or set of operators to be mapped
to a state
Examples
sympy.physics.quantum.operatorset.state_to_operators(state, **options)
Returns the operator or set of operators corresponding to the given eigenstate
A global function for mapping state classes to their associated operators or sets of oper-
ators. It takes either a state class or instance.
This function can handle both instances of a given state or just the class itself (i.e. both
XKet() and XKet)
There are multiple use cases to consider:
1) A state class is passed: In this case, we first try instantiating a default instance of
the class. If this succeeds, then we try to call state._state_to_operators on that instance.
If the creation of the default instance or if the calling of _state_to_operators fails, then
either an operator class or set of operator classes is returned. Otherwise, the appropriate
operator instances are returned.
2) A state instance is returned: Here, state._state_to_operators is called for the instance.
If this fails, then a class or set of operator classes is returned. Otherwise, the instances
are returned.
In either case, if the state’s class does not exist in state_mapping, None is returned.
Parameters arg: StateBase class or instance (or subclasses)
The class or instance of the state to be mapped to an operator or set
of operators
Examples
Qapply
options : dict
A dict of key/value pairs that determine how the operator actions are
carried out.
The following options are valid:
• dagger: try to apply Dagger operators to the left (default: False).
• ip_doit: call .doit() in inner products when they are encountered
(default: True).
Returns e : Expr
The original expression, but with the operators applied to states.
Examples
Represent
This function will have the logic for representing instances of its class in the basis set
having a class named FooBasis.
Parameters expr : Expr
The expression to represent.
basis : Operator, basis set
An object that contains the information about the basis set. If an oper-
ator is used, the basis is assumed to be the orthonormal eigenvectors
of that operator. In general though, the basis argument can be any
object that contains the basis set information.
options : dict
Key/value pairs of options that are passed to the underlying method
that finds the representation. These options can be used to control
how the representation is done. For example, this is where the size
of the basis set would be set.
Returns e : Expr
The SymPy expression of the represented quantum expression.
Examples
Here we subclass Operator and Ket to create the z-spin operator and its spin 1/2 up
eigenstate. By defining the _represent_SzOp method, the ket can be represented in the
z-spin basis.
Here we see an example of representations in a continuous basis. We see that the result
of representing various combinations of cartesian position operators and kets give us
continuous expressions involving DiracDelta functions.
sympy.physics.quantum.represent.rep_innerproduct(expr, **options)
Returns an innerproduct like representation (e.g. <x'|x>) for the given state.
Attempts to calculate inner product with a bra from the specified basis. Should only be
passed an instance of KetBase or BraBase
Parameters expr : KetBase or BraBase
The expression to be represented
Examples
sympy.physics.quantum.represent.rep_expectation(expr, **options)
Returns an <x'|A|x> type representation for the given operator.
Parameters expr : Operator
Operator to be represented in the specified basis
Examples
Note: This is mostly used internally by represent(). Examples are given merely to show
the use cases.
Parameters orig_expr : quantum expression
The original expression which was to be represented
result: Expr
The resulting representation that we wish to integrate over
Examples
sympy.physics.quantum.represent.get_basis(expr, **options)
Returns a basis state instance corresponding to the basis specified in options=s. If no
basis is specified, the function tries to form a default basis state of the given expression.
There are three behaviors:
1. The basis specified in options is already an instance of StateBase. If this is the case,
it is simply returned. If the class is specified but not an instance, a default instance
is returned.
2. The basis specified is an operator or set of operators. If this is the case, the opera-
tor_to_state mapping method is used.
3. No basis is specified. If expr is a state, then a default instance of its class is returned.
If expr is an operator, then it is mapped to the corresponding state. If it is neither,
then we cannot obtain the basis state.
If the basis cannot be mapped, then it is not changed.
This will be called from within represent, and represent will only pass QExpr’s.
TODO (?): Support for Muls and other types of expressions?
Parameters expr : Operator or StateBase
Expression whose basis is sought
Examples
sympy.physics.quantum.represent.enumerate_states(*args, **options)
Returns instances of the given state with dummy indices appended
Operates in two different modes:
1. Two arguments are passed to it. The first is the base state which is to be indexed,
and the second argument is a list of indices to append.
2. Three arguments are passed. The first is again the base state to be indexed. The
second is the start index for counting. The final argument is the number of kets you
wish to receive.
Tries to call state._enumerate_state. If this fails, returns an empty list
Parameters args : list
See list of operation modes above for explanation
Examples
Spin
References
[R483]
Examples
With symbolic Euler angles and calculating the inverse rotation operator:
Examples
Return the Wigner-D matrix element for a defined rotation, both numerical and sym-
bolic:
Examples
Return the Wigner-D matrix element for a defined rotation, both numerical and sym-
bolic:
class sympy.physics.quantum.spin.WignerD
Wigner-D function
The Wigner D-function gives the matrix elements of the rotation operator in the jm-
representation. For the Euler angles α, β, γ, the D-function is defined such that:
References
[R484]
Examples
class sympy.physics.quantum.spin.JxKet
Eigenket of Jx.
See JzKet for the usage of spin eigenstates.
See also:
class sympy.physics.quantum.spin.JxBra
Eigenbra of Jx.
See JzKet for the usage of spin eigenstates.
See also:
class sympy.physics.quantum.spin.JyKet
Eigenket of Jy.
See JzKet for the usage of spin eigenstates.
See also:
class sympy.physics.quantum.spin.JyBra
Eigenbra of Jy.
See JzKet for the usage of spin eigenstates.
See also:
class sympy.physics.quantum.spin.JzKet
Eigenket of Jz.
Spin state which is an eigenstate of the Jz operator. Uncoupled states, that is states
representing the interaction of multiple separate spin states, are defined as a tensor
product of states.
Parameters j : Number, Symbol
Examples
Normal States:
Defining simple spin states, both numerical and symbolic:
Rewriting the JzKet in terms of eigenkets of the Jx operator: Note: that the resulting
eigenstates are JxKet’s
>>> JzKet(1,1).rewrite(”Jx”)
|1,-1>/2 - sqrt(2)*|1,0>/2 + |1,1>/2
Get the vector representation of a state in terms of the basis elements of the Jx operator:
Uncoupled States:
Define an uncoupled state as a TensorProduct between two Jz eigenkets:
A TensorProduct can be rewritten, in which case the eigenstates that make up the tensor
product is rewritten to the new basis:
>>> TensorProduct(JzKet(1,1),JxKet(1,1)).rewrite('Jz')
|1,1>x|1,-1>/2 + sqrt(2)*|1,1>x|1,0>/2 + |1,1>x|1,1>/2
The represent method for TensorProduct’s gives the vector representation of the state.
Note that the state in the product basis is the equivalent of the tensor product of the
vector representation of the component eigenstates:
>>> represent(TensorProduct(JzKet(1,0),JzKet(1,1)))
Matrix([
[0],
[0],
[0],
[1],
[0],
[0],
[0],
[0],
[0]])
>>> represent(TensorProduct(JzKet(1,1),JxKet(1,1)), basis=Jz)
Matrix([
[ 1/2],
[sqrt(2)/2],
[ 1/2],
[ 0],
[ 0],
[ 0],
[ 0],
[ 0],
[ 0]])
class sympy.physics.quantum.spin.JzBra
Eigenbra of Jz.
See the JzKet for the usage of spin eigenstates.
See also:
class sympy.physics.quantum.spin.JxKetCoupled
Coupled eigenket of Jx.
See JzKetCoupled for the usage of coupled spin eigenstates.
See also:
class sympy.physics.quantum.spin.JxBraCoupled
Coupled eigenbra of Jx.
See JzKetCoupled for the usage of coupled spin eigenstates.
See also:
class sympy.physics.quantum.spin.JyKetCoupled
Coupled eigenket of Jy.
See JzKetCoupled for the usage of coupled spin eigenstates.
See also:
class sympy.physics.quantum.spin.JyBraCoupled
Coupled eigenbra of Jy.
See JzKetCoupled for the usage of coupled spin eigenstates.
See also:
class sympy.physics.quantum.spin.JzKetCoupled
Coupled eigenket of Jz
Spin state that is an eigenket of Jz which represents the coupling of separate spin spaces.
The arguments for creating instances of JzKetCoupled are j, m, jn and an optional jcou-
pling argument. The j and m options are the total angular momentum quantum num-
bers, as used for normal states (e.g. JzKet).
The other required parameter in jn, which is a tuple defining the jn angular momentum
quantum numbers of the product spaces. So for example, if a state represented the
coupling of the product basis state |j1 , m1 ⟩ × |j2 , m2 ⟩, the jn for this state would be (j1,
j2).
The final option is jcoupling, which is used to define how the spaces specified by jn
are coupled, which includes both the order these spaces are coupled together and the
quantum numbers that arise from these couplings. The jcoupling parameter itself is
a list of lists, such that each of the sublists defines a single coupling between the spin
spaces. If there are N coupled angular momentum spaces, that is jn has N elements, then
there must be N-1 sublists. Each of these sublists making up the jcoupling parameter
have length 3. The first two elements are the indices of the product spaces that are
considered to be coupled together. For example, if we want to couple j1 and j4 , the
indices would be 1 and 4. If a state has already been coupled, it is referenced by the
smallest index that is coupled, so if j2 and j4 has already been coupled to some j24 , then
this value can be coupled by referencing it with index 2. The final element of the sublist
is the quantum number of the coupled state. So putting everything together, into a
valid sublist for jcoupling, if j1 and j2 are coupled to an angular momentum space with
quantum number j12 with the value j12, the sublist would be (1,2,j12), N-1 of these
sublists are used in the list for jcoupling.
Note the jcoupling parameter is optional, if it is not specified, the default coupling is
taken. This default value is to coupled the spaces in order and take the quantum number
of the coupling to be the maximum value. For example, if the spin spaces are j1 , j2 , j3 ,
j4 , then the default coupling couples j1 and j2 to j12 = j1 + j2 , then, j12 and j3 are coupled
to j123 = j12 + j3 , and finally j123 and j4 to j = j123 + j4 . The jcoupling value that would
correspond to this is:
((1,2,j1+j2),(1,3,j1+j2+j3))
See also:
Examples
Defining coupled spin states for more than 2 coupled spaces with various coupling pa-
rameters:
Rewriting the JzKetCoupled in terms of eigenkets of the Jx operator: Note: that the
resulting eigenstates are JxKetCoupled
>>> JzKetCoupled(1,1,(1,1)).rewrite(”Jx”)
|1,-1,j1=1,j2=1>/2 - sqrt(2)*|1,0,j1=1,j2=1>/2 + |1,1,j1=1,j2=1>/2
The rewrite method can be used to convert a coupled state to an uncoupled state. This
is done by passing coupled=False to the rewrite function:
Get the vector representation of a state in terms of the basis elements of the Jx operator:
class sympy.physics.quantum.spin.JzBraCoupled
Coupled eigenbra of Jz.
See the JzKetCoupled for the usage of coupled spin eigenstates.
See also:
sympy.physics.quantum.spin.couple(expr, jcoupling_list=None)
Couple a tensor product of spin states
This function can be used to couple an uncoupled tensor product of spin states. All of the
eigenstates to be coupled must be of the same class. It will return a linear combination
of eigenstates that are subclasses of CoupledSpinState determined by Clebsch-Gordan
angular momentum coupling coefficients.
Parameters expr : Expr
An expression involving TensorProducts of spin states to be coupled.
Each state must be a subclass of SpinState and they all must be the
same class.
jcoupling_list : list or tuple
Elements of this list are sub-lists of length 2 specifying the order of
the coupling of the spin spaces. The length of this must be N-1, where
N is the number of states in the tensor product to be coupled. The
elements of this sublist are the same as the first two elements of each
sublist in the jcoupling parameter defined for JzKetCoupled. If this
parameter is not specified, the default value is taken, which couples
the first and second product basis spaces, then couples this new cou-
pled space to the third product space, etc
Examples
Numerical coupling of three spaces using the default coupling method, i.e. first and
second spaces couple, then this couples to the third space:
Perform this same coupling, but we define the coupling to first couple the first and third
spaces:
Examples
State
dual
Return the dual state of this one.
classmethod dual_class()
Return the class used to construct the dual.
operators
Return the operator(s) that this state is an eigenstate of
class sympy.physics.quantum.state.State
General abstract quantum state used as a base class for Ket and Bra.
class sympy.physics.quantum.state.Ket
A general time-independent Ket in quantum mechanics.
Inherits from State and KetBase. This class should be used as the base class for all
physical, time-independent Kets in a system. This class and its subclasses will be the
main classes that users will use for expressing Kets in Dirac notation [R485].
Parameters args : tuple
The list of numbers or parameters that uniquely specify the ket.
This will usually be its symbol or its quantum numbers. For time-
dependent state, this will include the time.
References
[R485]
Examples
>>> k.dual
<psi|
>>> k.dual_class()
<class 'sympy.physics.quantum.state.Bra'>
>>> k0 = Ket(0)
>>> k1 = Ket(1)
>>> 2*I*k0 - 4*k1
2*I*|0> - 4*|1>
>>> n, m = symbols('n,m')
>>> k = Ket(n,m)
>>> k
|nm>
class sympy.physics.quantum.state.Bra
A general time-independent Bra in quantum mechanics.
Inherits from State and BraBase. A Bra is the dual of a Ket [R486]. This class and
its subclasses will be the main classes that users will use for expressing Bras in Dirac
notation.
Parameters args : tuple
The list of numbers or parameters that uniquely specify the ket.
This will usually be its symbol or its quantum numbers. For time-
dependent state, this will include the time.
References
[R486]
Examples
>>> b.dual
|psi>
>>> b.dual_class()
<class 'sympy.physics.quantum.state.Ket'>
Like Kets, Bras can have compound labels and be manipulated in a similar manner:
>>> n, m = symbols('n,m')
>>> b = Bra(n,m) - I*Bra(m,n)
>>> b
-I*<mn| + <nm|
>>> b.subs(n,m)
<mm| - I*<mm|
class sympy.physics.quantum.state.TimeDepState
Base class for a general time-dependent quantum state.
This class is used as a base class for any time-dependent state. The main difference
between this class and the time-independent state is that this class takes a second argu-
ment that is the time in addition to the usual label argument.
Parameters args : tuple
The list of numbers or parameters that uniquely specify the ket.
This will usually be its symbol or its quantum numbers. For time-
dependent state, this will include the time as the final argument.
label
The label of the state.
time
The time of the state.
class sympy.physics.quantum.state.TimeDepBra
General time-dependent Bra in quantum mechanics.
This inherits from TimeDepState and BraBase and is the main class that should be used
for Bras that vary with time. Its dual is a TimeDepBra.
Parameters args : tuple
The list of numbers or parameters that uniquely specify the ket.
This will usually be its symbol or its quantum numbers. For time-
dependent state, this will include the time as the final argument.
Examples
class sympy.physics.quantum.state.TimeDepKet
General time-dependent Ket in quantum mechanics.
This inherits from TimeDepState and KetBase and is the main class that should be used
for Kets that vary with time. Its dual is a TimeDepBra.
Parameters args : tuple
The list of numbers or parameters that uniquely specify the ket.
This will usually be its symbol or its quantum numbers. For time-
dependent state, this will include the time as the final argument.
Examples
class sympy.physics.quantum.state.Wavefunction
Class for representations in continuous bases
This class takes an expression and coordinates in its constructor. It can be used to easily
calculate normalizations and probabilities.
Parameters expr : Expr
The expression representing the functional form of the w.f.
coords : Symbol or tuple
The coordinates to be integrated over, and their bounds
Examples
Particle in a box, specifying bounds in the more primitive way of using Piecewise:
>>> from sympy import Symbol, Piecewise, pi, N
>>> from sympy.functions import sqrt, sin
>>> from sympy.physics.quantum.state import Wavefunction
>>> x = Symbol('x', real=True)
>>> n = 1
>>> L = 1
>>> g = Piecewise((0, x < 0), (0, x > L), (sqrt(2//L)*sin(n*pi*x/L), True))
>>> f = Wavefunction(g, x)
>>> f.norm
1
>>> f.is_normalized
True
>>> p = f.prob()
>>> p(0)
0
>>> p(L)
0
>>> p(0.5)
(continues on next page)
Additionally, you can specify the bounds of the function and the indices in a more compact
way:
>>> from sympy import symbols, pi, diff
>>> from sympy.functions import sqrt, sin
>>> from sympy.physics.quantum.state import Wavefunction
>>> x, L = symbols('x,L', positive=True)
>>> n = symbols('n', integer=True, positive=True)
>>> g = sqrt(2/L)*sin(n*pi*x/L)
>>> f = Wavefunction(g, (x, 0, L))
>>> f.norm
1
>>> f(L+1)
0
>>> f(L-1)
sqrt(2)*sin(pi*n*(L - 1)/L)/sqrt(L)
>>> f(-1)
0
>>> f(0.85)
sqrt(2)*sin(0.85*pi*n/L)/sqrt(L)
>>> f(0.85, n=1, L=1)
sqrt(2)*sin(0.85*pi)
>>> f.is_commutative
False
All arguments are automatically sympified, so you can define the variables as strings
rather than symbols:
>>> expr = x**2
>>> f = Wavefunction(expr, 'x')
>>> type(f.variables[0])
<class 'sympy.core.symbol.Symbol'>
expr
Return the expression which is the functional form of the Wavefunction
Examples
is_commutative
Override Function’s is_commutative so that order is preserved in represented ex-
pressions
is_normalized
Returns true if the Wavefunction is properly normalized
Examples
limits
Return the limits of the coordinates which the w.f. depends on If no limits are spec-
ified, defaults to (-oo, oo).
Examples
norm
Return the normalization of the specified functional form.
This function integrates over the coordinates of the Wavefunction, with the bounds
specified.
Examples
normalize()
Return a normalized version of the Wavefunction
Examples
prob()
Return the absolute magnitude of the w.f., |ψ(x)|2
Examples
variables
Return the coordinates which the wavefunction depends on
Examples
Quantum Computation
Circuit Plot
Gates
eval_controls(qubit)
Return True/False to indicate if the controls are satisfied.
gate
The non-controlled gate that will be applied to the targets.
min_qubits
The minimum number of qubits this gate needs to act on.
nqubits
The total number of qubits this gate acts on.
For controlled gate subclasses this includes both target and control qubits, so that,
for examples the CNOT gate acts on 2 qubits.
plot_gate(circ_plot, gate_idx)
Plot the controlled gate. If simplify_cgate is true, simplify C-X and C-Z gates into
their more familiar forms.
targets
A tuple of target qubits.
class sympy.physics.quantum.gate.UGate
General gate specified by a set of targets and a target matrix.
Parameters label : tuple
A tuple of the form (targets, U), where targets is a tuple of the target
qubits and U is a unitary matrix with dimension of len(targets).
get_target_matrix(format=’sympy’)
The matrix rep. of the target part of the gate.
Parameters format : str
The format string (‘sympy’,’numpy’, etc.)
targets
A tuple of target qubits.
class sympy.physics.quantum.gate.OneQubitGate
A single qubit unitary gate base class.
class sympy.physics.quantum.gate.TwoQubitGate
A two qubit unitary gate base class.
class sympy.physics.quantum.gate.IdentityGate
The single qubit identity gate.
Parameters target : int
The target qubit this gate will apply to.
class sympy.physics.quantum.gate.HadamardGate
The single qubit Hadamard gate.
Parameters target : int
The target qubit this gate will apply to.
Examples
class sympy.physics.quantum.gate.XGate
The single qubit X, or NOT, gate.
Parameters target : int
The target qubit this gate will apply to.
class sympy.physics.quantum.gate.YGate
The single qubit Y gate.
Parameters target : int
The target qubit this gate will apply to.
class sympy.physics.quantum.gate.ZGate
The single qubit Z gate.
Parameters target : int
The target qubit this gate will apply to.
class sympy.physics.quantum.gate.TGate
The single qubit pi/8 gate.
This gate rotates the phase of the state by pi/4 if the state is |1> and does nothing if the
state is |0>.
Parameters target : int
The target qubit this gate will apply to.
class sympy.physics.quantum.gate.PhaseGate
The single qubit phase, or S, gate.
This gate rotates the phase of the state by pi/2 if the state is |1> and does nothing if the
state is |0>.
Parameters target : int
The target qubit this gate will apply to.
class sympy.physics.quantum.gate.SwapGate
Two qubit SWAP gate.
This gate swap the values of the two qubits.
Parameters label : tuple
A tuple of the form (target1, target2).
decompose(**options)
Decompose the SWAP gate into CNOT gates.
class sympy.physics.quantum.gate.CNotGate
Two qubit controlled-NOT.
This gate performs the NOT or X gate on the target qubit if the control qubits all have
the value 1.
Parameters label : tuple
A tuple of the form (control, target).
Examples
controls
A tuple of control qubits.
gate
The non-controlled gate that will be applied to the targets.
min_qubits
The minimum number of qubits this gate needs to act on.
targets
A tuple of target qubits.
sympy.physics.quantum.gate.CNOT
alias of sympy.physics.quantum.gate.CNotGate (page 1753)
sympy.physics.quantum.gate.SWAP
alias of sympy.physics.quantum.gate.SwapGate (page 1753)
sympy.physics.quantum.gate.H
alias of sympy.physics.quantum.gate.HadamardGate (page 1752)
sympy.physics.quantum.gate.X
alias of sympy.physics.quantum.gate.XGate (page 1753)
sympy.physics.quantum.gate.Y
alias of sympy.physics.quantum.gate.YGate (page 1753)
sympy.physics.quantum.gate.Z
alias of sympy.physics.quantum.gate.ZGate (page 1753)
sympy.physics.quantum.gate.T
alias of sympy.physics.quantum.gate.TGate (page 1753)
sympy.physics.quantum.gate.S
alias of sympy.physics.quantum.gate.PhaseGate (page 1753)
sympy.physics.quantum.gate.Phase
alias of sympy.physics.quantum.gate.PhaseGate (page 1753)
sympy.physics.quantum.gate.normalized(normalize)
Set flag controlling normalization of Hadamard gates by 1/sqrt(2).
This is a global setting that can be used to simplify the look of various expressions, by
leaving off the leading 1/sqrt(2) of the Hadamard gate.
Parameters normalize : bool
Should the Hadamard gate include the 1/sqrt(2) normalization factor?
When True, the Hadamard gate will have the 1/sqrt(2). When False,
the Hadamard gate will not have this factor.
sympy.physics.quantum.gate.gate_sort(circuit)
Sorts the gates while keeping track of commutation relations
This function uses a bubble sort to rearrange the order of gate application. Keeps track
of Quantum computations special commutation relations (e.g. things that apply to the
same Qubit do not commute with each other)
circuit is the Mul of gates that are to be sorted.
sympy.physics.quantum.gate.gate_simp(circuit)
Simplifies gates symbolically
It first sorts gates using gate_sort. It then applies basic simplification rules to the circuit,
e.g., XGate**2 = Identity
sympy.physics.quantum.gate.random_circuit(ngates, nqubits, gate_space=(<class
’sympy.physics.quantum.gate.XGate’>,
<class ’sympy.physics.quantum.gate.YGate’>,
<class ’sympy.physics.quantum.gate.ZGate’>,
<class ’sympy.physics.quantum.gate.PhaseGate’>,
<class ’sympy.physics.quantum.gate.TGate’>,
<class ’sympy.physics.quantum.gate.HadamardGate’>,
<class ’sympy.physics.quantum.gate.CNotGate’>,
<class ’sympy.physics.quantum.gate.SwapGate’>))
Return a random circuit of ngates and nqubits.
This uses an equally weighted sample of (X, Y, Z, S, T, H, CNOT, SWAP) gates.
Parameters ngates : int
The number of gates in the circuit.
nqubits : int
The number of qubits in the circuit.
gate_space : tuple
A tuple of the gate classes that will be used in the circuit. Repeating
gate classes multiple times in this tuple will increase the frequency
they appear in the random circuit.
class sympy.physics.quantum.gate.CGateS
Version of CGate that allows gate simplifications. I.e. cnot looks like an oplus, cphase
has dots, etc.
Grover’s Algorithm
• Generalize the algorithm for an unknown function that returns 1 on multiple qubit states,
not just one.
• Implement _represent_ZGate in OracleGate
class sympy.physics.quantum.grover.OracleGate
A black box gate.
The gate marks the desired qubits of an unknown function by flipping the sign of the
qubits. The unknown function returns true when it finds its desired qubits and false
otherwise.
Parameters qubits : int
Number of qubits.
oracle : callable
A callable function that returns a boolean on a computational basis.
Examples
Apply an Oracle gate that flips the sign of |2> on different qubits:
search_function
The unknown function that helps find the sought after qubits.
targets
A tuple of target qubits.
class sympy.physics.quantum.grover.WGate
General n qubit W Gate in Grover’s algorithm.
The gate performs the operation 2|phi><phi| - 1 on some qubits. |phi> = (tensor
product of n Hadamards)*(|0> with n qubits)
Parameters nqubits : int
The number of qubits to operate on
sympy.physics.quantum.grover.superposition_basis(nqubits)
Creates an equal superposition of the computational basis.
Parameters nqubits : int
The number of qubits.
Returns state : Qubit
An equal superposition of the computational basis with nqubits.
Examples
sympy.physics.quantum.grover.grover_iteration(qstate, oracle)
Applies one application of the Oracle and W Gate, WV.
Parameters qstate : Qubit
A superposition of qubits.
oracle : OracleGate
The black box operator that flips the sign of the desired basis qubits.
Returns Qubit : The qubits after applying the Oracle and W gate.
Examples
Examples
QFT
Qubit
Examples
>>> q.nqubits
4
>>> len(q)
4
>>> q.dimension
4
>>> q.qubit_values
(0, 1, 0, 1)
>>> q.flip(1)
|0111>
>>> ip = Dagger(q)*q
>>> ip
<0101|0101>
>>> ip.doit()
1
class sympy.physics.quantum.qubit.QubitBra
A multi-qubit bra in the computational (z) basis.
We use the normal convention that the least significant qubit is on the right, so |00001>
has a 1 in the least significant qubit.
Parameters values : list, str
The qubit values as a list of ints ([0,0,0,1,1,]) or a string (‘011’).
See also:
class sympy.physics.quantum.qubit.IntQubit
A qubit ket that store integers as binary numbers in qubit values.
The differences between this class and Qubit are:
Examples
>>> q = IntQubit(Qubit('101'))
>>> q
|5>
>>> q.as_int()
5
>>> q.nqubits
3
>>> q.qubit_values
(1, 0, 1)
>>> Qubit(q)
|101>
class sympy.physics.quantum.qubit.IntQubitBra
A qubit bra that store integers as binary numbers in qubit values.
sympy.physics.quantum.qubit.qubit_to_matrix(qubit, format=’sympy’)
Converts an Add/Mul of Qubit objects into it’s matrix representation
This function is the inverse of matrix_to_qubit and is a shorthand for repre-
sent(qubit).
sympy.physics.quantum.qubit.matrix_to_qubit(matrix)
Convert from the matrix repr. to a sum of Qubit objects.
Parameters matrix : Matrix, numpy.matrix, scipy.sparse
The matrix to build the Qubit representation of. This works with
sympy matrices, numpy matrices and scipy.sparse sparse matrices.
Examples
sympy.physics.quantum.qubit.matrix_to_density(mat)
Works by finding the eigenvectors and eigenvalues of the matrix. We know we can de-
compose rho by doing: sum(EigenVal*|Eigenvect><Eigenvect|)
sympy.physics.quantum.qubit.measure_all(qubit, format=’sympy’, normal-
ize=True)
Perform an ensemble measurement of all qubits.
Parameters qubit : Qubit, Add
The qubit to measure. This can be any Qubit or a linear combination
of them.
format : str
The format of the intermediate matrices to use. Possible values
are (‘sympy’,’numpy’,’scipy.sparse’). Currently only ‘sympy’ is imple-
mented.
Returns result : list
A list that consists of primitive states and their probabilities.
Examples
>>> c = H(0)*H(1)*Qubit('00')
>>> c
H(0)*H(1)*|00>
>>> q = qapply(c)
>>> measure_all(q)
[(|00>, 1/4), (|01>, 1/4), (|10>, 1/4), (|11>, 1/4)]
format : str
The format of the intermediate matrices to use. Possible values
are (‘sympy’,’numpy’,’scipy.sparse’). Currently only ‘sympy’ is imple-
mented.
Returns result : list
A list that consists of primitive states and their probabilities.
Examples
>>> c = H(0)*H(1)*Qubit('00')
>>> c
H(0)*H(1)*|00>
>>> q = qapply(c)
>>> measure_partial(q, (0,))
[(sqrt(2)*|00>/2 + sqrt(2)*|10>/2, 1/2), (sqrt(2)*|01>/2 + sqrt(2)*|11>/2, 1/2)]
Shor’s Algorithm
Analytic Solutions
Particle in a Box
Optics Module
Abstract
Gaussian Optics
Gaussian optics.
The module implements:
• Ray transfer matrices for geometrical and gaussian optics.
See RayTransferMatrix, GeometricRay and BeamParameter
• Conjugation relations for geometrical and gaussian optics.
See geometric_conj*, gauss_conj and conjugate_gauss_beams
The conventions for the distances are as follows:
focal distance positive for convergent lenses
object distance positive for real objects
image distance positive for real images
class sympy.physics.optics.gaussopt.RayTransferMatrix
Base class for a Ray Transfer Matrix.
It should be used if there isn’t already a more specific subclass mentioned in See Also.
Parameters parameters : A, B, C and D or 2x2 matrix (Matrix(2, 2, [A, B, C,
D]))
See also:
GeometricRay (page 1769), BeamParameter (page 1770), FreeSpace (page 1766), Fla-
tRefraction (page 1767), CurvedRefraction (page 1767), FlatMirror (page 1768),
CurvedMirror (page 1768), ThinLens (page 1768)
References
[R465]
Examples
>>> mat.A
1
>>> f = Symbol('f')
>>> lens = ThinLens(f)
>>> lens
Matrix([
[ 1, 0],
[-1/f, 1]])
>>> lens.C
-1/f
Attributes
cols
rows
A
The A parameter of the Matrix.
Examples
B
The B parameter of the Matrix.
Examples
C
The C parameter of the Matrix.
Examples
D
The D parameter of the Matrix.
Examples
class sympy.physics.optics.gaussopt.FreeSpace
Ray Transfer Matrix for free space.
Parameters distance
See also:
RayTransferMatrix (page 1764)
Examples
Attributes
cols
rows
class sympy.physics.optics.gaussopt.FlatRefraction
Ray Transfer Matrix for refraction.
Parameters n1 : refractive index of one medium
n2 : refractive index of other medium
See also:
RayTransferMatrix (page 1764)
Examples
Attributes
cols
rows
class sympy.physics.optics.gaussopt.CurvedRefraction
Ray Transfer Matrix for refraction on curved interface.
Parameters R : radius of curvature (positive for concave)
n1 : refractive index of one medium
n2 : refractive index of other medium
See also:
RayTransferMatrix (page 1764)
Examples
Attributes
cols
rows
class sympy.physics.optics.gaussopt.FlatMirror
Ray Transfer Matrix for reflection.
See also:
RayTransferMatrix (page 1764)
Examples
Attributes
cols
rows
class sympy.physics.optics.gaussopt.CurvedMirror
Ray Transfer Matrix for reflection from curved surface.
Parameters R : radius of curvature (positive for concave)
See also:
RayTransferMatrix (page 1764)
Examples
Attributes
cols
rows
class sympy.physics.optics.gaussopt.ThinLens
Ray Transfer Matrix for a thin lens.
Parameters f : the focal distance
See also:
RayTransferMatrix (page 1764)
Examples
Attributes
cols
rows
class sympy.physics.optics.gaussopt.GeometricRay
Representation for a geometric ray in the Ray Transfer Matrix formalism.
Parameters h : height, and
angle : angle, or
matrix : a 2x1 matrix (Matrix(2, 1, [height, angle]))
See also:
RayTransferMatrix (page 1764)
Examples
Attributes
cols
rows
angle
The angle with the optical axis.
Examples
height
The distance from the optical axis.
Examples
class sympy.physics.optics.gaussopt.BeamParameter
Representation for a gaussian ray in the Ray Transfer Matrix formalism.
Parameters wavelen : the wavelength,
z : the distance to waist, and
w : the waist, or
z_r : the rayleigh range
See also:
RayTransferMatrix (page 1764)
References
[R466], [R467]
Examples
>>> p.q.n()
1.0 + 5.92753330865999*I
>>> p.w_0.n()
0.00100000000000000
>>> p.z_r.n()
5.92753330865999
divergence
Half of the total angular spread.
Examples
gouy
The Gouy phase.
Examples
q
The complex parameter representing the beam.
Examples
radius
The radius of curvature of the phase front.
Examples
w
The beam radius at 1/e2 intensity.
See also:
Examples
w_0
The beam waist (minimal radius).
See also:
Examples
waist_approximation_limit
The minimal waist for which the gauss beam approximation is valid.
The gauss beam is a solution to the paraxial equation. For curvatures that are too
great it is not a valid approximation.
Examples
sympy.physics.optics.gaussopt.waist2rayleigh(w, wavelen)
Calculate the rayleigh range from the waist of a gaussian beam.
See also:
rayleigh2waist (page 1773), BeamParameter (page 1770)
Examples
sympy.physics.optics.gaussopt.rayleigh2waist(z_r, wavelen)
Calculate the waist from the rayleigh range of a gaussian beam.
See also:
waist2rayleigh (page 1772), BeamParameter (page 1770)
Examples
sympy.physics.optics.gaussopt.geometric_conj_ab(a, b)
Conjugation relation for geometrical beams under paraxial conditions.
Takes the distances to the optical element and returns the needed focal distance.
See also:
geometric_conj_af (page 1773), geometric_conj_bf (page 1774)
Examples
sympy.physics.optics.gaussopt.geometric_conj_af(a, f)
Conjugation relation for geometrical beams under paraxial conditions.
Takes the object distance (for geometric_conj_af) or the image distance (for geomet-
ric_conj_bf) to the optical element and the focal distance. Then it returns the other
distance needed for conjugation.
See also:
geometric_conj_ab (page 1773)
Examples
sympy.physics.optics.gaussopt.geometric_conj_bf(a, f)
Conjugation relation for geometrical beams under paraxial conditions.
Takes the object distance (for geometric_conj_af) or the image distance (for geomet-
ric_conj_bf) to the optical element and the focal distance. Then it returns the other
distance needed for conjugation.
See also:
geometric_conj_ab (page 1773)
Examples
sympy.physics.optics.gaussopt.gaussian_conj(s_in, z_r_in, f)
Conjugation relation for gaussian beams.
Parameters s_in : the distance to optical element from the waist
z_r_in : the rayleigh range of the incident beam
f : the focal length of the optical element
Returns a tuple containing (s_out, z_r_out, m)
s_out : the distance between the new waist and the optical element
z_r_out : the rayleigh range of the emergent beam
m : the ration between the new and the old waists
Examples
sympy.physics.optics.gaussopt.conjugate_gauss_beams(wavelen, waist_in,
waist_out, **kwargs)
Find the optical setup conjugating the object/image waists.
Parameters wavelen : the wavelength of the beam
waist_in and waist_out : the waists to be conjugated
f : the focal distance of the element used in the conjugation
Returns a tuple containing (s_in, s_out, f)
s_in : the distance before the optical element
s_out : the distance after the optical element
f : the focal distance of the optical element
Examples
Medium
Contains
• Medium
class sympy.physics.optics.medium.Medium
This class represents an optical medium. The prime reason to implement this is to facil-
itate refraction, Fermat’s principle, etc.
An optical medium is a material through which electromagnetic waves propagate. The
permittivity and permeability of the medium define how electromagnetic waves propa-
gate in it.
Parameters name: string
References
[R468]
Examples
intrinsic_impedance
Returns intrinsic impedance of the medium.
The intrinsic impedance of a medium is the ratio of the transverse components of the
electric and magnetic fields of the electromagnetic wave travelling in the medium.
In a region with no electrical conductivity it simplifies to the square root of ratio of
magnetic permeability to electric permittivity.
Examples
permeability
Returns magnetic permeability of the medium.
Examples
permittivity
Returns electric permittivity of the medium.
Examples
refractive_index
Returns refractive index of the medium.
Examples
speed
Returns speed of the electromagnetic wave travelling in the medium.
Examples
Utilities
Contains
• refraction_angle
• deviation
• brewster_angle
• lens_makers_formula
• mirror_formula
• lens_formula
• hyperfocal_distance
• transverse_magnification
sympy.physics.optics.utils.refraction_angle(incident, medium1, medium2, nor-
mal=None, plane=None)
This function calculates transmitted vector after refraction at planar surface. medium1
and medium2 can be M edium or any sympifiable object.
Examples
Examples
sympy.physics.optics.utils.brewster_angle(medium1, medium2)
This function calculates the Brewster’s angle of incidence to Medium 2 from Medium 1
in radians.
Parameters medium 1 : Medium or sympifiable
Refractive index of Medium 1
medium 2 : Medium or sympifiable
Refractive index of Medium 1
Examples
Examples
sympy.physics.optics.utils.mirror_formula(focal_length=None, u=None,
v=None)
This function provides one of the three parameters when two of them are supplied. This
is valid only for paraxial rays.
Parameters focal_length : sympifiable
Focal length of the mirror.
u : sympifiable
Distance of object from the pole on the principal axis.
v : sympifiable
Distance of the image from the pole on the principal axis.
Examples
Examples
sympy.physics.optics.utils.hyperfocal_distance(f, N, c)
Parameters f: sympifiable
Focal length of a given lens
N: sympifiable
F-number of a given lens
c: sympifiable
Circle of Confusion (CoC) of a given image format
Example
sympy.physics.optics.utils.transverse_magnification(si, so)
Calculates the transverse magnification, which is the ratio of the image size to the object
size.
Parameters so: sympifiable
Lens-object distance
si: sympifiable
Lens-image distance
Example
Waves
This module has all the classes and functions related to waves in optics.
Contains
• TWave
class sympy.physics.optics.waves.TWave(amplitude, frequency=None, phase=0,
time_period=None, n=n)
This is a simple transverse sine wave travelling in a one-dimensional space. Basic prop-
erties are required at the time of creation of the object, but they can be changed later
with respective methods provided.
It is represented as A × cos(k ∗ x − ω × t + ϕ), where A is the amplitude, ω is the angular
velocity, k is the wavenumber (spatial frequency), x is a spatial variable to represent the
position on the dimension on which the wave propagates, and ϕ is the phase angle of the
wave.
Raises ValueError : When neither frequency nor time period is provided
or they are not consistent.
TypeError : When anything other than TWave objects is added.
Examples
Arguments
amplitude
Returns the amplitude of the wave.
Examples
angular_velocity
Returns the angular velocity of the wave, in radians per second.
Examples
frequency
Returns the frequency of the wave, in cycles per second.
Examples
phase
Returns the phase angle of the wave, in radians.
Examples
speed
Returns the propagation speed of the wave, in meters per second. It is dependent
on the propagation medium.
Examples
time_period
Returns the temporal period of the wave, in seconds per cycle.
Examples
wavelength
Returns the wavelength (spatial period) of the wave, in meters per cycle. It depends
on the medium of the wave.
Examples
wavenumber
Returns the wavenumber of the wave, in radians per meter.
Examples
Continuum Mechanics
Abstract
Beam
Beam (Docstrings)
Beam
This module can be used to solve 2D beam bending problems with singularity functions in
mechanics.
class sympy.physics.continuum_mechanics.beam.Beam(length, elastic_modulus,
second_moment, vari-
able=x, base_char=’C’)
A Beam is a structural element that is capable of withstanding load primarily by resist-
ing against bending. Beams are characterized by their cross sectional profile(Second
moment of area), their length and their material.
Note: While solving a beam bending problem, a user should choose its own sign con-
vention and should stick to it. The results will automatically follow the chosen sign
convention.
Examples
There is a beam of length 4 meters. A constant distributed load of 6 N/m is applied from
half of the beam till the end. There are two simple supports below the beam, one at the
starting point and another at the ending point of the beam. The deflection of the beam
at the end is restricted.
Using the sign convention of downwards forces being positive.
>>> from sympy.physics.continuum_mechanics.beam import Beam
>>> from sympy import symbols, Piecewise
>>> E, I = symbols('E, I')
>>> R1, R2 = symbols('R1, R2')
>>> b = Beam(4, E, I)
>>> b.apply_load(R1, 0, -1)
>>> b.apply_load(6, 2, 0)
>>> b.apply_load(R2, 4, -1)
>>> b.bc_deflection = [(0, 0), (4, 0)]
>>> b.boundary_conditions
{'deflection': [(0, 0), (4, 0)], 'slope': []}
>>> b.load
R1*SingularityFunction(x, 0, -1) + R2*SingularityFunction(x, 4, -1) +␣
,→6*SingularityFunction(x, 2, 0)
>>> b.bending_moment()
-3*SingularityFunction(x, 0, 1) + 3*SingularityFunction(x, 2, 2) -␣
,→9*SingularityFunction(x, 4, 1)
>>> b.slope()
(-3*SingularityFunction(x, 0, 2)/2 + SingularityFunction(x, 2, 3) -␣
,→9*SingularityFunction(x, 4, 2)/2 + 7)/(E*I)
>>> b.deflection()
(7*x - SingularityFunction(x, 0, 3)/2 + SingularityFunction(x, 2, 4)/4 -␣
,→3*SingularityFunction(x, 4, 3)/2)/(E*I)
>>> b.deflection().rewrite(Piecewise)
(7*x - Piecewise((x**3, x > 0), (0, True))/2
- 3*Piecewise(((x - 4)**3, x - 4 > 0), (0, True))/2
+ Piecewise(((x - 2)**4, x - 2 > 0), (0, True))/4)/(E*I)
applied_loads
Returns a list of all loads applied on the beam object. Each load in the list is a tuple
of form (value, start, order, end).
Examples
Examples
+ 2*SingularityFunction(x, 3, 0) + 2*SingularityFunction(x, 3, 2)
apply_support(loc, type=’fixed’)
This method applies support to a particular beam object.
Parameters loc : Sympifyable
Location of point at which support is applied.
type : String
Determines type of Beam support applied. To apply support structure
with - zero degree of freedom, type = “fixed” - one degree of freedom,
type = “pin” - two degrees of freedom, type = “roller”
Examples
bending_moment()
Returns a Singularity Function expression which represents the bending moment
curve of the Beam object.
Examples
boundary_conditions
Returns a dictionary of boundary conditions applied on the beam. The dictionary has
three kewwords namely moment, slope and deflection. The value of each keyword is
a list of tuple, where each tuple contains loaction and value of a boundary condition
in the format (location, value).
Examples
Here the deflection of the beam should be 2 at 0. Similarly, the slope of the beam
should be 1 at 0.
deflection()
Returns a Singularity Function expression which represents the elastic curve or
deflection of the Beam object.
Examples
elastic_modulus
Young’s Modulus of the Beam.
join(beam, via=’fixed’)
This method joins two beams to make a new composite beam system. Passed Beam
class instance is attached to the right end of calling object. This method can be used
to form beams having Discontinuous values of Elastic modulus or Second moment.
Parameters beam : Beam class object
The Beam object which would be connected to the right of calling
object.
via : String
States the way two Beam object would get connected - For ax-
ially fixed Beams, via=”fixed” - For Beams connected via hinge,
via=”hinge”
Examples
There is a cantilever beam of length 4 meters. For first 2 meters its moment of
inertia is 1.5 ∗ I and I for the other end. A pointload of magnitude 4 N is applied from
the top at its free end.
>>> b.slope()
(((80*SingularityFunction(x, 0, 1) - 10*SingularityFunction(x, 0, 2) +␣
,→10*SingularityFunction(x, 4, 2))/I - 120/I)/E + 80.0/
,→(E*I))*SingularityFunction(x, 2, 0)
+ 0.666666666666667*(80*SingularityFunction(x, 0, 1) -␣
,→10*SingularityFunction(x, 0, 2) + 10*SingularityFunction(x, 4,␣
,→2))*SingularityFunction(x, 0, 0)/(E*I)
- 0.666666666666667*(80*SingularityFunction(x, 0, 1) -␣
,→10*SingularityFunction(x, 0, 2) + 10*SingularityFunction(x, 4,␣
,→2))*SingularityFunction(x, 2, 0)/(E*I)
length
Length of the Beam.
load
Returns a Singularity Function expression which represents the load distribution
curve of the Beam object.
Examples
max_bmoment()
Returns maximum Shear force and its coordinate in the Beam object.
max_deflection()
Returns point of max deflection and its coresponding deflection value in a Beam
object.
max_shear_force()
Returns maximum Shear force and its coordinate in the Beam object.
plot_bending_moment(subs=None)
Returns a plot for Bending moment present in the Beam object.
Parameters subs : dictionary
Python dictionary containing Symbols as key and their corresponding
values.
Examples
+ 5000*SingularityFunction(x, 4, 2) - 31250*SingularityFunction(x, 8, 1)
- 5000*SingularityFunction(x, 8, 2) for x over (0.0, 8.0)
plot_deflection(subs=None)
Returns a plot for deflection curve of the Beam object.
Parameters subs : dictionary
Examples
+ 1.04166666666667e-5*SingularityFunction(x, 2, 3) + 5.20833333333333e-
,→6*SingularityFunction(x, 4, 4)
- 6.51041666666667e-5*SingularityFunction(x, 8, 3) - 5.20833333333333e-
,→6*SingularityFunction(x, 8, 4)
plot_loading_results(subs=None)
Returns Axes object containing subplots of Shear Force, Bending Moment, Slope
and Deflection of the Beam object.
Parameters subs : dictionary
Python dictionary containing Symbols as key and their corresponding
values.
.. note::
This method only works if numpy and matplotlib libraries are installed
on the system.
Examples
plot_shear_force(subs=None)
Returns a plot for Shear force present in the Beam object.
Parameters subs : dictionary
Python dictionary containing Symbols as key and their corresponding
values.
Examples
+ 10000*SingularityFunction(x, 4, 1) - 31250*SingularityFunction(x, 8, 0)
- 10000*SingularityFunction(x, 8, 1) for x over (0.0, 8.0)
plot_slope(subs=None)
Returns a plot for slope of deflection curve of the Beam object.
Parameters subs : dictionary
Python dictionary containing Symbols as key and their corresponding
values.
Examples
+ 2.08333333333333e-5*SingularityFunction(x, 4, 3) - 0.
,→0001953125*SingularityFunction(x, 8, 2)
point_cflexure()
Returns a Set of point(s) with zero bending moment and where bending moment
curve of the beam object changes its sign from negative to positive or vice versa.
Examples
There is is 10 meter long overhanging beam. There are two simple supports below
the beam. One at the start and another one at a distance of 6 meters from the start.
Point loads of magnitude 10KN and 20KN are applied at 2 meters and 4 meters from
start respectively. A Uniformly distribute load of magnitude of magnitude 3KN/m is
also applied on top starting from 6 meters away from starting point till end. Using
the sign convention of upward forces and clockwise moment being positive.
reaction_loads
Returns the reaction forces in a dictionary.
Examples
+ 2*SingularityFunction(x, 3, 0) + 2*SingularityFunction(x, 3, 2)
>>> b.remove_load(-2, 2, 2, end = 3)
>>> b.load
-3*SingularityFunction(x, 0, -2) + 4*SingularityFunction(x, 2, -1)
second_moment
Second moment of area of the Beam.
shear_force()
Returns a Singularity Function expression which represents the shear force curve
of the Beam object.
Examples
applied from the top of the beam at the starting point. There are two simple supports
below the beam. One at the end and another one at a distance of 10 meters from
the start. The deflection is restricted at both the supports.
Using the sign convention of upward forces and clockwise moment being positive.
slope()
Returns a Singularity Function expression which represents the slope the elastic
curve of the Beam object.
Examples
solve_for_reaction_loads(*reactions)
Solves for the reaction forces.
Examples
variable
A symbol that can be used as a variable along the length of the beam while repre-
senting load distribution, shear force curve, bending moment, slope curve and the
deflection curve. By default, it is set to Symbol('x'), but this property is mutable.
Examples
Note: While solving a beam bending problem, a user should choose its own sign con-
vention and should stick to it. The results will automatically follow the chosen sign con-
vention. This class assumes that any kind of distributed load/moment is applied through
out the span of a beam.
References
[R461]
Examples
- m)/(2*E*I)]
>>> b.deflection()
[0, -l**2*q*x**2/(12*E*I) + l**2*x**2*(A*G*l**2*q - 2*A*G*l*m + 12*E*I*q)/
,→(8*E*I*(A*G*l**2 + 12*E*I))
Examples
There is a beam of length 4 meters. The slope at 0 should be 4 along the x-axis and
0 along others. At the other end of beam, deflection along all the three axes should
be zero.
Here the deflection of the beam should be 0 along all the three axes at 4. Similarly,
the slope of the beam should be 4 along x-axis and 0 along y and z axis at 0.
deflection()
Returns a three element list representing deflection curve along all the three axes.
load_vector
Returns a three element list representing the load vector.
moment_load_vector
Returns a three element list representing moment loads on Beam.
second_moment
Second moment of area of the Beam.
shear_force()
Returns a list of three expressions which represents the shear force curve of the
Beam object along all three axes.
shear_modulus
Young’s Modulus of the Beam.
slope()
Returns a three element list representing slope of deflection curve along all the three
axes.
solve_for_reaction_loads(*reaction)
Solves for the reaction forces.
Examples
torsional_moment()
Returns expression of Torsional moment present inside the Beam object.
To make this document easier to read, we are going to enable pretty printing.
Beam
Examples
Example 1
A beam of length 9 meters is having a fixed support at the start. A distributed constant load
of 8 kN/m is applied downward from the starting point till 5 meters away from the start. A
clockwise moment of 50 kN-m is applied at 5 meters away from the start of the beam. A
downward point load of 12 kN is applied at the end.
\\\\| 8 KN/m
\\\\|_____________
\\\\|| | | | | | | 50 KN-m 12 KN
\\\\|V V V V V V V | |
\\\\|________________|_______________V
\\\\|________________|_______________|
\\\\| :
\\\\|----------------|---------------|
5.0 m 4.0 m
Note: Since a user is free to choose its own sign convention we are considering the upward
forces and clockwise bending moment being positive.
Example 2
|| 8 N 120 Nm
\/______________________________________________|
|_______________________________________________|
/\ /\
|------------|---------------------------------|
10 m 20 m
Note: Using the sign convention of upward forces and clockwise moment being positive.
Example 3
A beam of length 6 meters is having a roller support at the start and a hinged support at
the end. A clockwise moment of 1.5 kN-m is applied at the mid of the beam. A constant
distributed load of 3 kN/m and a ramp load of 1 kN/m is applied from the mid till the end of
the beam.
Note: Using the sign convention of upward forces and clockwise moment being positive.
8 120
␣
,→──────────────────────────────────────────────────────────────────────────────────────────────────
E⋅I
Example 4
An overhanging beam of length 8 meters is pinned at 1 meter from starting point and sup-
ported by a roller 1 meter before the other end. It is subjected to a distributed constant load
of 10 KN/m from the starting point till 2 meters away from it. Two pointloads of 20KN and
8KN are applied at 5 meters and 7.5 meters away from the starting point respectively.
---> x
|
v y
10 KN/m
_____________ 20 KN 8 KN
| | | | | | | | |
V V V V V V V V V
_______________________________________________
|_______________________________________________|
/\ O
|-----|------|-----------------|----------|--|--|
1m 1m 3m 2m .5m .5m
>>> b.shear_force()
1 0 1 0 0 0
10⋅<x> - 26.0⋅<x - 1> - 10⋅<x - 2> + 20⋅<x - 5> - 22.0⋅<x - 7> + 8⋅<x - 7.5>
>>> b.bending_moment()
2 1 2 1 1 1
5⋅<x> - 26.0⋅<x - 1> - 5⋅<x - 2> + 20⋅<x - 5> - 22.0⋅<x - 7> + 8⋅<x - 7.5>
(continues on next page)
3 3
───────────────────────────────────────────────────────────────────────────────────────────────────
E⋅I
>>> b.deflection()
4 4 3 ␣
,→ 3
5⋅<x> 3 5⋅<x - 2> 10⋅<x - 5> ␣
,→ 3 4⋅<x - 7.5>
28.2916666666667⋅x + ────── - 4.33333333333333⋅<x - 1> - ────────── + ─────────── -␣
,→3.66666666666667⋅<x - 7> + ──────────── - 28.7083333333333
12 12 3 ␣
,→ 3
─────────────────────────────────────────────────────────────────────────────────────────────────────
E⋅I
Example 5
A cantilever beam of length 6 meters is under downward distributed constant load with mag-
nitude of 4.0 KN/m from starting point till 2 meters away from it. A ramp load of 1 kN/m
applied from the mid till the end of the beam. A point load of 12KN is also applied in same
direction 4 meters away from start.
---> x .
| . |
v y 12 KN . | |
| . | | |
V . | | | |
\\\\| 4 KN/m . | | | | |
\\\\|___________ . 1 KN/m | |
\\\\|| | | | | | . V V V V V V V
\\\\|V V V V V V |---------------|
\\\\|________________________________
\\\\|________________________________|
\\\\| : : :
\\\\|----------|-----|----|----------|
2.0 m 1m 1m 2.0 m
2 2
>>> b.shear_force()
-1 0 2 ␣
,→ 2
───────── - ─────── + 4⋅<x> - 4⋅<x - 2> + ──────── + 12⋅<x - 4> - <x - 6> -␣
,→────────
2 2 2 2
>>> b.bending_moment()
0 1 3 2 ␣
,→3
2 2 6 2 6
>>> b.bc_deflection = [(0, 0)]
>>> b.bc_slope = [(0, 0)]
>>> b.slope()
1 2 3 3 4 3 4
157⋅<x> 49⋅<x> 2⋅<x> 2⋅<x - 2> <x - 3> 2 <x - 6> <x - 6>
──────── - ─────── + ────── - ────────── + ──────── + 6⋅<x - 4> - ──────── - ────────
2 4 3 3 24 6 24
──────────────────────────────────────────────────────────────────────────────────────
E⋅I
>>> b.deflection()
2 3 4 4 5 4 5
157⋅<x> 49⋅<x> <x> <x - 2> <x - 3> 3 <x - 6> <x - 6>
──────── - ─────── + ──── - ──────── + ──────── + 2⋅<x - 4> - ──────── - ────────
4 12 6 6 120 24 120
──────────────────────────────────────────────────────────────────────────────────
E⋅I
Example 6
2 KN/m ---> x
_________________ |
| | | | | | | | | v y
V V V V V V V V V 5 KN-m
____________________________________________________|
O____________________________________________________|
(continues on next page)
Example 7
There is a beam of length l, fixed at both ends. A concentrated point load of magnitude F is
applied in downward direction at mid-point of the beam.
^ y
|
---> x
\\\\| F |\\\\
(continues on next page)
>>> b.load
-2 -2 -1 -1 -1
F⋅l⋅<x> F⋅l⋅<-l + x> F⋅<x> l F⋅<-l + x>
- ───────── + ────────────── + ─────── - F⋅<- ─ + x> + ────────────
8 8 2 2 2
>>> b.shear_force()
-1 -1 0 0 0
F⋅l⋅<x> F⋅l⋅<-l + x> F⋅<x> l F⋅<-l + x>
- ───────── + ────────────── + ────── - F⋅<- ─ + x> + ───────────
8 8 2 2 2
>>> b.bending_moment()
0 0 1 1 1
F⋅l⋅<x> F⋅l⋅<-l + x> F⋅<x> l F⋅<-l + x>
- ──────── + ───────────── + ────── - F⋅<- ─ + x> + ───────────
8 8 2 2 2
>>> b.slope()
2
l
1 1 2 F⋅<- ─ + x> 2
F⋅l⋅<x> F⋅l⋅<-l + x> F⋅<x> 2 F⋅<-l + x>
- ──────── + ───────────── + ────── - ──────────── + ───────────
8 8 4 2 4
────────────────────────────────────────────────────────────────
E⋅I
(continues on next page)
>>> b.deflection()
3
l
2 2 3 F⋅<- ─ + x> 3
F⋅l⋅<x> F⋅l⋅<-l + x> F⋅<x> 2 F⋅<-l + x>
- ──────── + ───────────── + ────── - ──────────── + ───────────
16 16 12 6 12
────────────────────────────────────────────────────────────────
E⋅I
Example 8
There is a beam of length 4*l, having a hinge connector at the middle. It is having a fixed
support at the start and also has two rollers at a distance of l and 4*l from the starting point.
A concentrated point load P is also applied at a distance of 3*l from the starting point.
---> x
\\\\| P |
\\\\| | v y
\\\\| V
\\\\|_____________________ _______________________
\\\\|_____________________O_______________________|
\\\\| /\ : /\
\\\\| oooo : oooo
\\\\|----------|-----------|----------|-----------|
l l l l
>>> b.load
-2 -1 -1 -1
P⋅l⋅<x> 3⋅P⋅<x> 5⋅P⋅<-l + x> -1 P⋅<-4⋅l + x>
- ───────── + ───────── - ────────────── + P⋅<-3⋅l + x> - ──────────────
4 4 4 2
(continues on next page)
>>> b.shear_force()
-1 0 0 0
P⋅l⋅<x> 3⋅P⋅<x> 5⋅P⋅<-l + x> 0 P⋅<-4⋅l + x>
- ───────── + ──────── - ───────────── + P⋅<-3⋅l + x> - ─────────────
4 4 4 2
>>> b.bending_moment()
0 1 1 1
P⋅l⋅<x> 3⋅P⋅<x> 5⋅P⋅<-l + x> 1 P⋅<-4⋅l + x>
- ──────── + ──────── - ───────────── + P⋅<-3⋅l + x> - ─────────────
4 4 4 2
>>> b.slope()
⎛ 2 2 2 2⎞ ⎛ 1 ␣
,→ 2 2 2⎞ ⎛ 1 2 ␣
,→ 2 2⎞
⎜5⋅P⋅l P⋅<-2⋅l + x> P⋅<-3⋅l + x> P⋅<-4⋅l + x> ⎟ 0 ⎜ P⋅l⋅<x> ␣
,→3⋅P⋅<x> 5⋅P⋅<-l + x> P⋅<-2⋅l + x> ⎟ 0 ⎜ P⋅l⋅<x> 3⋅P⋅<x> 5⋅P⋅<-l +␣
,→x> P⋅<-2⋅l + x> ⎟ 0
⎜────── - ───────────── + ───────────── - ─────────────⎟⋅<-2⋅l + x> ⎜- ──────── +␣
,→──────── - ───────────── + ─────────────⎟⋅<x> ⎜- ──────── + ──────── -␣
,→───────────── + ─────────────⎟⋅<-2⋅l + x>
⎝ 48 4 2 4 ⎠ ⎝ 4 ␣
,→ 8 8 4 ⎠ ⎝ 4 8 8 ␣
,→ 4 ⎠
──────────────────────────────────────────────────────────────────── +␣
,→──────────────────────────────────────────────────────────── -␣
,→───────────────────────────────────────────────────────────────────
E⋅I ␣
,→ E⋅I E⋅I
>>> b.deflection()
⎛ 2 3 3 3⎞ ⎛ 2 3 ␣
,→ 3 3⎞ ⎛ 3 2 ␣
,→ 3 3 3⎞
⎜ P⋅l⋅<x> P⋅<x> 5⋅P⋅<-l + x> P⋅<-2⋅l + x> ⎟ 0 ⎜ P⋅l⋅<x> P⋅<x> ␣
,→5⋅P⋅<-l + x> P⋅<-2⋅l + x> ⎟ 0 ⎜7⋅P⋅l 5⋅P⋅l ⋅(-2⋅l + x) P⋅<-2⋅l␣
,→+ x> P⋅<-3⋅l + x> P⋅<-4⋅l + x> ⎟ 0
⎜- ──────── + ────── - ───────────── + ─────────────⎟⋅<x> ⎜- ──────── + ────── -␣
,→───────────── + ─────────────⎟⋅<-2⋅l + x> ⎜────── + ───────────────── -␣
,→───────────── + ───────────── - ─────────────⎟⋅<-2⋅l + x>
⎝ 8 8 24 12 ⎠ ⎝ 8 8 ␣
,→ 24 12 ⎠ ⎝ 24 48 12 ␣
,→ 6 12 ⎠
────────────────────────────────────────────────────────── -␣
,→───────────────────────────────────────────────────────────────── +␣
,→────────────────────────────────────────────────────────────────────────────────────────
E⋅I ␣
,→ E⋅I ␣
,→ E⋅I
Example 9
There is a cantilever beam of length 4 meters. For first 2 meters its moment of inertia is
1.5*I and I for the other end. A pointload of magnitude 20 N is applied from the top at its
free end.
---> x
\\\\| |
\\\\| 20 N v y
\\\\|________________ |
\\\\| |_______________V
\\\\| 1.5*I _______I_______|
\\\\|________________|
\\\\| :
\\\\|----------------|---------------|
2.0 m 2.0 m
,→────────────────────────────────────────────────────────────
⎝ E E⋅I ⎠ ␣
,→E⋅I E⋅I
Example 10
A combined beam, with constant flexural rigidity E*I, is formed by joining a Beam of length
2*l to the right of another Beam of length l. The whole beam is fixed at both of its both end.
A point load of magnitude P is also applied from the top at a distance of 2*l from starting
point.
---> x
|
\\\\| P v y |\\\\
\\\\| | |\\\\
\\\\| V |\\\\
\\\\|____________ ________________________|\\\\
\\\\|____________O________________________|\\\\
\\\\| : : |\\\\
\\\\| : : |\\\\
\\\\|------------|------------|-----------|\\\\
l l l
>>> b.load
-2 -2 -1 -1
5⋅P⋅l⋅<x> 4⋅P⋅l⋅<-3⋅l + x> 5⋅P⋅<x> -1 13⋅P⋅<-3⋅l + x>
─────────── - ────────────────── - ───────── + P⋅<-2⋅l + x> - ─────────────────
18 9 18 18
>>> b.shear_force()
-1 -1 0 0
5⋅P⋅l⋅<x> 4⋅P⋅l⋅<-3⋅l + x> 5⋅P⋅<x> 0 13⋅P⋅<-3⋅l + x>
─────────── - ────────────────── - ──────── + P⋅<-2⋅l + x> - ────────────────
18 9 18 18
>>> b.bending_moment()
0 0 1 1
5⋅P⋅l⋅<x> 4⋅P⋅l⋅<-3⋅l + x> 5⋅P⋅<x> 1 13⋅P⋅<-3⋅l + x>
────────── - ───────────────── - ──────── + P⋅<-2⋅l + x> - ────────────────
18 9 18 18
(continues on next page)
>>> b.slope()
⎛ 1 2 2⎞ ⎛ 1 2 ␣
,→2⎞ ⎛ 2 1 2 2 ␣
,→ 2⎞
⎜5⋅P⋅l⋅<x> 5⋅P⋅<x> 5⋅P⋅<-l + x> ⎟ 0 ⎜5⋅P⋅l⋅<x> 5⋅P⋅<x> 5⋅P⋅<-l + x>␣
,→⎟ 0 ⎜P⋅l 4⋅P⋅l⋅<-3⋅l + x> 5⋅P⋅<-l + x> P⋅<-2⋅l + x> 13⋅P⋅<-
,→3⋅l + x> ⎟ 0
⎜────────── - ──────── + ─────────────⎟⋅<x> ⎜────────── - ──────── +␣
,→─────────────⎟⋅<-l + x> ⎜──── - ───────────────── - ───────────── +␣
,→───────────── - ────────────────⎟⋅<-l + x>
⎝ 18 36 36 ⎠ ⎝ 18 36 36 ␣
,→⎠ ⎝ 18 9 36 2 36␣
,→ ⎠
──────────────────────────────────────────── -␣
,→───────────────────────────────────────────────── +␣
,→───────────────────────────────────────────────────────────────────────────────────────
E⋅I E⋅I ␣
,→ E⋅I
>>> b.deflection()
⎛ 2 3 3⎞ ⎛ 2 3 ␣
,→3⎞ ⎛ 3 2 2 3 ␣
,→ 3 3⎞
⎜5⋅P⋅l⋅<x> 5⋅P⋅<x> 5⋅P⋅<-l + x> ⎟ 0 ⎜5⋅P⋅l⋅<x> 5⋅P⋅<x> 5⋅P⋅<-l + x>␣
,→⎟ 0 ⎜5⋅P⋅l P⋅l ⋅(-l + x) 2⋅P⋅l⋅<-3⋅l + x> 5⋅P⋅<-l + x> P⋅<-2⋅l␣
,→+ x> 13⋅P⋅<-3⋅l + x> ⎟ 0
⎜────────── - ──────── + ─────────────⎟⋅<x> ⎜────────── - ──────── +␣
,→─────────────⎟⋅<-l + x> ⎜────── + ───────────── - ───────────────── -␣
,→───────────── + ───────────── - ────────────────⎟⋅<-l + x>
,→───────────────────────────────────────────────────────────────────────────────────────────────────
E⋅I E⋅I ␣
,→ E⋅I
5.41.1 Introduction
The category theory module for SymPy will allow manipulating diagrams within a single cat-
egory, including drawing them in TikZ and deciding whether they are commutative or not.
The general reference work this module tries to follow is
[JoyOfCats] J. Adamek, H. Herrlich. G. E. Strecker: Abstract and Concrete
Categories. The Joy of Cats.
The latest version of this book should be available for free download from
katmat.math.uni-bremen.de/acc/acc.pdf
This section lists the classes which implement some of the basic notions in category theory:
objects, morphisms, categories, and diagrams.
class sympy.categories.Object
The base class for any kind of object in an abstract category.
While technically any instance of Basic will do, this class is the recommended way to
create abstract objects in abstract categories.
class sympy.categories.Morphism
The base class for any morphism in an abstract category.
In abstract categories, a morphism is an arrow between two category objects. The object
where the arrow starts is called the domain, while the object where the arrow ends is
called the codomain.
Two morphisms between the same pair of objects are considered to be the same mor-
phisms. To distinguish between morphisms between the same objects use NamedMor-
phism (page 1816).
It is prohibited to instantiate this class. Use one of the derived classes instead.
See also:
IdentityMorphism (page 1818), NamedMorphism (page 1816), CompositeMorphism
(page 1817)
codomain
Returns the codomain of the morphism.
Examples
compose(other)
Composes self with the supplied morphism.
The order of elements in the composition is the usual order, i.e., to construct g ◦ f
use g.compose(f).
Examples
domain
Returns the domain of the morphism.
Examples
class sympy.categories.NamedMorphism
Represents a morphism which has a name.
Names are used to distinguish between morphisms which have the same domain and
codomain: two named morphisms are equal if they have the same domains, codomains,
and names.
See also:
Morphism (page 1815)
Examples
name
Returns the name of the morphism.
Examples
class sympy.categories.CompositeMorphism
Represents a morphism which is a composition of other morphisms.
Two composite morphisms are equal if the morphisms they were obtained from (compo-
nents) are the same and were listed in the same order.
The arguments to the constructor for this class should be listed in diagram order: to
obtain the composition g ◦ f from the instances of Morphism (page 1815) g and f use
CompositeMorphism(f, g).
Examples
codomain
Returns the codomain of this composite morphism.
The codomain of the composite morphism is the codomain of its last component.
Examples
components
Returns the components of this composite morphism.
Examples
domain
Returns the domain of this composite morphism.
The domain of the composite morphism is the domain of its first component.
Examples
flatten(new_name)
Forgets the composite structure of this morphism.
If new_name is not empty, returns a NamedMorphism (page 1816) with the supplied
name, otherwise returns a Morphism (page 1815). In both cases the domain of the
new morphism is the domain of this composite morphism and the codomain of the
new morphism is the codomain of this composite morphism.
Examples
class sympy.categories.IdentityMorphism
Represents an identity morphism.
An identity morphism is a morphism with equal domain and codomain, which acts as an
identity with respect to composition.
See also:
Morphism (page 1815)
Examples
class sympy.categories.Category
An (abstract) category.
A category [JoyOfCats] is a quadruple K = (O, hom, id, ◦) consisting of
• a (set-theoretical) class O, whose members are called K-objects,
• for each pair (A, B) of K-objects, a set hom(A, B) whose members are called K-
morphisms from A to B,
• for a each K-object A, a morphism id : A → A, called the K-identity of A,
• a composition law ◦ associating with every K-morphisms f : A → B and g : B → C a
K-morphism g ◦ f : A → C, called the composite of f and g.
Composition is associative, K-identities are identities with respect to composition, and
the sets hom(A, B) are pairwise disjoint.
This class knows nothing about its objects and morphisms. Concrete cases of (abstract)
categories should be implemented as classes derived from this one.
Certain instances of Diagram (page 1820) can be asserted to be commutative in a Cate-
gory (page 1819) by supplying the argument commutative_diagrams in the constructor.
See also:
Diagram (page 1820)
Examples
commutative_diagrams
Returns the FiniteSet of diagrams which are known to be commutative in this cat-
egory.
name
Returns the name of this category.
Examples
objects
Returns the class of objects of this category.
Examples
class sympy.categories.Diagram
Represents a diagram in a certain category.
Informally, a diagram is a collection of objects of a category and certain morphisms
between them. A diagram is still a monoid with respect to morphism composition; i.e.,
identity morphisms, as well as all composites of morphisms included in the diagram
belong to the diagram. For a more formal approach to this notion see [Pare1970].
The components of composite morphisms are also added to the diagram. No properties
are assigned to such morphisms by default.
A commutative diagram is often accompanied by a statement of the following kind: “if
such morphisms with such properties exist, then such morphisms which such proper-
ties exist and the diagram is commutative”. To represent this, an instance of Diagram
(page 1820) includes a collection of morphisms which are the premises and another col-
lection of conclusions. premises and conclusions associate morphisms belonging to
the corresponding categories with the FiniteSet’s of their properties.
The set of properties of a composite morphism is the intersection of the sets of properties
of its components. The domain and codomain of a conclusion morphism should be among
the domains and codomains of the morphisms listed as the premises of a diagram.
No checks are carried out of whether the supplied object and morphisms do belong to
one and the same category.
References
Examples
conclusions
Returns the conclusions of this diagram.
Examples
hom(A, B)
Returns a 2-tuple of sets of morphisms between objects A and B: one set of mor-
phisms listed as premises, and the other set of morphisms listed as conclusions.
See also:
Object (page 1815), Morphism (page 1815)
Examples
is_subdiagram(diagram)
Checks whether diagram is a subdiagram of self. Diagram D′ is a subdiagram of D
if all premises (conclusions) of D′ are contained in the premises (conclusions) of D.
The morphisms contained both in D′ and D should have the same properties for D′
to be a subdiagram of D.
Examples
objects
Returns the FiniteSet of objects that appear in this diagram.
Examples
premises
Returns the premises of this diagram.
Examples
subdiagram_from_objects(objects)
If objects is a subset of the objects of self, returns a diagram which has as premises
all those premises of self which have a domains and codomains in objects, likewise
for conclusions. Properties are preserved.
Examples
This section lists the classes which allow automatic drawing of diagrams.
class sympy.categories.diagram_drawing.DiagramGrid(diagram, groups=None,
**hints)
Constructs and holds the fitting of the diagram into a grid.
The mission of this class is to analyse the structure of the supplied diagram and to place
its objects on a grid such that, when the objects and the morphisms are actually drawn,
the diagram would be “readable”, in the sense that there will not be many intersections of
moprhisms. This class does not perform any actual drawing. It does strive nevertheless
to offer sufficient metadata to draw a diagram.
Consider the following simple diagram.
Sometimes one sees the diagram as consisting of logical groups. One can advise Dia-
gramGrid as to such groups by employing the groups keyword argument.
Consider the following diagram:
>>> D = Object(”D”)
>>> f = NamedMorphism(A, B, ”f”)
>>> g = NamedMorphism(B, C, ”g”)
>>> h = NamedMorphism(D, A, ”h”)
>>> k = NamedMorphism(D, B, ”k”)
>>> diagram = Diagram([f, g, h, k])
Now, we can group the objects A and D to have them near one another:
DiagramGrid (page 1823) does not automatically guess which layout would suit the sup-
plied diagram better. Consider, for example, the following linear diagram:
>>> E = Object(”E”)
>>> f = NamedMorphism(A, B, ”f”)
>>> g = NamedMorphism(B, C, ”g”)
>>> h = NamedMorphism(C, D, ”h”)
>>> i = NamedMorphism(D, E, ”i”)
>>> diagram = Diagram([f, g, h, i])
When laid out with the generic layout, it does not get to look linear:
One may sometimes need to transpose the resulting layout. While this can always be
done by hand, DiagramGrid (page 1823) provides a hint for that purpose:
Separate hints can also be provided for each group. For an example, refer to tests/
test_drawing.py, and see the different ways in which the five lemma [FiveLemma] can
be laid out.
See also:
Diagram
References
[FiveLemma] http://en.wikipedia.org/wiki/Five_lemma
height
Returns the number of rows in this diagram layout.
Examples
morphisms
Returns those morphisms (and their properties) which are sufficiently meaningful
to be drawn.
Examples
width
Returns the number of columns in this diagram layout.
Examples
curving should be one of ”^”, ”_” to specify in which direction the arrow is going to
curve. curving_amount is a number describing how many unit’s the morphism is going
to curve:
looping_start and looping_end are currently only used for loop morphisms, those
which have the same domain and codomain. These two attributes should store a valid
Xy-pic direction and specify, correspondingly, the direction the arrow gets out into and
the direction the arrow gets back from:
label_displacement controls how far the arrow label is from the ends of the arrow. For
example, to position the arrow label near the arrow head, use “>”:
Finally, arrow_style is used to specify the arrow style. To get a dashed arrow, for ex-
ample, use “{–>}” as arrow style:
See also:
XypicDiagramDrawer (page 1828)
Notes
References
[Xypic] http://xy-pic.sourceforge.net/
class sympy.categories.diagram_drawing.XypicDiagramDrawer
Given a Diagram and the corresponding DiagramGrid (page 1823), produces the Xy-pic
representation of the diagram.
The most important method in this class is draw. Consider the following triangle dia-
gram:
To draw this diagram, its objects need to be laid out with a DiagramGrid (page 1823):
In some diagrams some morphisms are drawn as curved arrows. Consider the following
diagram:
>>> D = Object(”D”)
>>> E = Object(”E”)
>>> h = NamedMorphism(D, A, ”h”)
>>> k = NamedMorphism(D, B, ”k”)
>>> diagram = Diagram([f, g, h, k])
>>> grid = DiagramGrid(diagram)
>>> drawer = XypicDiagramDrawer()
>>> print(drawer.draw(diagram, grid))
\xymatrix{
A \ar[r]_{f} & B \ar[d]^{g} & D \ar[l]^{k} \ar@/_3mm/[ll]_{h} \\
(continues on next page)
To control how far the morphisms are curved by default, one can use the unit and de-
fault_curving_amount attributes:
In some diagrams, there are multiple curved morphisms between the same two objects.
To control by how much the curving changes between two such successive morphisms,
use default_curving_step:
>>> drawer.default_curving_step = 1
>>> h1 = NamedMorphism(A, D, ”h1”)
>>> diagram = Diagram([f, g, h, k, h1])
>>> grid = DiagramGrid(diagram)
>>> print(drawer.draw(diagram, grid))
\xymatrix{
A \ar[r]_{f} \ar@/^1cm/[rr]^{h_{1}} & B \ar[d]^{g} & D \ar[l]^{k} \ar@/_2cm/[ll]_
,→{h} \\
& C &
}
To draw this diagram, its objects need to be laid out with a DiagramGrid (page 1823):
The argument masked can be used to skip morphisms in the presentation of the
diagram:
Finally, the diagram_format argument can be used to specify the format string of
the diagram. For example, to increase the spacing by 1 cm, proceeding as follows:
sympy.categories.diagram_drawing.xypic_draw_diagram(diagram, masked=None,
diagram_format=”,
groups=None, **hints)
Provides a shortcut combining DiagramGrid (page 1823) and XypicDiagramDrawer
(page 1828). Returns an Xy-pic presentation of diagram. The argument masked is a
list of morphisms which will be not be drawn. The argument diagram_format is the
format string inserted after “xymatrix”. groups should be a set of logical groups. The
hints will be passed directly to the constructor of DiagramGrid (page 1823).
For more information about the arguments, see the docstrings of DiagramGrid
(page 1823) and XypicDiagramDrawer.draw.
See also:
XypicDiagramDrawer (page 1828), DiagramGrid (page 1823)
Examples
sympy.categories.diagram_drawing.preview_diagram(diagram, masked=None,
diagram_format=”,
groups=None, output=’png’,
viewer=None, euler=True,
**hints)
Combines the functionality of xypic_draw_diagram and sympy.printing.preview.
The arguments masked, diagram_format, groups, and hints are passed to
xypic_draw_diagram, while output, viewer, and ``euler are passed to preview.
See also:
xypic_diagram_drawer
Examples
5.42.1 Introduction
class sympy.diffgeom.Manifold
Object representing a mathematical manifold.
The only role that this object plays is to keep a list of all patches defined on the manifold.
It does not provide any means to study the topological characteristics of the manifold
that it represents.
class sympy.diffgeom.Patch
Object representing a patch on a manifold.
On a manifold one can have many patches that do not always include the whole manifold.
On these patches coordinate charts can be defined that permit the parameterization of
any point on the patch in terms of a tuple of real numbers (the coordinates).
This object serves as a container/parent for all coordinate system charts that can be
defined on the patch it represents.
Examples
class sympy.diffgeom.CoordSystem
Contains all coordinate transformation logic.
Examples
Define a Manifold and a Patch, and then define two coord systems on that patch:
Define a basis scalar field (i.e. a coordinate function), that takes a point and returns its
coordinates. It is an instance of BaseScalarField.
>>> rect.coord_function(0)(p)
-sqrt(2)/2
>>> rect.coord_function(1)(p)
sqrt(2)/2
Define a basis vector field (i.e. a unit vector field along the coordinate line). Vectors are
also differential operators on scalar fields. It is an instance of BaseVectorField.
>>> dx = rect.base_oneform(0)
>>> dx(v_x)
1
If you provide a list of names the fields will print nicely: - without provided names:
>>> x, v_x, dx
(rect_0, e_rect_0, drect_0)
base_oneform(coord_index)
Return a basis 1-form field.
The basis one-form field for this coordinate system. It is also an operator on vector
fields.
See the docstring of CoordSystem for examples.
base_oneforms()
Returns a list of all base oneforms.
For more details see the base_oneform method of this class.
base_vector(coord_index)
Return a basis vector field.
The basis vector field for this coordinate system. It is also an operator on scalar
fields.
See the docstring of CoordSystem for examples.
base_vectors()
Returns a list of all base vectors.
For more details see the base_vector method of this class.
Examples
coords(to_sys=None)
Coordinates of the point in a given coordinate system.
If to_sys is None it returns the coordinates in the system in which the point was
defined.
class sympy.diffgeom.BaseScalarField
Base Scalar Field over a Manifold for a given Coordinate System.
A scalar field takes a point as an argument and returns a scalar.
A base scalar field of a coordinate system takes a point and returns one of the coordinates
of that point in the coordinate system in question.
To define a scalar field you need to choose the coordinate system and the index of the
coordinate.
The use of the scalar field after its definition is independent of the coordinate system in
which it was defined, however due to limitations in the simplification routines you may
arrive at more complicated expression if you use unappropriate coordinate systems.
You can build complicated scalar fields by just building up SymPy expressions containing
BaseScalarField instances.
Examples
Examples of fields:
>>> fx = BaseScalarField(rect, 0)
>>> fy = BaseScalarField(rect, 1)
>>> (fx**2+fy**2).rcall(point)
r0**2
>>> g = Function('g')
>>> ftheta = BaseScalarField(polar, 1)
>>> fg = g(ftheta-pi)
>>> fg.rcall(point)
g(-pi)
class sympy.diffgeom.BaseVectorField
Vector Field over a Manifold.
A vector field is an operator taking a scalar field and returning a directional derivative
(which is also a scalar field).
A base vector field is the same type of operator, however the derivation is specifically
done with respect to a chosen coordinate.
To define a base vector field you need to choose the coordinate system and the index of
the coordinate.
The use of the vector field after its definition is independent of the coordinate system in
which it was defined, however due to limitations in the simplification routines you may
arrive at more complicated expression if you use unappropriate coordinate systems.
Examples
>>> g = Function('g')
>>> s_field = g(R2.x, R2.y)
>>> s_field.rcall(point_r)
g(x0, y0)
>>> s_field.rcall(point_p)
g(r0*cos(theta0), r0*sin(theta0))
Vector field:
>>> v = BaseVectorField(R2_r, 1)
>>> pprint(v(s_field))
/ d \|
|-----(g(x, xi_2))||
\dxi_2 /|xi_2=y
>>> pprint(v(s_field).rcall(point_r).doit())
d
---(g(x0, y0))
dy0
>>> pprint(v(s_field).rcall(point_p).doit())
/ d \|
|-----(g(r0*cos(theta0), xi_2))||
\dxi_2 /|xi_2=r0*sin(theta0)
Examples
Vector fields:
>>> c_xr
Commutator(e_x, e_r)
>>> simplify(c_xr(R2.y**2).doit())
-2*y**2*cos(theta)/(x**2 + y**2)
class sympy.diffgeom.Differential(form_field)
Return the differential (exterior derivative) of a form field.
The differential of a form (i.e. the exterior derivative) has a complicated definition in the
general case.
The differential df of the 0-form f is defined for any vector field v as df (v) = v(f ).
Examples
Vector fields:
>>> e_x, e_y, = R2.e_x, R2.e_y
Differentials:
>>> dg = Differential(s_field)
>>> dg
d(g(x, y))
>>> pprint(dg(e_x))
/ d \|
|-----(g(xi_1, y))||
\dxi_1 /|xi_1=x
>>> pprint(dg(e_y))
/ d \|
|-----(g(x, xi_2))||
\dxi_2 /|xi_2=y
class sympy.diffgeom.TensorProduct(*args)
Tensor product of forms.
The tensor product permits the creation of multilinear functionals (i.e. higher order
tensors) out of lower order fields (e.g. 1-forms and vector fields). However, the higher
tensors thus created lack the interesting features provided by the other type of product,
the wedge product, namely they are not antisymmetric and hence are not form fields.
Examples
You can make partial contraction for instance when ‘raising an index’. Putting None in
the second argument of rcall means that the respective position in the tensor product
is left as it is.
>>> TP = TensorProduct
>>> metric = TP(R2.dx, R2.dx) + 3*TP(R2.dy, R2.dy)
>>> metric.rcall(R2.e_y, None)
3*dy
>>> metric.rcall(R2.e_y)
3*dy
class sympy.diffgeom.WedgeProduct(*args)
Wedge product of forms.
In the context of integration only completely antisymmetric forms make sense. The
wedge product permits the creation of such forms.
Examples
Examples
The Lie derivative of a tensor field by another tensor field is equal to their commutator:
Examples
Examples
Examples
( ) d
( )
V (f ) γ(t) = dt f γ(t)
where the given vector_field is denoted as V . This holds for any value t for the param-
eter and any scalar field f .
This function returns the differential equation of γ(t) in terms of the coordinate system
coord_sys. The equations and expansions are necessarily done in coordinate-system-
dependent way as there is no other way to represent movement between points on the
manifold (i.e. there is no such thing as a difference of points for a general manifold).
Parameters vector_field
the vector field for which an integral curve will be given
param
the argument of the function γ from R to the curve
start_point
the point which corresponds to γ(0)
coord_sys
the coordinate system in which to give the equations
Returns a tuple of (equations, initial conditions)
See also:
intcurve_series (page 1842)
Examples
sympy.diffgeom.vectors_in_basis(expr, to_sys)
Transform all base vectors in base vectors of a specified coord basis.
While the new base vectors are in the new coordinate system basis, any coefficients are
kept in the old system.
Examples
sympy.diffgeom.twoform_to_matrix(expr)
Return the matrix representing the twoform.
For the twoform w return the matrix M such that M [i, j] = w(ei , ej ), where ei is the i-th
base vector field for the coordinate system in which the expression of w is given.
Examples
sympy.diffgeom.metric_to_Christoffel_1st(expr)
Return the nested list of Christoffel symbols for the given metric.
This returns the Christoffel symbol of first kind that represents the Levi-Civita connection
for the given metric.
Examples
sympy.diffgeom.metric_to_Christoffel_2nd(expr)
Return the nested list of Christoffel symbols for the given metric.
This returns the Christoffel symbol of second kind that represents the Levi-Civita con-
nection for the given metric.
Examples
sympy.diffgeom.metric_to_Riemann_components(expr)
Return the components of the Riemann tensor expressed in a given basis.
Given a metric it calculates the components of the Riemann tensor in the canonical basis
of the coordinate system in which the metric expression is given.
Examples
>>> non_trivial_metric
r**2*TensorProduct(dtheta, dtheta) + exp(2*r)*TensorProduct(dr, dr)
>>> riemann = metric_to_Riemann_components(non_trivial_metric)
>>> riemann[0, :, :, :]
[[[0, 0], [0, 0]], [[0, r*exp(-2*r)], [-r*exp(-2*r), 0]]]
>>> riemann[1, :, :, :]
[[[0, -1/r], [1/r, 0]], [[0, 0], [0, 0]]]
sympy.diffgeom.metric_to_Ricci_components(expr)
Return the components of the Ricci tensor expressed in a given basis.
Given a metric it calculates the components of the Ricci tensor in the canonical basis of
the coordinate system in which the metric expression is given.
Examples
The vector module provides tools for basic vector math and differential calculus with respect
to 3D Cartesian coordinate systems. This documentation provides an overview of all the
features offered, and relevant API.
Introduction
This page gives a brief conceptual overview of the functionality present in sympy.vector.
In vector math, we deal with two kinds of quantities – scalars and vectors.
A scalar is an entity which only has a magnitude – no direction. Examples of scalar quantities
include mass, electric charge, temperature, distance, etc.
A vector, on the other hand, is an entity that is characterized by a magnitude and a direction.
Examples of vector quantities are displacement, velocity, magnetic field, etc.
A scalar can be depicted just by a number, for e.g. a temperature of 300 K. On the other
hand, vectorial quantities like acceleration are usually denoted by a vector. Given a vector
V, the magnitude of the corresponding quantity can be calculated as the magnitude of the
vector itself ∥V∥, while the direction would be specified by a unit vector in the direction of the
original vector, ^ V
V = ∥V∥ .
Coordinate Systems
A coordinate system is an abstract mathematical entity used to define the notion of direc-
tions and locations in n-dimensional spaces. This module deals with 3-dimensional spaces,
with the conventional X, Y and Z axes defined with respect to each coordinate system.
Each coordinate system also has a special reference point called the ‘origin’ defined for it.
This point is used either while referring to locations in 3D space, or while calculating the
coordinates of pre-defined points with respect to the system.
It is a pretty well-known concept that there is no absolute notion of location or orientation in
space. Any given coordinate system defines a unique ‘perspective’ of quantifying positions
and directions. Therefore, even if we assume that all systems deal with the same units of mea-
surement, the expression of vectorial and scalar quantities differs according to the coordinate
system a certain observer deals with.
Consider two points P and Q in space. Assuming units to be common throughtout, the distance
between these points remains the same regardless of the coordinate system in which the
measurements are being made. However, the 3-D coordinates of each of the two points, as
well as the position vector of any of the points with respect to the other, do not. In fact, these
two quantities don’t make sense at all, unless they are being measured keeping in mind a
certain location and orientation of the measurer (essentially the coordinate system).
Therefore, it is quite clear that the orientation and location (of the origin) of a coordinate
system define the way different quantities will be expressed with respect to it. Neither of
the two properties can be measured on an absolute scale, but rather with respect to another
coordinate system. The orientation of one system with respect to another is measured using
the rotation matrix, while the relative position can be quantified via the position vector of one
system’s origin with respect to the other.
Fields
A field is a vector or scalar quantity that can be specified everywhere in space as a function
of position (Note that in general a field may also be dependent on time and other custom
variables). Since we only deal with 3D spaces in this module, a field is defined as a function
of the x, y and z coordinates corresponding to a location in the coordinate system. Here, x, y
and z act as scalar variables defining the position of a general point.
For example, temperature in 3 dimensional space (a temperature field) can be written as
T (x, y, z) – a scalar function of the position. An example of a scalar field in electromagnetism
is the electric potential.
In a similar manner, a vector field can be defined as a vectorial function of the location (x, y, z)
of any point in space.
For instance, every point on the earth may be considered to be in the gravitational force field
of the earth. We may specify the field by the magnitude and the direction of acceleration due
to gravity (i.e. force per unit mass ) ⃗g (x, y, z) at every point in space.
To give an example from electromagnetism, consider an electric potential of form 2x2 y, a
scalar field in 3D space. The corresponding conservative electric field can be computed as
the gradient of the electric potential function, and expressed as 4xy^i + 2x 2^
√ j. The magnitude of
this electric field can in turn be expressed as a scalar field of the form 4x4 + 16x2 y 2 .
Currently, sympy.vector is able to deal with the Cartesian (also called rectangular), spherical
and other curvilinear coordinate systems.
A 3D Cartesian coordinate system can be initialized in sympy.vector as
>>> from sympy.vector import CoordSys3D
>>> N = CoordSys3D('N')
The string parameter to the constructor denotes the name assigned to the system, and will
primarily be used for printing purposes.
Once a coordinate system (in essence, a CoordSys3D instance) has been defined, we can
access the orthonormal unit vectors (i.e. the ^
i, ^
j and ^
k vectors) and coordinate variables/base
scalars (i.e. the x, y and z variables) corresponding to it. We will talk about coordinate
variables in the later sections.
The basis vectors for the X, Y and Z axes can be accessed using the i, j and k properties
respectively.
>>> N.i
N.i
>>> type(N.i)
<class 'sympy.vector.vector.BaseVector'>
As seen above, the basis vectors are all instances of a class called BaseVector.
When a BaseVector is multiplied by a scalar (essentially any SymPy Expr), we get a VectorMul
- the product of a base vector and a scalar.
>>> 3*N.i
3*N.i
>>> type(3*N.i)
<class 'sympy.vector.vector.VectorMul'>
Addition of VectorMul and BaseVectors gives rise to formation of VectorAdd - except for
special cases, ofcourse.
>>> v = 2*N.i + N.j
>>> type(v)
<class 'sympy.vector.vector.VectorAdd'>
>>> v - N.j
2*N.i
>>> type(v - N.j)
<class 'sympy.vector.vector.VectorMul'>
What about a zero vector? It can be accessed using the zero attribute assigned to class
Vector. Since the notion of a zero vector remains the same regardless of the coordinate
system in consideration, we use Vector.zero wherever such a quantity is required.
>>> from sympy.vector import Vector
>>> Vector.zero
0
>>> type(Vector.zero)
<class 'sympy.vector.vector.VectorZero'>
>>> N.i + Vector.zero
N.i
(continues on next page)
All the classes shown above - BaseVector, VectorMul, VectorAdd and VectorZero are sub-
classes of Vector.
You should never have to instantiate objects of any of the subclasses of Vector. Using the
BaseVector instances assigned to a CoordSys3D instance and (if needed) Vector.zero as
building blocks, any sort of vectorial expression can be constructed with the basic mathemat-
ical operators +, -, *. and /.
In addition to the elementary mathematical operations, the vector operations of dot and cross
can also be performed on Vector.
The & and ^ operators have been overloaded for the dot and cross methods respectively.
>>> v1 & v2
-11
>>> v1 ^ v2
(-1)*N.i + (-3)*N.j + (-11)*N.k
However, this is not the recommended way of performing these operations. Using the original
methods makes the code clearer and easier to follow.
In addition to these operations, it is also possible to compute the outer products of Vector
instances in sympy.vector. More on that in a little bit.
The SymPy operations of simplify, trigsimp, diff, and factor work on Vector objects, with
the standard SymPy API.
In essence, the methods work on the measure numbers(The coefficients of the basis vectors)
present in the provided vectorial expression.
Points
As mentioned before, every coordinate system corresponds to a unique origin point. Points,
in general, have been implemented in sympy.vector in the form of the Point class.
To access the origin of system, use the origin property of the CoordSys3D class.
You can instantiate new points in space using the locate_new method of Point. The argu-
ments include the name(string) of the new Point, and its position vector with respect to the
‘parent’ Point.
Like Vector, a user never has to expressly instantiate an object of Point. This is because any
location in space (albeit relative) can be pointed at by using the origin of a CoordSys3D as
the reference, and then using locate_new on it and subsequent Point instances.
The position vector of a Point with respect to another Point can be computed using the
position_wrt method.
>>> P.position_wrt(Q)
b*N.j
>>> Q.position_wrt(N.origin)
a*N.i + c*N.k
Dyadics
Similar to Vector, Dyadic also has subsequent subclasses like BaseDyadic, DyadicMul, Dyad-
icAdd. As with Vector, a zero dyadic can be accessed from Dyadic.zero.
All basic mathematical operations work with Dyadic too.
>>> dyad = N.i.outer(N.k)
>>> dyad*3
3*(N.i|N.k)
>>> dyad - dyad
0
>>> dyad + 2*(N.j|N.i)
(N.i|N.k) + 2*(N.j|N.i)
dot and cross also work among Dyadic instances as well as between a Dyadic and Vector
(and also vice versa) - as per the respective mathematical definitions. As with Vector, & and
^ have been overloaded for dot and cross.
>>> d = N.i.outer(N.j)
>>> d.dot(N.j|N.j)
(N.i|N.j)
>>> d.dot(N.i)
0
>>> d.dot(N.j)
N.i
>>> N.i.dot(d)
N.j
>>> N.k ^ d
(N.j|N.j)
We will now look at how we can initialize new coordinate systems in sympy.vector, trans-
formed in user-defined ways with respect to already-existing systems.
We already know that the origin property of a CoordSys3D corresponds to the Point instance
denoting its origin reference point.
Consider a coordinate system N . Suppose we want to define a new system M , whose origin
is located at 3^
i + 4^
j + 5^
k from N ’s origin. In other words, the coordinates of M ’s origin from
N’s perspective happen to be (3, 4, 5). Moreover, this would also mean that the coordinates of
N ’s origin with respect to M would be (−3, −4, −5).
This can be achieved programmatically as follows -
It is worth noting that M ’s orientation is the same as that of N . This means that the rotation
matrix of :math: N with respect to M , and also vice versa, is equal to the identity matrix of
dimensions 3x3. The locate_new method initializes a CoordSys3D that is only translated in
space, not re-oriented, relative to the ‘parent’ system.
Similar to ‘locating’ new systems, sympy.vector also allows for initialization of new Co-
ordSys3D instances that are oriented in user-defined ways with respect to existing systems.
Suppose you have a coordinate system A.
You want to initialize a new coordinate system B, that is rotated with respect to A’s Z-axis by
an angle θ.
az
bz
B
by
A θ
ay
θ
ax bx
There are two ways to achieve this.
This is the easiest, cleanest, and hence the recommended way of doing it.
You would first have to initialize an AxisOrienter instance for storing the rotation informa-
tion.
orient_new also lets you orient new systems using multiple Orienter instances, provided
in an iterable. The rotations/orientations are applied to the new system in the order the
Orienter instances appear in the iterable.
The sympy.vector API provides the following four Orienter classes for orientation purposes-
1. AxisOrienter
2. BodyOrienter
3. SpaceOrienter
4. QuaternionOrienter
Please refer to the API of the respective classes in the docs of this module to know more.
In each of the above examples, the origin of the new coordinate system coincides with the
origin of the ‘parent’ system.
>>> B.position_wrt(A)
0
To compute the rotation matrix of any coordinate system with respect to another one, use the
rotation_matrix method.
What if you want to initialize a new system that is not only oriented in a pre-defined way, but
also translated with respect to the parent?
Each of the orient_new_<method of orientation> methods, as well as the orient_new
method, support a location keyword argument.
If a Vector is supplied as the value for this kwarg, the new system’s origin is automatically
defined to be located at that position vector with respect to the parent coordinate system.
Thus, the orientation methods also act as methods to support orientation+ location of the
new systems.
The most general way of creating user-defined system is to use transformation parameter
in CoordSys3D. Here we can define any transformation equations. If we are interested in
some typical curvilinear coordinate system different that Cartesian, we can also use some
predefined ones. It could be also possible to translate or rotate system by setting appropriate
transformation equations.
In CoordSys3D is also dedicated method, create_new which works similarly to methods like
locate_new, orient_new_axis etc.
As mentioned earlier, the same vector attains different expressions in different coordinate
systems. In general, the same is true for scalar expressions and dyadic tensors.
sympy.vector supports the expression of vector/scalar quantities in different coordinate sys-
tems using the express function.
For purposes of this section, assume the following initializations-
Apart from Vector instances, express also supports reexpression of scalars (general SymPy
Expr) and Dyadic objects.
express also accepts a second coordinate system for re-expressing Dyadic instances.
Coordinate Variables
The location of a coordinate system’s origin does not affect the re-expression of BaseVector
instances. However, it does affect the way BaseScalar instances are expressed in different
systems.
BaseScalar instances, are coordinate ‘symbols’ meant to denote the variables used in the
definition of vector/scalar fields in sympy.vector.
For example, consider the scalar field TN (x, y, z) = x + y + z defined in system N . Thus, at a
point with coordinates (a, b, c), the value of the field would be a + b + c. Now consider system
R, whose origin is located at (1, 2, 3) with respect to N (no change of orientation). A point with
coordinates (a, b, c) in R has coordinates (a + 1, b + 2, c + 3) in N . Therefore, the expression for
TN in R becomes TR (x, y, z) = x + y + z + 6.
Coordinate variables, if present in a vector/scalar/dyadic expression, can also be re-expressed
in a given coordinate system, by setting the variables keyword argument of express to True.
The above mentioned example, done programmatically, would look like this -
The to_matrix method of Vector and express_coordinates method of Point also return
different results depending on the coordinate system being provided.
Implementation in sympy.vector
Vector math and basic calculus operations with respect to vectors have already been elabo-
rated upon in the earlier section of this module’s documentation.
On the other hand, base scalars (or coordinate variables) are implemented in a special class
called BaseScalar, and are assigned to every coordinate system, one for each axis from X, Y
and Z. These coordinate variables are used to form the expressions of vector or scalar fields
in 3D space. For a system R, the X, Y and Z BaseScalars instances can be accessed using
the R.x, R.y and R.z expressions respectively.
Therefore, to generate the expression for the aforementioned electric potential field 2x2 y, you
would have to do
It is to be noted that BaseScalar instances can be used just like any other SymPy Symbol,
except that they store the information about the coordinate system and axis they correspond
to.
Scalar fields can be treated just as any other SymPy expression, for any math/calculus func-
tionality. Hence, to differentiate the above electric potential with respect to x (i.e. R.x), you
would use the diff method.
It is worth noting that having a BaseScalar in the expression implies that a ‘field’ changes
with position, in 3D space. Technically speaking, a simple Expr with no BaseScalar s is still
a field, though constant.
Like scalar fields, vector fields that vary with position can also be constructed using
BaseScalar s in the measure-number expressions.
The Del, or ‘Nabla’ operator - written as ∇ is commonly known as the vector differential
operator. Depending on its usage in a mathematical expression, it may denote the gradient
of a scalar field, the divergence of a vector field, or the curl of a vector field.
Essentially, ∇ is not technically an ‘operator’, but a convenient mathematical notation to de-
note any one of the aforementioned field operations.
In sympy.vector, ∇ has been implemented as the Del() class. The instance of this class is
independent of coordinate system. Hence, the ∇ operator would be accessible as Del().
Given below is an example of usage of the Del() class.
The above expression can be evaluated using the SymPy doit() routine.
>>> gradient_field.doit()
C.y*C.z*C.i + C.x*C.z*C.j + C.x*C.y*C.k
Usage of the ∇ notation in sympy.vector has been described in greater detail in the subse-
quent subsections.
Curl
Divergence
Divergence is a vector operator that measures the magnitude of a vector field’s source or sink
at a given point, in terms of a signed scalar.
The divergence operator always returns a scalar after operating on a vector.
In the 3D Cartesian system, the divergence of a 3D vector F, denoted by ∇ · F is given by:
∇·F= ∂U
∂x + ∂V
∂y + ∂W
∂z
Gradient
Consider a scalar field f (x, y, z) in 3D space. The gradient of this field is defined as the vector
of the 3 partial derivatives of f with respect to x, y and z in the X, Y and Z axes respectively.
In the 3D Cartesian system, the divergence of a scalar field f , denoted by ∇f is given by -
∂f^ ∂f^ ∂f ^
∇f = ∂x i + ∂y j + ∂z k
Computing the divergence of a vector field in sympy.vector can be accomplished in two ways.
One, by using the Del() class
Directional Derivative
Apart from the above three common applications of ∇, it is also possible to compute the
directional derivative of a field wrt a Vector in sympy.vector.
By definition, the directional derivative of a field F along a vector v at point x represents the
instantaneous rate of change of F moving through x with the velocity v. It is represented
mathematically as: (⃗v · ∇) F(x).
Directional derivatives of vector and scalar fields can be computed in sympy.vector using
the Del() class
>>> gradient(c.r*c.theta*c.z)
c.theta*c.z*c.i + c.z*c.j + c.r*c.theta*c.k
In vector calculus, a conservative field is a field that is the gradient of some scalar field.
Conservative fields have the property that their line integral over any path depends only on
the end-points, and is independent of the path travelled. A conservative vector field is also
said to be ‘irrotational’, since the curl of a conservative field is always zero.
In physics, conservative fields represent forces in physical systems where energy is con-
served.
To check if a vector field is conservative in sympy.vector, the is_conservative function can
be used.
A solenoidal field, on the other hand, is a vector field whose divergence is zero at all points
in space.
To check if a vector field is solenoidal in sympy.vector, the is_solenoidal function can be
used.
We have previously mentioned that every conservative field can be defined as the gradient of
some scalar field. This scalar field is also called the ‘scalar potential field’ corresponding to
the aforementioned conservative field.
The scalar_potential function in sympy.vector calculates the scalar potential field cor-
responding to a given conservative vector field in 3D space - minus the extra constant of
integration, of course.
Example of usage -
This section details the solution of two basic problems in vector math/calculus using the
sympy.vector package.
Quadrilateral problem
The Problem
OABC is any quadrilateral in 3D space. P is the midpoint of OA, Q is the midpoint of AB, R is
the midpoint of BC and S is the midpoint of OC. Prove that PQ is parallel to SR
Solution
The solution to this problem demonstrates the usage of Point, and basic operations on Vec-
tor.
Define a coordinate system
>>> O = Sys.origin
P is the midpoint of OA. Lets locate it with respect to O (you could also define it with respect
to A).
>>> PQ = Q.position_wrt(P)
>>> SR = R.position_wrt(S)
>>> PQ.cross(SR)
0
Since the cross product is a zero vector, the two vectors have to be parallel, thus proving that
PQ || SR.
See
The Problem
Solution
The scalar field f and the measure numbers of the vector field ⃗v are all functions of the
coordinate variables of the coordinate system in general. Hence, define SymPy functions
that way.
v1, v2 and v3 are the X, Y and Z components of the vector field respectively.
Define the vector field as vfield and the scalar field as sfield.
>>> vfield = v1(C.x, C.y, C.z)*C.i + v2(C.x, C.y, C.z)*C.j + v3(C.x, C.y, C.z)*C.k
>>> ffield = f(C.x, C.y, C.z)
Construct the expression for the LHS of the equation using Del().
Now, to prove the product rule, we would just need to equate the expanded and simplified
versions of the lhs and the rhs, so that the SymPy expressions match.
Thus, the general form of the third product rule mentioned above can be proven using sympy.
vector.
CoordSys3D
Examples
Examples
Examples
Using an AxisOrienter
Using a BodyOrienter
Using a SpaceOrienter
Using a QuaternionOrienter
Examples
Examples
A ‘Body’ fixed rotation is described by three angles and three body-fixed rotation
axes. To orient a coordinate system D with respect to N, each sequential rotation is
always about the orthogonal unit vectors fixed to D. For example, a ‘123’ rotation will
specify rotations about N.i, then D.j, then D.k. (Initially, D.i is same as N.i) Therefore,
is same as
Acceptable rotation orders are of length 3, expressed in XYZ or 123, and cannot
have a rotation about about an axis twice in a row.
Examples
The location of the new coordinate system’s origin wrt this system’s
origin. If not specified, the origins are taken to be coincident.
vector_names, variable_names : iterable(optional)
Iterables of 3 strings each, with custom names for base vectors and
base scalars of the new system respectively. Used for simple str print-
ing.
See also:
Examples
is same as
position_wrt(other)
Returns the position vector of the origin of this coordinate system with respect to
another Point/CoordSys3D.
Parameters other : Point/CoordSys3D
If other is a Point, the position of this system’s origin wrt it is re-
turned. If its an instance of CoordSyRect, the position wrt its origin
is returned.
Examples
rotation_matrix(other)
Returns the direction cosine matrix(DCM), also known as the ‘rotation matrix’ of
this coordinate system with respect to another system.
If v_a is a vector defined in system ‘A’ (in matrix format) and v_b is the same vector
defined in system ‘B’, then v_a = A.rotation_matrix(B) * v_b.
Examples
scalar_map(other)
Returns a dictionary which expresses the coordinate variables (base scalars) of this
frame in terms of the variables of otherframe.
Parameters otherframe : CoordSys3D
The other system to map the variables to.
Examples
Vector
class sympy.vector.vector.Vector
Super class for all Vector classes. Ideally, neither this class nor any of its subclasses
should be instantiated by the user.
components
Returns the components of this vector in the form of a Python dictionary mapping
BaseVector instances to the corresponding measure numbers.
Examples
cross(other)
Returns the cross product of this Vector with another Vector or Dyadic instance. The
cross product is a Vector, if ‘other’ is a Vector. If ‘other’ is a Dyadic, this returns a
Dyadic instance.
Parameters other: Vector/Dyadic
The Vector or Dyadic we are crossing with.
Examples
dot(other)
Returns the dot product of this Vector, either with another Vector, or a Dyadic, or
a Del operator. If ‘other’ is a Vector, returns the dot product scalar (Sympy expres-
sion). If ‘other’ is a Dyadic, the dot product is returned as a Vector. If ‘other’ is an
instance of Del, returns the directional derivative operator as a Python function. If
this function is applied to a scalar expression, it returns the directional derivative
of the scalar field wrt this Vector.
Parameters other: Vector/Dyadic/Del
The Vector or Dyadic we are dotting with, or a Del operator .
Examples
magnitude()
Returns the magnitude of this vector.
normalize()
Returns the normalized version of this vector.
outer(other)
Returns the outer product of this vector with another, in the form of a Dyadic in-
stance.
Parameters other : Vector
The Vector with respect to which the outer product is to be computed.
Examples
projection(other, scalar=False)
Returns the vector or scalar projection of the ‘other’ on ‘self’.
Examples
separate()
The constituents of this vector in different coordinate systems, as per its definition.
Returns a dict mapping each CoordSys3D to the corresponding constituent Vector.
Examples
to_matrix(system)
Returns the matrix form of this vector with respect to the specified coordinate sys-
tem.
Parameters system : CoordSys3D
The system wrt which the matrix form is to be computed
Examples
Dyadic
class sympy.vector.dyadic.Dyadic
Super class for all Dyadic-classes.
References
[R600], [R601]
components
Returns the components of this dyadic in the form of a Python dictionary mapping
BaseDyadic instances to the corresponding measure numbers.
cross(other)
Returns the cross product between this Dyadic, and a Vector, as a Vector instance.
Parameters other : Vector
The Vector that we are crossing this Dyadic with
Examples
dot(other)
Returns the dot product(also called inner product) of this Dyadic, with another
Dyadic or Vector. If ‘other’ is a Dyadic, this returns a Dyadic. Else, it returns a
Vector (unless an error is encountered).
Parameters other : Dyadic/Vector
The other Dyadic or Vector to take the inner product with
Examples
to_matrix(system, second_system=None)
Returns the matrix form of the dyadic with respect to one or two coordinate systems.
Parameters system : CoordSys3D
The coordinate system that the rows and columns of the matrix cor-
respond to. If a second system is provided, this only corresponds to
the rows of the matrix.
second_system : CoordSys3D, optional, default=None
The coordinate system that the columns of the matrix correspond to.
Examples
Del
class sympy.vector.deloperator.Del
Represents the vector differential operator, usually represented in mathematical expres-
sions as the ‘nabla’ symbol.
cross(vect, doit=False)
Represents the cross product between this operator and a given vector - equal to
the curl of the vector field.
Parameters vect : Vector
The vector whose curl is to be calculated.
doit : bool
If True, the result is returned after calling .doit() on each component.
Else, the returned expression contains Derivative instances
Examples
dot(vect, doit=False)
Represents the dot product between this operator and a given vector - equal to the
divergence of the vector field.
Parameters vect : Vector
The vector whose divergence is to be calculated.
doit : bool
If True, the result is returned after calling .doit() on each component.
Else, the returned expression contains Derivative instances
Examples
gradient(scalar_field, doit=False)
Returns the gradient of the given scalar field, as a Vector instance.
Parameters scalar_field : SymPy expression
The scalar field to calculate the gradient of.
doit : bool
If True, the result is returned after calling .doit() on each component.
Else, the returned expression contains Derivative instances
Examples
Orienter
class sympy.vector.orienters.Orienter
Super-class for all orienter classes.
rotation_matrix()
The rotation matrix corresponding to this orienter instance.
AxisOrienter
Examples
rotation_matrix(system)
The rotation matrix corresponding to this orienter instance.
Parameters system : CoordSys3D
The coordinate system wrt which the rotation matrix is to be com-
puted
BodyOrienter
Examples
A ‘Body’ fixed rotation is described by three angles and three body-fixed rotation
axes. To orient a coordinate system D with respect to N, each sequential rotation is
always about the orthogonal unit vectors fixed to D. For example, a ‘123’ rotation will
specify rotations about N.i, then D.j, then D.k. (Initially, D.i is same as N.i) Therefore,
is same as
Acceptable rotation orders are of length 3, expressed in XYZ or 123, and cannot
have a rotation about about an axis twice in a row.
>>> body_orienter1 = BodyOrienter(q1, q2, q3, '123')
>>> body_orienter2 = BodyOrienter(q1, q2, 0, 'ZXZ')
>>> body_orienter3 = BodyOrienter(0, 0, 0, 'XYX')
SpaceOrienter
Examples
is same as
>>> from sympy.vector import AxisOrienter
>>> axis_orienter1 = AxisOrienter(q1, N.i)
>>> B = N.orient_new('B', (axis_orienter1, ))
(continues on next page)
QuaternionOrienter
Examples
matrix_to_vector
sympy.vector.matrix_to_vector(matrix, system)
Converts a vector in matrix form to a Vector instance.
It is assumed that the elements of the Matrix represent the measure numbers of the
components of the vector along basis vectors of ‘system’.
Parameters matrix : SymPy Matrix, Dimensions: (3, 1)
The matrix to be converted to a vector
system : CoordSys3D
The coordinate system the vector is to be defined in
Examples
express
Examples
curl
Examples
divergence
Examples
>>> divergence(v1)
R.x*R.y + R.x*R.z + R.y*R.z
>>> v2 = 2*R.y*R.z*R.j
>>> divergence(v2)
2*R.z
gradient
Examples
is_conservative
sympy.vector.is_conservative(field)
Checks if a field is conservative.
Parameters field : Vector
The field to check for conservative property
Examples
is_solenoidal
sympy.vector.is_solenoidal(field)
Checks if a field is solenoidal.
Parameters field : Vector
The field to check for solenoidal property
Examples
scalar_potential
sympy.vector.scalar_potential(field, coord_sys)
Returns the scalar potential function of a field in a given coordinate system (without the
added integration constant).
Parameters field : Vector
The vector field whose scalar potential function is to be calculated
coord_sys : CoordSys3D
The coordinate system to do the calculation in
Examples
scalar_potential_difference
Examples
All contributions are welcome. If you’d like to improve something, look into the sources if
they contain the information you need (if not, please fix them), otherwise the documentation
generation needs to be improved (look in the doc/ directory).
SIX
6.1 Introduction
The purpose of this collection of documents is to provide users of SymPy with topics which
are not strictly tutorial or are longer than tutorials and tests. The documents will hopefully
fill a gap as SymPy matures and users find more ways to show how SymPy can be used in
more advanced topics.
6.2.1 Introduction
( dF ) 1 ( d2 F ) 1 ( d3 F )
F (x0 + h) = F (x0 ) + ∗ h + ∗ h2
+ ∗ h3 + ...
dx x0 2! dx2 x0 3! dx3 x0
In order to simplify the notation, we now define a set of coefficients cn , where
1 ( dn F )
cn := .
n! dxn x0
So now our series has the form:
In the following we will only use a finite grid of values xi with i running from 1, ..., N and the
corresponding values of our function F at these grid points denoted by Fi . So the problem is
how to generate approximate values for the derivatives of F with the constraint that we use
a subset of the finite set of pairs (xi , Fi ) of size N.
1889
SymPy Documentation, Release 1.3
What follows are manipulations using SymPy to formulate approximations for derivatives of
a given order and to assess its accuracy. First, we use SymPy to derive the approximations
by using a rather brute force method frequently covered in introductory treatments. Later
we shall make use of other SymPy functions which get the job done with more efficiency.
If we let x0 = xi , evaluate the series at xi+1 = xi + h and truncate all terms above O(h1 ) we can
solve for the single coefficient c1 and obtain an approximation to the first derivative:
( dF ) Fi+1 − Fi
≈ + O(h)
dx x0 h
where the O(h) refers to the lowest order term in the series in h. This establishes that the
derivative approximation is of first order accuracy. Put another way, if we decide that we can
only use the two pairs (xi , Fi ) and (xi+1 , Fi+1 ) we obtain a “first order accurate” derivative.
In addition to (xi , Fi ) we next use the two points (xi+1 , Fi+1 ) and (xi+2 , Fi+2 ). Then we have two
equations:
1 1
Fi+1 = Fi + c1 ∗ h + ∗ c2 ∗ h2 + ∗ c3 ∗ h3 + ...
2 3!
1 1
Fi+2 = Fi + c1 ∗ (2h) +
∗ c2 ∗ (2h)2 + ∗ c3 ∗ (2h)3 + ...
2 3!
If we again want to find the first derivative (c1 ), we can do that by eliminating the term in-
volving c2 from the two equations. We show how to do it using SymPy.
Now we make a matrix consisting of the coefficients of the c_i in the nth degree polynomial P.
Coefficients of ci evaluated at xi :
Coefficients of ci evaluated at xi + h:
Coefficients of ci evaluated at xi + 2 ∗ h:
>>> M = Matrix([[m11, m12, m13], [m21, m22, m23], [m31, m32, m33]])
>>> X = M.inv() * R
Note that all three coefficients make up the solution. The desired first derivative is coefficient
c1 which is X[1].
>>> print(together(X[1]))
(4*F_{i+1} - F_{i+2} - 3*F_{i})/(2*h)
Now that we have the matrix of coefficients we next form the right-hand-side and solve by
inverting M :
These two examples serve to show how one can directly find second order accurate first
derivatives using SymPy. The first example uses values of x and F at all three points xi , xi+1 ,
and xi+2 whereas the second example only uses values of x at the two points xi−1 and xi+1 and
thus is a bit more efficient.
From these two simple examples a general rule is that if one wants a first derivative to be
accurate to O(hn ) then one needs n+1 function values in the approximating polynomial (here
provided via the function P (x, x0, c, n)).
Now let’s assess the question of the accuracy of the centered difference result to see how we
determine that it is really second order. To do this we take the result for dF /dx and substitute
in the polynomial expansion for a higher order polynomial and see what we get. To this end,
we make a set of eight coefficients d and use them to perform the check:
>>> d = symbols('c:8')
>>> dfdxcheck = (P(x0+h, x0, d, 8) - P(x0-h, x0, d, 8))/(2*h)
>>> print(simplify(dfdxcheck)) # so the appropriate cancellation of terms involving␣
,→`h` happens
Thus we see that indeed the derivative is c1 with the next term in the series of order h2 .
However, it can quickly become rather tedious to generalize the direct method as presented
above when attempting to generate a derivative approximation to high order, such as 6 or 8
although the method certainly works and using the present method is certainly less tedious
than performing the calculations by hand.
As we have seen in the discussion above, the simple centered approximation for the first
derivative only uses two point values of the (xi , Fi ) pairs. This works fine until one encounters
the last point in the domain, say at i = N . Since our centered derivative approximation would
use data at the point (xN +1 , FN +1 ) we see that the derivative formula will not work. So, what to
do? Well, a simple way to handle this is to devise a different formula for this last point which
uses points for which we do have values. This is the so-called backward difference formula.
To obtain it, we can use the same direct approach, except now us the three points (xN , FN ),
(xN −1 , FN −1 ), and (xN −2 , FN −2 ) and center the approximation at (xN , FN ). Here is how it can be
done using SymPy:
Now we make a matrix consisting of the coefficients of the ci in the dth degree polynomial P
coefficients of ci evaluated at xi , xi−1 , and xi+1 :
>>> M = Matrix([[m11, m12, m13], [m21, m22, m23], [m31, m32, m33]])
>>> # matrix of the function values...actually a vector of right hand sides
>>> R = Matrix([[FN], [FNm1], [FNm2]])
>>> X = M.inv() * R
The first derivative is coefficient c1 which is X[1]. Thus the second order accurate approxima-
tion for the first derivative is:
>>> print(”The first derivative centered at the last point on the right is:”)
The first derivative centered at the last point on the right is:
>>> print(together(X[1]))
(-4*F_{N-1} + F_{N-2} + 3*F_{N})/(2*h)
Of course, we can devise a similar formula for the value of the derivative at the left end of the
set of points at (x1 , F1 ) in terms of values at (x2 , F2 ) and (x3 , F3 ).
Also, we note that output of formats appropriate to Fortran, C, etc. may be done in the
examples given above.
Next we show how to perform these and many other discritizations of derivatives, but using
a much more efficient approach originally due to Bengt Fornberg and now incorported into
SymPy.
SEVEN
7.1 Introduction
When debugging comparisons and hashes in SymPy, it is necessary to understand when ex-
actly Python calls each method. Unfortunately, the official Python documentation for this is
not very detailed (see the docs for rich comparison, __cmp__() and __hash__() methods).
We wrote this guide to fill in the missing gaps. After reading it, you should be able to under-
stand which methods do (and do not) get called and the order in which they are called.
7.2 Hashing
Every Python class has a __hash__() method, the default implementation of which is:
def __hash__(self):
return id(self)
You can reimplement it to return a different integer that you compute on your own. hash(x)
just calls x.__hash__(). Python builtin classes usually redefine the __hash__() method. For
example, an int has something like this:
def __hash__(self):
return int(self)
def __hash__(self):
raise TypeError(”list objects are unhashable”)
The general idea about hashes is that if two objects have a different hash, they are not equal,
but if they have the same hash, they might be equal. (This is usually called a “hash collision”
and you need to use the methods described in the next section to determine if the objects
really are equal).
The only requirement from the Python side is that the hash value mustn’t change after it is
returned by the __hash__() method.
Please be aware that hashing is platform-dependent. This means that you can get different
hashes for the same SymPy object on different platforms. This affects for instance sorting of
sympy expressions. You can also get SymPy objects printed in different order.
1895
SymPy Documentation, Release 1.3
When developing, you have to be careful about this, especially when writing tests. It is pos-
sible that your test runs on a 32-bit platform, but not on 64-bit. An example:
If you get this order of solutions, you are probably running 32-bit system. On a 64-bit system
you would get the following:
r = [i.evalf(2) for i in r]
assert r == [1.0 + 1.7*I, 2.0 - 1.0*I, 2.0 + I, 1.0 - 1.7*I]
it will fail on a 64-bit platforms, even if it works for your 32-bit system. You can avoid this by
using the sorted() or set() Python built-in:
r = [i.evalf(2) for i in r]
assert set(r) == set([1.0 + 1.7*I, 2.0 - 1.0*I, 2.0 + I, 1.0 - 1.7*I])
This approach does not work for doctests since they always compare strings that would be
printed after a prompt. In that case you could make your test print results using a combination
of str() and sorted():
or, if you don’t want to show the values as strings, then sympify the results or the sorted list:
The printing of SymPy expressions might be also affected, so be careful with doctests. If you
get the following on a 32-bit system:
Let a, b and c be instances of any one of the Python classes. As can be easily checked by the
Python script (page 1898) at the end of this guide, if you write:
a == b
a.__eq__(b)
b.__eq__(a)
a.__cmp__(b)
b.__cmp__(a)
id(a) == id(b)
a != b
a.__ne__(b)
b.__ne__(a)
a.__cmp__(b)
b.__cmp__(a)
id(a) == id(b)
If you write:
a < b
a.__lt__(b)
b.__gt__(a)
a.__cmp__(b)
b.__cmp__(a)
id(a) < id(b)
If you write:
a <= b
a.__le__(b)
b.__ge__(a)
a.__cmp__(b)
b.__cmp__(a)
id(a) <= id(b)
sorted([a, b, c])
Python calls the same chain of methods as for the b < a and c < b comparisons.
If you write any of the following:
a in {d: 5}
a in set([d, d, d])
set([a, b]) == set([a, b])
If hash(a) != hash(d) then the result of the statement a in {d: 5} is immediately False
(remember how hashes work in general). If hash(a) == hash(d)) Python goes through the
method resolution of the == operator as shown above.
In the method resolution for <, <=, ==, !=, >=, > and sorted([a, b, c]) operators the
__hash__() method is not called, so in these cases it doesn’t matter what it returns. The
__hash__() method is only called for sets and dictionaries.
In the official Python documentation you can read about hashable and non-hashable objects.
In reality, you don’t have to think about it, you just follow the method resolution described
here. E.g. if you try to use lists as dictionary keys, the list’s __hash__() method will be called
and it returns an exception.
In SymPy, every instance of any subclass of Basic is immutable. Technically this means,
that its behavior through all the methods above mustn’t change once the instance is created.
Especially, the hash value mustn’t change (as already stated above) or else objects will get
mixed up in dictionaries and wrong values will be returned for a given key, etc….
The above method resolution can be verified using the following program:
class A(object):
def __hash__(self):
print ”%s.__hash__()” % (self.a)
return self._hash
def show(s):
print ”--- %s ” % s + ”-”*40
eval(s)
a = A(”a”, 1)
b = A(”b”, 2)
c = A(”c”, 3)
d = A(”d”, 1)
show(”a == b”)
show(”a != b”)
show(”a < b”)
show(”a <= b”)
show(”a > b”)
show(”a >= b”)
show(”sorted([a, b, c])”)
show(”{d: 5}”)
show(”a in {d: 5}”)
show(”set([d, d, d])”)
show(”a in set([d, d, d])”)
show(”set([a, b])”)
--- a == b ----------------------------------------
a.__eq__(b)
b.__eq__(a)
a.__cmp__(b)
b.__cmp__(a)
--- a != b ----------------------------------------
a.__ne__(b)
b.__ne__(a)
a.__cmp__(b)
b.__cmp__(a)
--- a < b ----------------------------------------
a.__lt__(b)
b.__gt__(a)
a.__cmp__(b)
b.__cmp__(a)
--- a <= b ----------------------------------------
a.__le__(b)
b.__ge__(a)
a.__cmp__(b)
b.__cmp__(a)
--- a > b ----------------------------------------
a.__gt__(b)
b.__lt__(a)
a.__cmp__(b)
b.__cmp__(a)
--- a >= b ----------------------------------------
a.__ge__(b)
b.__le__(a)
a.__cmp__(b)
b.__cmp__(a)
--- sorted([a, b, c]) ----------------------------------------
b.__lt__(a)
a.__gt__(b)
b.__cmp__(a)
a.__cmp__(b)
c.__lt__(b)
b.__gt__(c)
c.__cmp__(b)
b.__cmp__(c)
--- {d: 5} ----------------------------------------
d.__hash__()
--- a in {d: 5} ----------------------------------------
d.__hash__()
a.__hash__()
d.__eq__(a)
a.__eq__(d)
d.__cmp__(a)
a.__cmp__(d)
--- set([d, d, d]) ----------------------------------------
d.__hash__()
(continues on next page)
EIGHT
WIKI
SymPy has a public wiki located at http://wiki.sympy.org. Users should feel free to contribute
to this wiki anything interesting/useful.
8.1 FAQ
FAQ is one of the most useful wiki pages. It has answers to frequently-asked questions.
1903
SymPy Documentation, Release 1.3
NINE
SYMPY PAPERS
A list of papers which either use or mention SymPy, as well as presentations given about
SymPy at conferences can be seen at SymPy Papers.
1905
SymPy Documentation, Release 1.3
TEN
PLANET SYMPY
1907
SymPy Documentation, Release 1.3
ELEVEN
SYMPY LOGOS
SymPy has a collection of official logos, which can be generated from sympy.svg in your local
copy of SymPy by:
$ cd doc
$ make logo # will be stored in the _build/logo subdirectory
The license of all the logos is the same as SymPy: BSD. See the LICENSE file in the trunk for
more information.
1909
SymPy Documentation, Release 1.3
TWELVE
You can see a list of blogs/news/magazines mentioning SymPy (that we know of) on our wiki
page.
1911
SymPy Documentation, Release 1.3
THIRTEEN
ABOUT
1913
SymPy Documentation, Release 1.3
• Google: SymPy has received generous financial support from Google in various years
through the Google Summer of Code program by providing stipends:
– in 2007 for 5 students (GSoC 2007)
13.3 License
Unless stated otherwise, all files in the SymPy project, SymPy’s webpage (and wiki), all images
and all documentation including this User’s Guide are licensed using the new BSD license:
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ”AS IS”
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE FOR
ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
DAMAGE.
--------------------------------------------------------------------------------
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ”AS IS”
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE FOR
ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
(continues on next page)
--------------------------------------------------------------------------------
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ”AS IS”
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE FOR
ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
DAMAGE.
FOURTEEN
CITING SYMPY
@article{10.7717/peerj-cs.103,
title = {SymPy: symbolic computing in Python},
author = {Meurer, Aaron and Smith, Christopher P. and Paprocki, Mateusz and \v{C}ert\
,→'{i}k, Ond\v{r}ej and Kirpichev, Sergey B. and Rocklin, Matthew and Kumar, AMiT and␣
,→Ivanov, Sergiu and Moore, Jason K. and Singh, Sartaj and Rathnayake, Thilina and␣
,→Vig, Sean and Granger, Brian E. and Muller, Richard P. and Bonazzi, Francesco and␣
,→Gupta, Harsh and Vats, Shivam and Johansson, Fredrik and Pedregosa, Fabian and␣
,→Curry, Matthew J. and Terrel, Andy R. and Rou\v{c}ka, \v{S}t\v{e}p\'{a}n and Saboo,␣
,→Ashutosh and Fernando, Isuru and Kulal, Sumith and Cimrman, Robert and Scopatz,␣
,→Anthony},
year = 2017,
month = jan,
keywords = {Python, Computer algebra system, Symbolics},
abstract = {
SymPy is an open source computer algebra system written in pure Python.␣
,→It is built with a focus on extensibility and ease of use, through both interactive␣
,→popular symbolic library for the scientific Python ecosystem. This paper presents␣
},
volume = 3,
pages = {e103},
journal = {PeerJ Computer Science},
issn = {2376-5992},
url = {https://doi.org/10.7717/peerj-cs.103},
doi = {10.7717/peerj-cs.103}
}
SymPy is BSD licensed, so you are free to use it whatever you like, be it academic, commercial,
creating forks or derivatives, as long as you copy the BSD statement if you redistribute it
(see the LICENSE file for details). That said, although not required by the SymPy license,
if it is convenient for you, please cite SymPy when using it in your work and also consider
1931
SymPy Documentation, Release 1.3
contributing all your changes back, so that we can incorporate it and all of us will benefit in
the end.
[R49] http://en.wikipedia.org/wiki/Negative_number
[R50] http://en.wikipedia.org/wiki/Parity_%28mathematics%29
[R51] http://en.wikipedia.org/wiki/Imaginary_number
[R52] http://en.wikipedia.org/wiki/Composite_number
[R53] http://en.wikipedia.org/wiki/Irrational_number
[R54] http://en.wikipedia.org/wiki/Prime_number
[R55] http://en.wikipedia.org/wiki/Finite
[R56] https://docs.python.org/3/library/math.html#math.isfinite
[R57] http://docs.scipy.org/doc/numpy/reference/generated/numpy.isfinite.html
[R58] http://en.wikipedia.org/wiki/Zero
[R59] http://en.wikipedia.org/wiki/1_%28number%29
[R60] http://en.wikipedia.org/wiki/%E2%88%921_%28number%29
[R61] http://en.wikipedia.org/wiki/One_half
[R62] http://en.wikipedia.org/wiki/NaN
[R63] http://en.wikipedia.org/wiki/Infinity
[R64] http://en.wikipedia.org/wiki/E_%28mathematical_constant%29
[R65] http://en.wikipedia.org/wiki/Imaginary_unit
[R66] http://en.wikipedia.org/wiki/Pi
[R67] http://en.wikipedia.org/wiki/Euler%E2%80%93Mascheroni_constant
[R68] http://en.wikipedia.org/wiki/Catalan%27s_constant
[R69] http://en.wikipedia.org/wiki/Golden_ratio
[R70] https://en.wikipedia.org/wiki/Generalizations_of_Fibonacci_numbers#
Tribonacci_numbers
[R71] http://en.wikipedia.org/wiki/Exponentiation
[R72] http://en.wikipedia.org/wiki/Exponentiation#Zero_to_the_power_of_zero
[R73] http://en.wikipedia.org/wiki/Indeterminate_forms
1933
SymPy Documentation, Release 1.3
[R74] This implementation detail is that Python provides no reliable method to de-
termine that a chained inequality is being built. Chained comparison operators
are evaluated pairwise, using “and” logic (see http://docs.python.org/2/reference/
expressions.html#notin). This is done in an efficient way, so that each object be-
ing compared is only evaluated once and the comparison can short-circuit. For
example, 1 > 2 > 3 is evaluated by Python as (1 > 2) and (2 > 3). The and
operator coerces each side into a bool, returning the object itself when it short-
circuits. The bool of the –Than operators will raise TypeError on purpose, be-
cause SymPy cannot determine the mathematical ordering of symbolic expres-
sions. Thus, if we were to compute x > y > z, with x, y, and z being Symbols,
Python converts the statement (roughly) into these steps:
(1) x > y > z
(2) (x > y) and (y > z)
(3) (GreaterThanObject) and (y > z)
(4) (GreaterThanObject.__nonzero__()) and (y > z)
(5) TypeError
Because of the “and” added at step 2, the statement gets turned into a weak
ternary statement, and the first object’s __nonzero__ method will raise TypeError.
Thus, creating a chained inequality is not possible.
In Python, there is no way to override the and operator, or to control
how it short circuits, so it is impossible to make something like x > y >
z work. There was a PEP to change this, PEP 335, but it was officially
closed in March, 2012.
[R75] This implementation detail is that Python provides no reliable method to de-
termine that a chained inequality is being built. Chained comparison operators
are evaluated pairwise, using “and” logic (see http://docs.python.org/2/reference/
expressions.html#notin). This is done in an efficient way, so that each object be-
ing compared is only evaluated once and the comparison can short-circuit. For
example, 1 > 2 > 3 is evaluated by Python as (1 > 2) and (2 > 3). The and
operator coerces each side into a bool, returning the object itself when it short-
circuits. The bool of the –Than operators will raise TypeError on purpose, be-
cause SymPy cannot determine the mathematical ordering of symbolic expres-
sions. Thus, if we were to compute x > y > z, with x, y, and z being Symbols,
Python converts the statement (roughly) into these steps:
(1) x > y > z
(2) (x > y) and (y > z)
(3) (GreaterThanObject) and (y > z)
(4) (GreaterThanObject.__nonzero__()) and (y > z)
(5) TypeError
Because of the “and” added at step 2, the statement gets turned into a weak
ternary statement, and the first object’s __nonzero__ method will raise TypeError.
Thus, creating a chained inequality is not possible.
In Python, there is no way to override the and operator, or to control
how it short circuits, so it is impossible to make something like x > y >
z work. There was a PEP to change this, PEP 335, but it was officially
closed in March, 2012.
1934 Bibliography
SymPy Documentation, Release 1.3
[R76] This implementation detail is that Python provides no reliable method to de-
termine that a chained inequality is being built. Chained comparison operators
are evaluated pairwise, using “and” logic (see http://docs.python.org/2/reference/
expressions.html#notin). This is done in an efficient way, so that each object be-
ing compared is only evaluated once and the comparison can short-circuit. For
example, 1 > 2 > 3 is evaluated by Python as (1 > 2) and (2 > 3). The and
operator coerces each side into a bool, returning the object itself when it short-
circuits. The bool of the –Than operators will raise TypeError on purpose, be-
cause SymPy cannot determine the mathematical ordering of symbolic expres-
sions. Thus, if we were to compute x > y > z, with x, y, and z being Symbols,
Python converts the statement (roughly) into these steps:
(1) x > y > z
(2) (x > y) and (y > z)
(3) (GreaterThanObject) and (y > z)
(4) (GreaterThanObject.__nonzero__()) and (y > z)
(5) TypeError
Because of the “and” added at step 2, the statement gets turned into a weak
ternary statement, and the first object’s __nonzero__ method will raise TypeError.
Thus, creating a chained inequality is not possible.
In Python, there is no way to override the and operator, or to control
how it short circuits, so it is impossible to make something like x > y >
z work. There was a PEP to change this, PEP 335, but it was officially
closed in March, 2012.
[R77] This implementation detail is that Python provides no reliable method to de-
termine that a chained inequality is being built. Chained comparison operators
are evaluated pairwise, using “and” logic (see http://docs.python.org/2/reference/
expressions.html#notin). This is done in an efficient way, so that each object be-
ing compared is only evaluated once and the comparison can short-circuit. For
example, 1 > 2 > 3 is evaluated by Python as (1 > 2) and (2 > 3). The and
operator coerces each side into a bool, returning the object itself when it short-
circuits. The bool of the –Than operators will raise TypeError on purpose, be-
cause SymPy cannot determine the mathematical ordering of symbolic expres-
sions. Thus, if we were to compute x > y > z, with x, y, and z being Symbols,
Python converts the statement (roughly) into these steps:
(1) x > y > z
(2) (x > y) and (y > z)
(3) (GreaterThanObject) and (y > z)
(4) (GreaterThanObject.__nonzero__()) and (y > z)
(5) TypeError
Because of the “and” added at step 2, the statement gets turned into a weak
ternary statement, and the first object’s __nonzero__ method will raise TypeError.
Thus, creating a chained inequality is not possible.
In Python, there is no way to override the and operator, or to control
how it short circuits, so it is impossible to make something like x > y >
z work. There was a PEP to change this, PEP 335, but it was officially
closed in March, 2012.
Bibliography 1935
SymPy Documentation, Release 1.3
1936 Bibliography
SymPy Documentation, Release 1.3
[R450] http://en.wikipedia.org/wiki/Continued_fraction
[R451] http://en.wikipedia.org/wiki/Periodic_continued_fraction
[R452] K. Rosen. Elementary Number theory and its applications. Addison-Wesley, 3 Sub
edition, pages 379-381, January 1992.
[R453] http://en.wikipedia.org/wiki/M%C3%B6bius_function
[R454] Thomas Koshy “Elementary Number Theory with Applications”
[R455] http://en.wikipedia.org/wiki/Egyptian_fraction
[R456] https://en.wikipedia.org/wiki/Greedy_algorithm_for_Egyptian_fractions
[R457] http://www.ics.uci.edu/~eppstein/numth/egypt/conflict.html
[R458] http://ami.ektf.hu/uploads/papers/finalpdf/AMI_42_from129to134.pdf
[R78] http://en.wikipedia.org/wiki/Vigenere_cipher
[R79] http://web.archive.org/web/20071116100808/ http://filebox.vt.edu/users/
batman/kryptos.html (short URL: https://goo.gl/ijr22d)
[R80] en.wikipedia.org/wiki/Hill_cipher
[R81] Lester S. Hill, Cryptography in an Algebraic Alphabet, The American Mathemat-
ical Monthly Vol.36, June-July 1929, pp.306-312.
[R82] http://en.wikipedia.org/wiki/Morse_code
[R83] http://en.wikipedia.org/wiki/Morse_code
[G83] Solomon Golomb, Shift register sequences, Aegean Park Press, Laguna Hills, Ca,
1967
[M84] James L. Massey, “Shift-Register Synthesis and BCH Decoding.” IEEE Trans. on
Information Theory, vol. 15(1), pp. 122-127, Jan 1969.
[R37] Michael Karr, “Summation in Finite Terms”, Journal of the ACM, Volume 28 Is-
sue 2, April 1981, Pages 305-350 http://dl.acm.org/citation.cfm?doid=322248.
322255
[R38] http://en.wikipedia.org/wiki/Summation#Capital-sigma_notation
[R39] http://en.wikipedia.org/wiki/Empty_sum
[R40] https://en.wikipedia.org/wiki/Absolute_convergence
[R41] https://en.wikipedia.org/wiki/Convergence_tests
[R42] Michael Karr, “Summation in Finite Terms”, Journal of the ACM, Volume 28 Is-
sue 2, April 1981, Pages 305-350 http://dl.acm.org/citation.cfm?doid=322248.
322255
[R43] Michael Karr, “Summation in Finite Terms”, Journal of the ACM, Volume 28 Is-
sue 2, April 1981, Pages 305-350 http://dl.acm.org/citation.cfm?doid=322248.
322255
[R44] http://en.wikipedia.org/wiki/Multiplication#Capital_Pi_notation
[R45] http://en.wikipedia.org/wiki/Empty_product
[R46] https://en.wikipedia.org/wiki/Infinite_product
Bibliography 1937
SymPy Documentation, Release 1.3
[R47] Michael Karr, “Summation in Finite Terms”, Journal of the ACM, Volume 28 Is-
sue 2, April 1981, Pages 305-350 http://dl.acm.org/citation.cfm?doid=322248.
322255
[R48] Marko Petkovsek, Herbert S. Wilf, Doron Zeilberger, A = B, AK Peters, Ltd.,
Wellesley, MA, USA, 1997, pp. 73–100
[R86] https://en.wikipedia.org/wiki/Cooley%E2%80%93Tukey_FFT_algorithm
[R87] http://mathworld.wolfram.com/FastFourierTransform.html
[R88] https://en.wikipedia.org/wiki/Cooley%E2%80%93Tukey_FFT_algorithm
[R89] http://mathworld.wolfram.com/FastFourierTransform.html
[R90] http://www.apfloat.org/ntt.html
[R91] http://mathworld.wolfram.com/NumberTheoreticTransform.html
[R92] https://en.wikipedia.org/wiki/Discrete_Fourier_transform_(general%29
[R93] http://www.apfloat.org/ntt.html
[R94] http://mathworld.wolfram.com/NumberTheoreticTransform.html
[R95] https://en.wikipedia.org/wiki/Discrete_Fourier_transform_(general%29
[R96] https://en.wikipedia.org/wiki/Hadamard_transform
[R97] https://en.wikipedia.org/wiki/Fast_Walsh%E2%80%93Hadamard_transform
[R98] https://en.wikipedia.org/wiki/Hadamard_transform
[R99] https://en.wikipedia.org/wiki/Fast_Walsh%E2%80%93Hadamard_transform
[R100] https://en.wikipedia.org/wiki/M%C3%B6bius_inversion_formula
[R101] https://people.csail.mit.edu/rrw/presentations/subset-conv.pdf
[R102] https://arxiv.org/pdf/1211.0189.pdf
[R103] https://en.wikipedia.org/wiki/M%C3%B6bius_inversion_formula
[R104] https://people.csail.mit.edu/rrw/presentations/subset-conv.pdf
[R105] https://arxiv.org/pdf/1211.0189.pdf
[R106] https://en.wikipedia.org/wiki/Convolution_theorem
[R107] https://en.wikipedia.org/wiki/Discrete_Fourier_transform_(general%29
[R108] https://en.wikipedia.org/wiki/Convolution_theorem
[R109] https://en.wikipedia.org/wiki/Discrete_Fourier_transform_(general%29
[R110] https://www.radioeng.cz/fulltexts/2002/02_03_40_42.pdf
[R111] https://en.wikipedia.org/wiki/Hadamard_transform
[R112] https://people.csail.mit.edu/rrw/presentations/subset-conv.pdf
[R113] https://people.csail.mit.edu/rrw/presentations/subset-conv.pdf
[R114] https://people.csail.mit.edu/rrw/presentations/subset-conv.pdf
[R25] https://docs.scipy.org/doc/numpy/user/basics.types.html
[R26] https://en.wikipedia.org/wiki/Newton%27s_method
[R153] http://en.wikipedia.org/wiki/Complex_conjugation
1938 Bibliography
SymPy Documentation, Release 1.3
[R154] http://en.wikipedia.org/wiki/Trigonometric_functions
[R155] http://dlmf.nist.gov/4.14
[R156] http://functions.wolfram.com/ElementaryFunctions/Sin
[R157] http://mathworld.wolfram.com/TrigonometryAngles.html
[R158] http://en.wikipedia.org/wiki/Trigonometric_functions
[R159] http://dlmf.nist.gov/4.14
[R160] http://functions.wolfram.com/ElementaryFunctions/Cos
[R161] http://en.wikipedia.org/wiki/Trigonometric_functions
[R162] http://dlmf.nist.gov/4.14
[R163] http://functions.wolfram.com/ElementaryFunctions/Tan
[R164] http://en.wikipedia.org/wiki/Trigonometric_functions
[R165] http://dlmf.nist.gov/4.14
[R166] http://functions.wolfram.com/ElementaryFunctions/Cot
[R167] http://en.wikipedia.org/wiki/Trigonometric_functions
[R168] http://dlmf.nist.gov/4.14
[R169] http://functions.wolfram.com/ElementaryFunctions/Sec
[R170] http://en.wikipedia.org/wiki/Trigonometric_functions
[R171] http://dlmf.nist.gov/4.14
[R172] http://functions.wolfram.com/ElementaryFunctions/Csc
[R173] http://en.wikipedia.org/wiki/Sinc_function
[R174] http://en.wikipedia.org/wiki/Inverse_trigonometric_functions
[R175] http://dlmf.nist.gov/4.23
[R176] http://functions.wolfram.com/ElementaryFunctions/ArcSin
[R177] http://en.wikipedia.org/wiki/Inverse_trigonometric_functions
[R178] http://dlmf.nist.gov/4.23
[R179] http://functions.wolfram.com/ElementaryFunctions/ArcCos
[R180] http://en.wikipedia.org/wiki/Inverse_trigonometric_functions
[R181] http://dlmf.nist.gov/4.23
[R182] http://functions.wolfram.com/ElementaryFunctions/ArcTan
[R183] http://en.wikipedia.org/wiki/Inverse_trigonometric_functions
[R184] http://dlmf.nist.gov/4.23
[R185] http://functions.wolfram.com/ElementaryFunctions/ArcCot
[R186] http://en.wikipedia.org/wiki/Inverse_trigonometric_functions
[R187] http://dlmf.nist.gov/4.23
[R188] http://functions.wolfram.com/ElementaryFunctions/ArcSec
[R189] http://reference.wolfram.com/language/ref/ArcSec.html
Bibliography 1939
SymPy Documentation, Release 1.3
[R190] http://en.wikipedia.org/wiki/Inverse_trigonometric_functions
[R191] http://dlmf.nist.gov/4.23
[R192] http://functions.wolfram.com/ElementaryFunctions/ArcCsc
[R193] http://en.wikipedia.org/wiki/Inverse_trigonometric_functions
[R194] http://en.wikipedia.org/wiki/Atan2
[R195] http://functions.wolfram.com/ElementaryFunctions/ArcTan2
[R196] http://en.wikipedia.org/wiki/Hyperbolic_function
[R197] http://dlmf.nist.gov/4.37
[R198] http://functions.wolfram.com/ElementaryFunctions/ArcSech/
[R199] http://en.wikipedia.org/wiki/Hyperbolic_function
[R200] http://dlmf.nist.gov/4.37
[R201] http://functions.wolfram.com/ElementaryFunctions/ArcCsch/
[R202] “Concrete mathematics” by Graham, pp. 87
[R203] http://mathworld.wolfram.com/CeilingFunction.html
[R204] “Concrete mathematics” by Graham, pp. 87
[R205] http://mathworld.wolfram.com/FloorFunction.html
[R206] http://en.wikipedia.org/wiki/Fractional_part
[R207] http://mathworld.wolfram.com/FractionalPart.html
[R208] http://en.wikipedia.org/wiki/Lambert_W_function
[R209] http://en.wikipedia.org/wiki/Directed_complete_partial_order
[R210] http://en.wikipedia.org/wiki/Lattice_%28order%29
[R211] http://en.wikipedia.org/wiki/Square_root
[R212] http://en.wikipedia.org/wiki/Principal_value
[R115] http://en.wikipedia.org/wiki/Bell_number
[R116] http://mathworld.wolfram.com/BellNumber.html
[R117] http://mathworld.wolfram.com/BellPolynomial.html
[R118] http://en.wikipedia.org/wiki/Bernoulli_number
[R119] http://en.wikipedia.org/wiki/Bernoulli_polynomial
[R120] http://mathworld.wolfram.com/BernoulliNumber.html
[R121] http://mathworld.wolfram.com/BernoulliPolynomial.html
[R122] https://www.johndcook.com/blog/binomial_coefficients/
[R123] http://en.wikipedia.org/wiki/Catalan_number
[R124] http://mathworld.wolfram.com/CatalanNumber.html
[R125] http://functions.wolfram.com/GammaBetaErf/CatalanNumber/
[R126] http://geometer.org/mathcircles/catalan.pdf
[R127] http://en.wikipedia.org/wiki/Euler_numbers
1940 Bibliography
SymPy Documentation, Release 1.3
[R128] http://mathworld.wolfram.com/EulerNumber.html
[R129] http://en.wikipedia.org/wiki/Alternating_permutation
[R130] http://mathworld.wolfram.com/AlternatingPermutation.html
[R131] http://dlmf.nist.gov/24.2#ii
[R132] http://en.wikipedia.org/wiki/Subfactorial
[R133] http://mathworld.wolfram.com/Subfactorial.html
[R134] https://en.wikipedia.org/wiki/Double_factorial
[R135] http://mathworld.wolfram.com/FallingFactorial.html
[R136] http://en.wikipedia.org/wiki/Fibonacci_number
[R137] http://mathworld.wolfram.com/FibonacciNumber.html
[R138] https://en.wikipedia.org/wiki/Generalizations_of_Fibonacci_numbers#
Tribonacci_numbers
[R139] http://mathworld.wolfram.com/TribonacciNumber.html
[R140] https://oeis.org/A000073
[R141] http://en.wikipedia.org/wiki/Harmonic_number
[R142] http://functions.wolfram.com/GammaBetaErf/HarmonicNumber/
[R143] http://functions.wolfram.com/GammaBetaErf/HarmonicNumber2/
[R144] http://en.wikipedia.org/wiki/Lucas_number
[R145] http://mathworld.wolfram.com/LucasNumber.html
[R146] https://en.wikipedia.org/wiki/Genocchi_number
[R147] http://mathworld.wolfram.com/GenocchiNumber.html
[R148] https://en.wikipedia.org/wiki/Partition_(number_theory%29
[R149] https://en.wikipedia.org/wiki/Pentagonal_number_theorem
[R150] https://en.wikipedia.org/wiki/Pochhammer_symbol
[R151] http://en.wikipedia.org/wiki/Stirling_numbers_of_the_first_kind
[R152] http://en.wikipedia.org/wiki/Stirling_numbers_of_the_second_kind
[R213] http://mathworld.wolfram.com/DeltaFunction.html
[R214] Regarding to the value at 0, Mathematica defines H(0) = 1, but Maple uses H(0)
= undefined. Different application areas may have specific conventions. For ex-
ample, in control theory, it is common practice to assume H(0) == 0 to match
the Laplace transform of a DiracDelta distribution.
[R215] http://mathworld.wolfram.com/HeavisideStepFunction.html
[R216] http://dlmf.nist.gov/1.16#iv
[R217] https://en.wikipedia.org/wiki/Singularity_function
[R218] http://en.wikipedia.org/wiki/Gamma_function
[R219] http://dlmf.nist.gov/5
[R220] http://mathworld.wolfram.com/GammaFunction.html
Bibliography 1941
SymPy Documentation, Release 1.3
[R221] http://functions.wolfram.com/GammaBetaErf/Gamma/
[R222] http://en.wikipedia.org/wiki/Gamma_function
[R223] http://dlmf.nist.gov/5
[R224] http://mathworld.wolfram.com/LogGammaFunction.html
[R225] http://functions.wolfram.com/GammaBetaErf/LogGamma/
[R226] http://en.wikipedia.org/wiki/Polygamma_function
[R227] http://mathworld.wolfram.com/PolygammaFunction.html
[R228] http://functions.wolfram.com/GammaBetaErf/PolyGamma/
[R229] http://functions.wolfram.com/GammaBetaErf/PolyGamma2/
[R230] http://en.wikipedia.org/wiki/Digamma_function
[R231] http://mathworld.wolfram.com/DigammaFunction.html
[R232] http://functions.wolfram.com/GammaBetaErf/PolyGamma2/
[R233] http://en.wikipedia.org/wiki/Trigamma_function
[R234] http://mathworld.wolfram.com/TrigammaFunction.html
[R235] http://functions.wolfram.com/GammaBetaErf/PolyGamma2/
[R236] http://en.wikipedia.org/wiki/Incomplete_gamma_function#Upper_incomplete_
Gamma_function
[R237] Abramowitz, Milton; Stegun, Irene A., eds. (1965), Chapter 6, Section 5, Hand-
book of Mathematical Functions with Formulas, Graphs, and Mathematical Tables
[R238] http://dlmf.nist.gov/8
[R239] http://functions.wolfram.com/GammaBetaErf/Gamma2/
[R240] http://functions.wolfram.com/GammaBetaErf/Gamma3/
[R241] http://en.wikipedia.org/wiki/Exponential_integral#Relation_with_other_
functions
[R242] http://en.wikipedia.org/wiki/Incomplete_gamma_function#Lower_incomplete_
Gamma_function
[R243] Abramowitz, Milton; Stegun, Irene A., eds. (1965), Chapter 6, Section 5, Hand-
book of Mathematical Functions with Formulas, Graphs, and Mathematical Tables
[R244] http://dlmf.nist.gov/8
[R245] http://functions.wolfram.com/GammaBetaErf/Gamma2/
[R246] http://functions.wolfram.com/GammaBetaErf/Gamma3/
[R247] http://en.wikipedia.org/wiki/Beta_function
[R248] http://mathworld.wolfram.com/BetaFunction.html
[R249] http://dlmf.nist.gov/5.12
[R250] http://en.wikipedia.org/wiki/Error_function
[R251] http://dlmf.nist.gov/7
[R252] http://mathworld.wolfram.com/Erf.html
[R253] http://functions.wolfram.com/GammaBetaErf/Erf
1942 Bibliography
SymPy Documentation, Release 1.3
[R254] http://en.wikipedia.org/wiki/Error_function
[R255] http://dlmf.nist.gov/7
[R256] http://mathworld.wolfram.com/Erfc.html
[R257] http://functions.wolfram.com/GammaBetaErf/Erfc
[R258] http://en.wikipedia.org/wiki/Error_function
[R259] http://mathworld.wolfram.com/Erfi.html
[R260] http://functions.wolfram.com/GammaBetaErf/Erfi
[R261] http://functions.wolfram.com/GammaBetaErf/Erf2/
[R262] http://en.wikipedia.org/wiki/Error_function#Inverse_functions
[R263] http://functions.wolfram.com/GammaBetaErf/InverseErf/
[R264] http://en.wikipedia.org/wiki/Error_function#Inverse_functions
[R265] http://functions.wolfram.com/GammaBetaErf/InverseErfc/
[R266] http://functions.wolfram.com/GammaBetaErf/InverseErf2/
[R267] http://en.wikipedia.org/wiki/Fresnel_integral
[R268] http://dlmf.nist.gov/7
[R269] http://mathworld.wolfram.com/FresnelIntegrals.html
[R270] http://functions.wolfram.com/GammaBetaErf/FresnelS
[R271] The converging factors for the fresnel integrals by John W. Wrench Jr. and Vicki
Alley
[R272] http://en.wikipedia.org/wiki/Fresnel_integral
[R273] http://dlmf.nist.gov/7
[R274] http://mathworld.wolfram.com/FresnelIntegrals.html
[R275] http://functions.wolfram.com/GammaBetaErf/FresnelC
[R276] The converging factors for the fresnel integrals by John W. Wrench Jr. and Vicki
Alley
[R277] http://dlmf.nist.gov/6.6
[R278] http://en.wikipedia.org/wiki/Exponential_integral
[R279] Abramowitz & Stegun, section 5: http://people.math.sfu.ca/~cbm/aands/page_
228.htm
[R280] http://dlmf.nist.gov/8.19
[R281] http://functions.wolfram.com/GammaBetaErf/ExpIntegralE/
[R282] http://en.wikipedia.org/wiki/Exponential_integral
[R283] http://en.wikipedia.org/wiki/Logarithmic_integral
[R284] http://mathworld.wolfram.com/LogarithmicIntegral.html
[R285] http://dlmf.nist.gov/6
[R286] http://mathworld.wolfram.com/SoldnersConstant.html
[R287] http://en.wikipedia.org/wiki/Logarithmic_integral
Bibliography 1943
SymPy Documentation, Release 1.3
[R288] http://mathworld.wolfram.com/LogarithmicIntegral.html
[R289] http://dlmf.nist.gov/6
[R290] http://en.wikipedia.org/wiki/Trigonometric_integral
[R291] http://en.wikipedia.org/wiki/Trigonometric_integral
[R292] http://en.wikipedia.org/wiki/Trigonometric_integral
[R293] http://en.wikipedia.org/wiki/Trigonometric_integral
[R294] Abramowitz, Milton; Stegun, Irene A., eds. (1965), “Chapter 9”, Handbook of
Mathematical Functions with Formulas, Graphs, and Mathematical Tables
[R295] Luke, Y. L. (1969), The Special Functions and Their Approximations, Volume 1
[R296] http://en.wikipedia.org/wiki/Bessel_function
[R297] http://functions.wolfram.com/Bessel-TypeFunctions/BesselJ/
[R298] http://functions.wolfram.com/Bessel-TypeFunctions/BesselY/
[R299] http://functions.wolfram.com/Bessel-TypeFunctions/BesselI/
[R300] http://functions.wolfram.com/Bessel-TypeFunctions/BesselK/
[R301] http://functions.wolfram.com/Bessel-TypeFunctions/HankelH1/
[R302] http://functions.wolfram.com/Bessel-TypeFunctions/HankelH2/
[R303] http://dlmf.nist.gov/10.47
[R304] http://dlmf.nist.gov/10.47
[R305] http://en.wikipedia.org/wiki/Airy_function
[R306] http://dlmf.nist.gov/9
[R307] http://www.encyclopediaofmath.org/index.php/Airy_functions
[R308] http://mathworld.wolfram.com/AiryFunctions.html
[R309] http://en.wikipedia.org/wiki/Airy_function
[R310] http://dlmf.nist.gov/9
[R311] http://www.encyclopediaofmath.org/index.php/Airy_functions
[R312] http://mathworld.wolfram.com/AiryFunctions.html
[R313] http://en.wikipedia.org/wiki/Airy_function
[R314] http://dlmf.nist.gov/9
[R315] http://www.encyclopediaofmath.org/index.php/Airy_functions
[R316] http://mathworld.wolfram.com/AiryFunctions.html
[R317] http://en.wikipedia.org/wiki/Airy_function
[R318] http://dlmf.nist.gov/9
[R319] http://www.encyclopediaofmath.org/index.php/Airy_functions
[R320] http://mathworld.wolfram.com/AiryFunctions.html
[R321] http://en.wikipedia.org/wiki/B-spline
[R322] http://dlmf.nist.gov/25.11
1944 Bibliography
SymPy Documentation, Release 1.3
[R323] http://en.wikipedia.org/wiki/Hurwitz_zeta_function
[R324] http://en.wikipedia.org/wiki/Dirichlet_eta_function
[R325] Bateman, H.; Erdelyi, A. (1953), Higher Transcendental Functions, Vol. I, New
York: McGraw-Hill. Section 1.11.
[R326] http://dlmf.nist.gov/25.14
[R327] http://en.wikipedia.org/wiki/Lerch_transcendent
[R328] http://en.wikipedia.org/wiki/Stieltjes_constants
[R329] Luke, Y. L. (1969), The Special Functions and Their Approximations, Volume 1
[R330] http://en.wikipedia.org/wiki/Generalized_hypergeometric_function
[R331] Luke, Y. L. (1969), The Special Functions and Their Approximations, Volume 1
[R332] http://en.wikipedia.org/wiki/Meijer_G-function
[R333] http://en.wikipedia.org/wiki/Elliptic_integrals
[R334] http://functions.wolfram.com/EllipticIntegrals/EllipticK
[R335] http://en.wikipedia.org/wiki/Elliptic_integrals
[R336] http://functions.wolfram.com/EllipticIntegrals/EllipticF
[R337] http://en.wikipedia.org/wiki/Elliptic_integrals
[R338] http://functions.wolfram.com/EllipticIntegrals/EllipticE2
[R339] http://functions.wolfram.com/EllipticIntegrals/EllipticE
[R340] http://en.wikipedia.org/wiki/Elliptic_integrals
[R341] http://functions.wolfram.com/EllipticIntegrals/EllipticPi3
[R342] http://functions.wolfram.com/EllipticIntegrals/EllipticPi
[R343] http://en.wikipedia.org/wiki/Mathieu_function
[R344] http://dlmf.nist.gov/28
[R345] http://mathworld.wolfram.com/MathieuBase.html
[R346] http://functions.wolfram.com/MathieuandSpheroidalFunctions/MathieuS/
[R347] http://en.wikipedia.org/wiki/Mathieu_function
[R348] http://dlmf.nist.gov/28
[R349] http://mathworld.wolfram.com/MathieuBase.html
[R350] http://functions.wolfram.com/MathieuandSpheroidalFunctions/MathieuC/
[R351] http://en.wikipedia.org/wiki/Mathieu_function
[R352] http://dlmf.nist.gov/28
[R353] http://mathworld.wolfram.com/MathieuBase.html
[R354] http://functions.wolfram.com/MathieuandSpheroidalFunctions/MathieuSPrime/
[R355] http://en.wikipedia.org/wiki/Mathieu_function
[R356] http://dlmf.nist.gov/28
[R357] http://mathworld.wolfram.com/MathieuBase.html
Bibliography 1945
SymPy Documentation, Release 1.3
[R358] http://functions.wolfram.com/MathieuandSpheroidalFunctions/MathieuCPrime/
[R359] http://en.wikipedia.org/wiki/Jacobi_polynomials
[R360] http://mathworld.wolfram.com/JacobiPolynomial.html
[R361] http://functions.wolfram.com/Polynomials/JacobiP/
[R362] http://en.wikipedia.org/wiki/Jacobi_polynomials
[R363] http://mathworld.wolfram.com/JacobiPolynomial.html
[R364] http://functions.wolfram.com/Polynomials/JacobiP/
[R365] http://en.wikipedia.org/wiki/Gegenbauer_polynomials
[R366] http://mathworld.wolfram.com/GegenbauerPolynomial.html
[R367] http://functions.wolfram.com/Polynomials/GegenbauerC3/
[R368] http://en.wikipedia.org/wiki/Chebyshev_polynomial
[R369] http://mathworld.wolfram.com/ChebyshevPolynomialoftheFirstKind.html
[R370] http://mathworld.wolfram.com/ChebyshevPolynomialoftheSecondKind.html
[R371] http://functions.wolfram.com/Polynomials/ChebyshevT/
[R372] http://functions.wolfram.com/Polynomials/ChebyshevU/
[R373] http://en.wikipedia.org/wiki/Chebyshev_polynomial
[R374] http://mathworld.wolfram.com/ChebyshevPolynomialoftheFirstKind.html
[R375] http://mathworld.wolfram.com/ChebyshevPolynomialoftheSecondKind.html
[R376] http://functions.wolfram.com/Polynomials/ChebyshevT/
[R377] http://functions.wolfram.com/Polynomials/ChebyshevU/
[R378] http://en.wikipedia.org/wiki/Legendre_polynomial
[R379] http://mathworld.wolfram.com/LegendrePolynomial.html
[R380] http://functions.wolfram.com/Polynomials/LegendreP/
[R381] http://functions.wolfram.com/Polynomials/LegendreP2/
[R382] http://en.wikipedia.org/wiki/Associated_Legendre_polynomials
[R383] http://mathworld.wolfram.com/LegendrePolynomial.html
[R384] http://functions.wolfram.com/Polynomials/LegendreP/
[R385] http://functions.wolfram.com/Polynomials/LegendreP2/
[R386] http://en.wikipedia.org/wiki/Hermite_polynomial
[R387] http://mathworld.wolfram.com/HermitePolynomial.html
[R388] http://functions.wolfram.com/Polynomials/HermiteH/
[R389] http://en.wikipedia.org/wiki/Laguerre_polynomial
[R390] http://mathworld.wolfram.com/LaguerrePolynomial.html
[R391] http://functions.wolfram.com/Polynomials/LaguerreL/
[R392] http://functions.wolfram.com/Polynomials/LaguerreL3/
[R393] http://en.wikipedia.org/wiki/Laguerre_polynomial#Assoc_laguerre_polynomials
1946 Bibliography
SymPy Documentation, Release 1.3
[R394] http://mathworld.wolfram.com/AssociatedLaguerrePolynomial.html
[R395] http://functions.wolfram.com/Polynomials/LaguerreL/
[R396] http://functions.wolfram.com/Polynomials/LaguerreL3/
[R397] http://en.wikipedia.org/wiki/Spherical_harmonics
[R398] http://mathworld.wolfram.com/SphericalHarmonic.html
[R399] http://functions.wolfram.com/Polynomials/SphericalHarmonicY/
[R400] http://dlmf.nist.gov/14.30
[R401] http://en.wikipedia.org/wiki/Spherical_harmonics
[R402] http://mathworld.wolfram.com/SphericalHarmonic.html
[R403] http://functions.wolfram.com/Polynomials/SphericalHarmonicY/
[R404] http://en.wikipedia.org/wiki/Spherical_harmonics
[R405] http://mathworld.wolfram.com/SphericalHarmonic.html
[R406] http://functions.wolfram.com/Polynomials/SphericalHarmonicY/
[R407] http://en.wikipedia.org/wiki/Kronecker_delta
[WikiPappus] “Pappus’s Hexagon Theorem” Wikipedia, the Free Encyclopedia. Web. 26 Apr.
2013. <http://en.wikipedia.org/wiki/Pappus’s_hexagon_theorem>
[R408] https://hal.inria.fr/inria-00070025/document
[R409] http://www.risc.jku.at/publications/download/risc_2244/DIPLFORM.pdf
[BlogPost] http://nessgrh.wordpress.com/2011/07/07/tricky-branch-cuts/
[R410] http://en.wikipedia.org/wiki/Cauchy_principal_value
[R411] http://mathworld.wolfram.com/CauchyPrincipalValue.html
[R412] http://en.wikipedia.org/wiki/Gaussian_quadrature
[R413] http://people.sc.fsu.edu/~jburkardt/cpp_src/legendre_rule/legendre_rule.html
[R414] http://en.wikipedia.org/wiki/Gauss%E2%80%93Laguerre_quadrature
[R415] http://people.sc.fsu.edu/~jburkardt/cpp_src/laguerre_rule/laguerre_rule.html
[R416] http://en.wikipedia.org/wiki/Gauss-Hermite_Quadrature
[R417] http://people.sc.fsu.edu/~jburkardt/cpp_src/hermite_rule/hermite_rule.html
[R418] http://people.sc.fsu.edu/~jburkardt/cpp_src/gen_hermite_rule/gen_hermite_
rule.html
[R419] http://en.wikipedia.org/wiki/Gauss%E2%80%93Laguerre_quadrature
[R420] http://people.sc.fsu.edu/~jburkardt/cpp_src/gen_laguerre_rule/gen_laguerre_
rule.html
[R421] http://en.wikipedia.org/wiki/Chebyshev%E2%80%93Gauss_quadrature
[R422] http://people.sc.fsu.edu/~jburkardt/cpp_src/chebyshev1_rule/chebyshev1_rule.
html
[R423] http://en.wikipedia.org/wiki/Chebyshev%E2%80%93Gauss_quadrature
[R424] http://people.sc.fsu.edu/~jburkardt/cpp_src/chebyshev2_rule/chebyshev2_rule.
html
Bibliography 1947
SymPy Documentation, Release 1.3
[R425] http://en.wikipedia.org/wiki/Gauss%E2%80%93Jacobi_quadrature
[R426] http://people.sc.fsu.edu/~jburkardt/cpp_src/jacobi_rule/jacobi_rule.html
[R427] http://people.sc.fsu.edu/~jburkardt/cpp_src/gegenbauer_rule/gegenbauer_rule.
html
[R428] en.wikipedia.org/wiki/Quine-McCluskey_algorithm
[R429] en.wikipedia.org/wiki/Quine-McCluskey_algorithm
[R430] http://en.wikipedia.org/wiki/Gaussian_elimination
[R431] https://en.wikipedia.org/wiki/Moore-Penrose_pseudoinverse
[R432] https://en.wikipedia.org/wiki/Moore-Penrose_pseudoinverse#Obtaining_all_
solutions_of_a_linear_system
[Wester1999] Michael J. Wester, A Critique of the Mathematical Abilities of CA Systems, 1999,
http://www.math.unm.edu/~wester/cas/book/Wester.pdf
[Kozen89] D. Kozen, S. Landau, Polynomial decomposition algorithms, Journal of Symbolic
Computation 7 (1989), pp. 445-456
[Liao95] Hsin-Chao Liao, R. Fateman, Evaluation of the heuristic polynomial GCD, Interna-
tional Symposium on Symbolic and Algebraic Computation (ISSAC), ACM Press,
Montreal, Quebec, Canada, 1995, pp. 240–247
[Gathen99] J. von zur Gathen, J. Gerhard, Modern Computer Algebra, First Edition, Cam-
bridge University Press, 1999
[Weisstein09] Eric W. Weisstein, Cyclotomic Polynomial, From MathWorld - A Wolfram Web
Resource, http://mathworld.wolfram.com/CyclotomicPolynomial.html
[Wang78] P. S. Wang, An Improved Multivariate Polynomial Factoring Algorithm, Math. of
Computation 32, 1978, pp. 1215–1231
[Geddes92] K. Geddes, S. R. Czapor, G. Labahn, Algorithms for Computer Algebra, Springer,
1992
[Monagan93] Michael Monagan, In-place Arithmetic for Polynomials over Z_n, Proceedings
of DISCO ‘92, Springer-Verlag LNCS, 721, 1993, pp. 22–34
[Kaltofen98] E. Kaltofen, V. Shoup, Subquadratic-time Factoring of Polynomials over Finite
Fields, Mathematics of Computation, Volume 67, Issue 223, 1998, pp. 1179–1197
[Shoup95] V. Shoup, A New Polynomial Factorization Algorithm and its Implementation,
Journal of Symbolic Computation, Volume 20, Issue 4, 1995, pp. 363–397
[Gathen92] J. von zur Gathen, V. Shoup, Computing Frobenius Maps and Factoring Polyno-
mials, ACM Symposium on Theory of Computing, 1992, pp. 187–224
[Shoup91] V. Shoup, A Fast Deterministic Algorithm for Factoring Polynomials over Finite
Fields of Small Characteristic, In Proceedings of International Symposium on
Symbolic and Algebraic Computation, 1991, pp. 14–21
[Cox97] D. Cox, J. Little, D. O’Shea, Ideals, Varieties and Algorithms, Springer, Second
Edition, 1997
[Ajwa95] I.A. Ajwa, Z. Liu, P.S. Wang, Groebner Bases Algorithm, https://citeseer.ist.psu.
edu/myciteseer/login, 1995
[Bose03] N.K. Bose, B. Buchberger, J.P. Guiver, Multidimensional Systems Theory and Ap-
plications, Springer, 2003
1948 Bibliography
SymPy Documentation, Release 1.3
[Giovini91] A. Giovini, T. Mora, “One sugar cube, please” or Selection strategies in Buch-
berger algorithm, ISSAC ‘91, ACM
[Bronstein93] M. Bronstein, B. Salvy, Full partial fraction decomposition of rational functions,
Proceedings ISSAC ‘93, ACM Press, Kiev, Ukraine, 1993, pp. 157–160
[Buchberger01] B. Buchberger, Groebner Bases: A Short Introduction for Systems Theorists,
In: R. Moreno-Diaz, B. Buchberger, J. L. Freire, Proceedings of EUROCAST‘01,
February, 2001
[Davenport88] J.H. Davenport, Y. Siret, E. Tournier, Computer Algebra Systems and Algo-
rithms for Algebraic Computation, Academic Press, London, 1988, pp. 124–128
[Greuel2008] G.-M. Greuel, Gerhard Pfister, A Singular Introduction to Commutative Algebra,
Springer, 2008
[Atiyah69] M.F. Atiyah, I.G. MacDonald, Introduction to Commutative Algebra, Addison-
Wesley, 1969
[Collins67] G.E. Collins, Subresultants and Reduced Polynomial Remainder Sequences. J.
ACM 14 (1967) 128-142
[BrownTraub71] W.S. Brown, J.F. Traub, On Euclid’s Algorithm and the Theory of Subresul-
tants. J. ACM 18 (1971) 505-514
[Brown78] W.S. Brown, The Subresultant PRS Algorithm. ACM Transaction of Mathematical
Software 4 (1978) 237-249
[Monagan00] M. Monagan and A. Wittkopf, On the Design and Implementation of Brown’s
Algorithm over the Integers and Number Fields, Proceedings of ISSAC 2000, pp.
225-233, ACM, 2000.
[Brown71] W.S. Brown, On Euclid’s Algorithm and the Computation of Polynomial Greatest
Common Divisors, J. ACM 18, 4, pp. 478-504, 1971.
[Hoeij04] M. van Hoeij and M. Monagan, Algorithms for polynomial GCD computation over
algebraic function fields, Proceedings of ISSAC 2004, pp. 297-304, ACM, 2004.
[Wang81] P.S. Wang, A p-adic algorithm for univariate partial fractions, Proceedings of SYM-
SAC 1981, pp. 212-217, ACM, 1981.
[Hoeij02] M. van Hoeij and M. Monagan, A modular GCD algorithm over number fields
presented with multiple extensions, Proceedings of ISSAC 2002, pp. 109-116,
ACM, 2002
[ManWright94] Yiu-Kwong Man and Francis J. Wright, “Fast Polynomial Dispersion Compu-
tation and its Application to Indefinite Summation”, Proceedings of the Interna-
tional Symposium on Symbolic and Algebraic Computation, 1994, Pages 175-180
http://dl.acm.org/citation.cfm?doid=190347.190413
[Koepf98] Wolfram Koepf, “Hypergeometric Summation: An Algorithmic Approach to Sum-
mation and Special Function Identities”, Advanced lectures in mathematics,
Vieweg, 1998
[Abramov71] S. A. Abramov, “On the Summation of Rational Functions”, USSR Computational
Mathematics and Mathematical Physics, Volume 11, Issue 4, 1971, Pages 324-330
[Man93] Yiu-Kwong Man, “On Computing Closed Forms for Indefinite Summations”, Jour-
nal of Symbolic Computation, Volume 16, Issue 4, 1993, Pages 355-376 http:
//www.sciencedirect.com/science/article/pii/S0747717183710539
[Kapur1994] Deepak Kapur, Tushar Saxena, and Lu Yang. “Algebraic and geometric rea-
soning using Dixon resultants”, In Proceedings of the international symposium
Bibliography 1949
SymPy Documentation, Release 1.3
1950 Bibliography
SymPy Documentation, Release 1.3
Bibliography 1951
SymPy Documentation, Release 1.3
[R551] http://mathworld.wolfram.com/Chi-SquaredDistribution.html
[R552] http://en.wikipedia.org/wiki/Dagum_distribution
[R553] http://en.wikipedia.org/wiki/Erlang_distribution
[R554] http://mathworld.wolfram.com/ErlangDistribution.html
[R555] http://en.wikipedia.org/wiki/Exponential_distribution
[R556] http://mathworld.wolfram.com/ExponentialDistribution.html
[R557] http://en.wikipedia.org/wiki/F-distribution
[R558] http://mathworld.wolfram.com/F-Distribution.html
[R559] http://en.wikipedia.org/wiki/Fisher%27s_z-distribution
[R560] http://mathworld.wolfram.com/Fishersz-Distribution.html
[R561] http://en.wikipedia.org/wiki/Fr%C3%A9chet_distribution
[R562] http://en.wikipedia.org/wiki/Gamma_distribution
[R563] http://mathworld.wolfram.com/GammaDistribution.html
[R564] http://en.wikipedia.org/wiki/Inverse-gamma_distribution
[R565] http://en.wikipedia.org/wiki/Kumaraswamy_distribution
[R566] http://en.wikipedia.org/wiki/Laplace_distribution
[R567] http://mathworld.wolfram.com/LaplaceDistribution.html
[R568] http://en.wikipedia.org/wiki/Logistic_distribution
[R569] http://mathworld.wolfram.com/LogisticDistribution.html
[R570] http://en.wikipedia.org/wiki/Lognormal
[R571] http://mathworld.wolfram.com/LogNormalDistribution.html
[R572] http://en.wikipedia.org/wiki/Maxwell_distribution
[R573] http://mathworld.wolfram.com/MaxwellDistribution.html
[R574] http://en.wikipedia.org/wiki/Nakagami_distribution
[R575] http://en.wikipedia.org/wiki/Normal_distribution
[R576] http://mathworld.wolfram.com/NormalDistributionFunction.html
[R577] http://en.wikipedia.org/wiki/Pareto_distribution
[R578] http://mathworld.wolfram.com/ParetoDistribution.html
[R579] http://en.wikipedia.org/wiki/U-quadratic_distribution
[R580] http://en.wikipedia.org/wiki/Raised_cosine_distribution
[R581] http://en.wikipedia.org/wiki/Rayleigh_distribution
[R582] http://mathworld.wolfram.com/RayleighDistribution.html
[R583] http://en.wikipedia.org/wiki/Student_t-distribution
[R584] http://mathworld.wolfram.com/Studentst-Distribution.html
[R585] http://en.wikipedia.org/wiki/Triangular_distribution
[R586] http://mathworld.wolfram.com/TriangularDistribution.html
1952 Bibliography
SymPy Documentation, Release 1.3
[R587] http://en.wikipedia.org/wiki/Uniform_distribution_%28continuous%29
[R588] http://mathworld.wolfram.com/UniformDistribution.html
[R589] http://en.wikipedia.org/wiki/Uniform_sum_distribution
[R590] http://mathworld.wolfram.com/UniformSumDistribution.html
[R591] http://en.wikipedia.org/wiki/Von_Mises_distribution
[R592] http://mathworld.wolfram.com/vonMisesDistribution.html
[R593] http://en.wikipedia.org/wiki/Weibull_distribution
[R594] http://mathworld.wolfram.com/WeibullDistribution.html
[R595] http://en.wikipedia.org/wiki/Wigner_semicircle_distribution
[R596] http://mathworld.wolfram.com/WignersSemicircleLaw.html
[R532] S. A. Abramov, M. Bronstein and M. Petkovsek, On polynomial solutions of linear
operator equations, in: T. Levelt, ed., Proc. ISSAC ‘95, ACM Press, New York,
1995, 290-296.
[R533] M. Petkovsek, Hypergeometric solutions of linear recurrences with polynomial
coefficients, J. Symbolic Computation, 14 (1992), 243-264.
[R534] M. Petkovsek, H. S. Wilf, D. Zeilberger, A = B, 1996.
[R535] S. A. Abramov, Rational solutions of linear difference and q-difference equations
with polynomial coefficients, in: T. Levelt, ed., Proc. ISSAC ‘95, ACM Press, New
York, 1995, 285-289
[R536] M. Petkovsek, Hypergeometric solutions of linear recurrences with polynomial
coefficients, J. Symbolic Computation, 14 (1992), 243-264.
[R537] M. Petkovsek, H. S. Wilf, D. Zeilberger, A = B, 1996.
[R509] Methods to solve Ax^2 + Bxy + Cy^2 + Dx + Ey + F = 0, [online], Available:
http://www.alpertron.com.ar/METHODS.HTM
[R510] Solving the equation ax^2+ bxy + cy^2 + dx + ey + f= 0, [online], Available:
http://www.jpr2718.org/ax2p.pdf
[R511] Solving the generalized Pell equation x**2 - D*y**2 = N, John P. Robertson, July
31, 2004, Pages 16 - 17. [online], Available: http://www.jpr2718.org/pell.pdf
[R512] A. Nitaj, “L’algorithme de Cornacchia”
[R513] Solving the diophantine equation ax**2 + by**2 = m by Cornacchia’s method,
[online], Available: http://www.numbertheory.org/php/cornacchia.html
[R514] Solving the generalized Pell equation x**2 - D*y**2 = N, John P. Robertson, July
31, 2004, Page 15. http://www.jpr2718.org/pell.pdf
[R515] Solving the equation ax^2 + bxy + cy^2 + dx + ey + f = 0, John P.Robertson,
May 8, 2003, Page 7 - 11. http://www.jpr2718.org/ax2p.pdf
[R516] Solving the equation ax^2 + bxy + cy^2 + dx + ey + f = 0, John P.Robertson,
May 8, 2003, Page 7 - 11. http://www.jpr2718.org/ax2p.pdf
[R517] Efficient Solution of Rational Conices, J. E. Cremona and D. Rusin, Mathematics
of Computation, Volume 00, Number 0.
[R518] Representing an integer as a sum of three squares, [online], Available: http://
www.proofwiki.org/wiki/Integer_as_Sum_of_Three_Squares
Bibliography 1953
SymPy Documentation, Release 1.3
1954 Bibliography
SymPy Documentation, Release 1.3
Bibliography 1955
SymPy Documentation, Release 1.3
[R470] http://en.wikipedia.org/wiki/Commutator
[R471] Varshalovich, D A, Quantum Theory of Angular Momentum. 1988.
[R472] Varshalovich, D A, Quantum Theory of Angular Momentum. 1988.
[R473] Varshalovich, D A, Quantum Theory of Angular Momentum. 1988.
[R474] http://en.wikipedia.org/wiki/Commutator
[R475] http://en.wikipedia.org/wiki/Hermitian_adjoint
[R476] http://en.wikipedia.org/wiki/Hermitian_transpose
[R479] http://en.wikipedia.org/wiki/Inner_product
[R477] http://en.wikipedia.org/wiki/Hilbert_space
[R478] http://en.wikipedia.org/wiki/Fock_space
[R480] http://en.wikipedia.org/wiki/Operator_%28physics%29
[R481] http://en.wikipedia.org/wiki/Observable
[R482] http://en.wikipedia.org/wiki/Outer_product
[R483] Varshalovich, D A, Quantum Theory of Angular Momentum. 1988.
[R484] Varshalovich, D A, Quantum Theory of Angular Momentum. 1988.
[R485] http://en.wikipedia.org/wiki/Bra-ket_notation
[R486] http://en.wikipedia.org/wiki/Bra-ket_notation
[R465] http://en.wikipedia.org/wiki/Ray_transfer_matrix_analysis
[R466] http://en.wikipedia.org/wiki/Complex_beam_parameter
[R467] http://en.wikipedia.org/wiki/Gaussian_beam
[R468] http://en.wikipedia.org/wiki/Optical_medium
[R461] http://homes.civil.aau.dk/jc/FemteSemester/Beams3D.pdf
[WikiDel] http://en.wikipedia.org/wiki/Del
[R600] http://en.wikipedia.org/wiki/Dyadic_tensor
[R601] Kane, T., Levinson, D. Dynamics Theory and Applications. 1985 McGraw-Hill
[Dyadics] http://en.wikipedia.org/wiki/Dyadics
[DyadicProducts] http://en.wikipedia.org/wiki/Dyadic_product
[DelOperator] http://en.wikipedia.org/wiki/Del
1956 Bibliography
PYTHON MODULE INDEX
s sympy.combinatorics.testutil, 294
sympy, 801 sympy.combinatorics.util, 288
sympy.assumptions, 1105 sympy.core.add, 163
sympy.assumptions.ask, 1105 sympy.core.assumptions, 98
sympy.assumptions.assume, 1121 sympy.core.basic, 100
sympy.assumptions.handlers, 1125 sympy.core.cache, 100
sympy.assumptions.handlers.calculus, sympy.core.compatibility, 204
1125 sympy.core.containers, 203
sympy.assumptions.handlers.ntheory, sympy.core.core, 114
1127 sympy.core.evalf, 202
sympy.assumptions.handlers.order, 1127 sympy.core.expr, 115
sympy.assumptions.handlers.sets, 1128 sympy.core.exprtools, 206
sympy.assumptions.refine, 1123 sympy.core.function, 181
sympy.calculus, 1501 sympy.core.mod, 166
sympy.calculus.euler, 1501 sympy.core.mul, 160
sympy.calculus.finite_diff, 1505 sympy.core.multidimensional, 181
sympy.calculus.singularities, 1502 sympy.core.numbers, 142
sympy.categories, 1815 sympy.core.power, 157
sympy.categories.diagram_drawing, 1823 sympy.core.relational, 166
sympy.codegen.algorithms, 441 sympy.core.singleton, 114
sympy.codegen.approximations, 411 sympy.core.symbol, 137
sympy.codegen.ast, 412 sympy.core.sympify, 95
sympy.codegen.cfunctions, 429 sympy.crypto.crypto, 343
sympy.codegen.cnodes, 433 sympy.diffgeom, 1832
sympy.codegen.cutils, 443 sympy.discrete, 382
sympy.codegen.cxxnodes, 434 sympy.discrete.convolutions, 388
sympy.codegen.fnodes, 434 sympy.discrete.transforms, 382
sympy.codegen.futils, 443 sympy.functions, 446
sympy.codegen.pyutils, 443 sympy.functions.special.bessel, 534
sympy.codegen.rewriting, 409 sympy.functions.special.beta_functions,
sympy.combinatorics.generators, 237 512
sympy.combinatorics.graycode, 281 sympy.functions.special.elliptic_integrals,
sympy.combinatorics.group_constructs, 557
294 sympy.functions.special.error_functions,
sympy.combinatorics.named_groups, 285 514
sympy.combinatorics.partitions, 208 sympy.functions.special.gamma_functions,
sympy.combinatorics.perm_groups, 238 505
sympy.combinatorics.permutations, 213 sympy.functions.special.mathieu_functions,
sympy.combinatorics.polyhedron, 267 560
sympy.combinatorics.prufer, 270 sympy.functions.special.polynomials,
sympy.combinatorics.subsets, 274 563
sympy.combinatorics.tensor_can, 295 sympy.functions.special.singularity_functions,
1957
SymPy Documentation, Release 1.3
1961
SymPy Documentation, Release 1.3
1962 Index
SymPy Documentation, Release 1.3
Index 1963
SymPy Documentation, Release 1.3
angular_momentum() sympy.physics.quantum.grover),
(sympy.physics.mechanics.rigidbody.RigidBody 1757
method), 1684 apply_load() (sympy.physics.continuum_mechanics.beam.Be
angular_velocity (sympy.physics.optics.waves.TWave method), 1786
attribute), 1783 apply_load() (sympy.physics.continuum_mechanics.beam.Be
AnnihilateBoson (class in method), 1798
sympy.physics.secondquant), 1522 apply_moment_load()
AnnihilateFermion (class in (sympy.physics.continuum_mechanics.beam.Beam3D
sympy.physics.secondquant), 1523 method), 1799
annotated() (in module apply_operator() (sympy.physics.secondquant.AnnihilateBos
sympy.printing.pretty.pretty_symbology), method), 1522
1084 apply_operator() (sympy.physics.secondquant.AnnihilateFer
ANP (class in sympy.polys.polyclasses), 951 method), 1523
AntiCommutator (class in apply_operator() (sympy.physics.secondquant.CreateBoson
sympy.physics.quantum.anticommutator), method), 1523
1708 apply_operator() (sympy.physics.secondquant.CreateFermio
antidivisor_count() (in module method), 1525
sympy.ntheory.factor_), 324 apply_operators() (in module
antidivisors() (in module sympy.physics.secondquant), 1527
sympy.ntheory.factor_), 323 apply_support() (sympy.physics.continuum_mechanics.beam
antihermitian, 100 method), 1787
antihermitian (sympy.assumptions.ask.AssumptionKeys
apply_torque() (sympy.physics.mechanics.body.Body
attribute), 1105 method), 1707
AntiSymmetricTensor (class in applyfunc() (sympy.matrices.sparse.SparseMatrix
sympy.physics.secondquant), 1535 method), 793
aother (sympy.functions.special.hyper.meijerg applyfunc() (sympy.physics.vector.dyadic.Dyadic
attribute), 557 method), 1604
ap (sympy.functions.special.hyper.hyper at- applyfunc() (sympy.physics.vector.vector.Vector
tribute), 554 method), 1599
ap (sympy.functions.special.hyper.meijerg at- approximation() (sympy.core.numbers.NumberSymbol
tribute), 557 method), 148
apart() (in module sympy.polys.partfrac), 902 arbitrary_point() (sympy.geometry.curve.Curve
apart() (sympy.core.expr.Expr method), 115 method), 632
apart_list() (in module sympy.polys.partfrac), arbitrary_point() (sympy.geometry.ellipse.Ellipse
902 method), 636
apoapsis (sympy.geometry.ellipse.Ellipse at- arbitrary_point() (sympy.geometry.line.LinearEntity
tribute), 636 method), 604
apothem (sympy.geometry.polygon.RegularPolygon arbitrary_point() (sympy.geometry.plane.Plane
attribute), 658 method), 674
append() (sympy.plotting.plot.Plot method), arbitrary_point() (sympy.geometry.polygon.Polygon
1089 method), 651
applied_loads (sympy.physics.continuum_mechanics.beam.Beam
Arcsin() (in module sympy.stats), 1225
attribute), 1786 are_collinear() (sympy.geometry.point.Point3D
AppliedPredicate (class in static method), 599
sympy.assumptions.assume), 1121 are_concurrent() (sympy.geometry.line.LinearEntity
apply() (sympy.printing.pretty.stringpict.prettyForm static method), 604
static method), 1086 are_concurrent() (sympy.geometry.plane.Plane
apply() (sympy.simplify.epathtools.EPath static method), 674
method), 1209 are_coplanar() (sympy.geometry.point.Point
apply_finite_diff() (in module class method), 591
sympy.calculus.finite_diff), 1505 are_similar() (in module
apply_force() (sympy.physics.mechanics.body.Body sympy.geometry.util), 588
method), 1707 area (sympy.geometry.ellipse.Ellipse at-
apply_grover() (in module tribute), 637
1964 Index
SymPy Documentation, Release 1.3
Index 1965
SymPy Documentation, Release 1.3
1966 Index
SymPy Documentation, Release 1.3
Index 1967
SymPy Documentation, Release 1.3
1968 Index
SymPy Documentation, Release 1.3
Index 1969
SymPy Documentation, Release 1.3
1970 Index
SymPy Documentation, Release 1.3
Index 1971
SymPy Documentation, Release 1.3
1972 Index
SymPy Documentation, Release 1.3
Index 1973
SymPy Documentation, Release 1.3
1974 Index
SymPy Documentation, Release 1.3
Index 1975
SymPy Documentation, Release 1.3
1976 Index
SymPy Documentation, Release 1.3
Index 1977
SymPy Documentation, Release 1.3
1978 Index
SymPy Documentation, Release 1.3
Index 1979
SymPy Documentation, Release 1.3
1980 Index
SymPy Documentation, Release 1.3
Index 1981
SymPy Documentation, Release 1.3
1982 Index
SymPy Documentation, Release 1.3
Index 1983
SymPy Documentation, Release 1.3
1984 Index
SymPy Documentation, Release 1.3
Index 1985
SymPy Documentation, Release 1.3
1986 Index
SymPy Documentation, Release 1.3
Index 1987
SymPy Documentation, Release 1.3
1988 Index
SymPy Documentation, Release 1.3
Index 1989
SymPy Documentation, Release 1.3
1990 Index
SymPy Documentation, Release 1.3
Index 1991
SymPy Documentation, Release 1.3
1992 Index
SymPy Documentation, Release 1.3
Index 1993
SymPy Documentation, Release 1.3
1994 Index
SymPy Documentation, Release 1.3
Index 1995
SymPy Documentation, Release 1.3
indent_code() (sympy.printing.ccode.C89CodePrinter
init_printing() (in module
method), 1051 sympy.physics.vector.printing), 1614
indent_code() (sympy.printing.fcode.FCodePrinterinject() (sympy.polys.domains.compositedomain.CompositeD
method), 1059 method), 937
indent_code() (sympy.printing.jscode.JavascriptCodePrinter
inject() (sympy.polys.domains.domain.Domain
method), 1062 method), 932
indent_code() (sympy.printing.julia.JuliaCodePrinter
inject() (sympy.polys.domains.simpledomain.SimpleDomain
method), 1064 method), 937
indent_code() (sympy.printing.octave.OctaveCodePrinter
inject() (sympy.polys.polyclasses.DMP
method), 1067 method), 947
indent_code() (sympy.printing.rcode.RCodePrinterinject() (sympy.polys.polytools.Poly method),
method), 1055 865
indent_code() (sympy.printing.rust.RustCodePrinter
InnerProduct (class in
method), 1069 sympy.physics.quantum.innerproduct),
index() (sympy.combinatorics.permutations.Permutation1714
method), 224 InnerProduct (class in
index() (sympy.core.containers.Tuple sympy.physics.secondquant), 1527
method), 203 inradius (sympy.geometry.polygon.RegularPolygon
index() (sympy.physics.secondquant.FixedBosonicBasis attribute), 662
method), 1528 inradius (sympy.geometry.polygon.Triangle
index() (sympy.physics.secondquant.VarBosonicBasis attribute), 669
method), 1528 IntBaseType (class in sympy.codegen.ast),
index() (sympy.polys.rings.PolyRing method), 422
1000 intcurve_diffequ() (in module
Indexed (class in sympy.tensor.indexed), 1409 sympy.diffgeom), 1843
IndexedBase (class in sympy.tensor.indexed), intcurve_series() (in module sympy.diffgeom),
1410 1842
indices (sympy.tensor.indexed.Indexed at- integer, 99
tribute), 1409 Integer (class in sympy.core.numbers), 147
indices_contain_equal_information integer (sympy.assumptions.ask.AssumptionKeys
(sympy.functions.special.tensor_functions.KroneckerDelta
attribute), 1109
attribute), 577 integer_elements (sympy.assumptions.ask.AssumptionKeys
indices_contain_equal_information attribute), 1109
(sympy.physics.secondquant.KroneckerDelta
integer_nthroot() (in module
attribute), 1519 sympy.core.power), 160
inertia() (in module IntegerPartition (class in
sympy.physics.mechanics.functions), sympy.combinatorics.partitions),
1687 210
inertia_of_point_mass() (in module IntegerRing (class in sympy.polys.domains),
sympy.physics.mechanics.functions), 938
1687 Integers (class in sympy.sets.fancysets), 1178
inf (sympy.sets.sets.Set attribute), 1164 Integral (class in sympy.integrals), 714
infinite, 99 Integral.is_commutative (in module
infinite (sympy.assumptions.ask.AssumptionKeys sympy.integrals), 714
attribute), 1109 integrand() (sympy.functions.special.hyper.meijerg
infinite (sympy.series.formal.FormalPowerSeries method), 557
attribute), 1154 integrate() (in module sympy.integrals), 711
infinitesimal (sympy.assumptions.ask.AssumptionKeys
integrate() (sympy.core.expr.Expr method),
attribute), 1109 128
infinitesimals() (in module integrate() (sympy.holonomic.holonomic.HolonomicFunction
sympy.solvers.ode), 1268 method), 682
Infinity (class in sympy.core.numbers), 152 integrate() (sympy.polys.polyclasses.DMP
infinity (sympy.assumptions.ask.AssumptionKeys method), 947
attribute), 1109 integrate() (sympy.polys.polytools.Poly
1996 Index
SymPy Documentation, Release 1.3
Index 1997
SymPy Documentation, Release 1.3
1998 Index
SymPy Documentation, Release 1.3
Index 1999
SymPy Documentation, Release 1.3
is_only_q_creator (sympy.physics.secondquant.CreateFermion
attribute), 947
attribute), 1525 is_quadratic (sympy.polys.polytools.Poly at-
is_open (sympy.sets.sets.Set attribute), 1165 tribute), 869
is_parallel() (sympy.geometry.line.LinearEntity is_radical() (sympy.polys.agca.ideals.Ideal
method), 606 method), 920
is_parallel() (sympy.geometry.plane.Plane is_rational_function() (sympy.core.expr.Expr
method), 676 method), 132
is_perpendicular() (sympy.geometry.line.LinearEntity
is_reduced() (in module
method), 607 sympy.polys.groebnertools), 1024
is_perpendicular() (sympy.geometry.plane.Planeis_right() (sympy.geometry.polygon.Triangle
method), 676 method), 670
is_polynomial() (sympy.core.expr.Expr is_right_unbounded (sympy.sets.sets.Interval
method), 131 attribute), 1170
is_positive() (sympy.polys.domains.AlgebraicField
is_scalar_multiple() (sympy.geometry.point.Point
method), 941 method), 594
is_positive() (sympy.polys.domains.domain.Domainis_scalene() (sympy.geometry.polygon.Triangle
method), 933 method), 670
is_positive() (sympy.polys.domains.ExpressionDomain
is_sequence() (in module
method), 944 sympy.core.compatibility), 205
is_positive() (sympy.polys.domains.FractionFieldis_similar() (sympy.geometry.entity.GeometryEntity
method), 942 method), 585
is_positive() (sympy.polys.domains.PolynomialRing
is_similar() (sympy.geometry.line.LinearEntity
method), 939 method), 607
is_primary() (sympy.polys.agca.ideals.Ideal is_similar() (sympy.geometry.polygon.Triangle
method), 920 method), 670
is_prime() (sympy.polys.agca.ideals.Ideal is_simple() (sympy.functions.special.delta_functions.DiracDe
method), 920 method), 500
is_primitive (sympy.polys.polyclasses.DMP is_Singleton (sympy.combinatorics.permutations.Permutatio
attribute), 947 attribute), 226
is_primitive (sympy.polys.polytools.Poly at- is_solenoidal() (in module
tribute), 869 sympy.physics.vector.fieldfunctions),
is_primitive() (sympy.combinatorics.perm_groups.PermutationGroup
1625
method), 252 is_solenoidal() (in module sympy.vector),
is_primitive_root() (in module 1885
sympy.ntheory.residue_ntheory), is_solvable (sympy.combinatorics.perm_groups.Permutation
334 attribute), 253
is_principal() (sympy.polys.agca.ideals.Ideal is_sqf (sympy.polys.polyclasses.DMP at-
method), 920 tribute), 947
is_proper_subset() (sympy.sets.sets.Set is_sqf (sympy.polys.polytools.Poly attribute),
method), 1165 869
is_proper_superset() (sympy.sets.sets.Set is_strictly_decreasing() (in module
method), 1165 sympy.calculus.singularities), 1503
is_q_annihilator (sympy.physics.secondquant.AnnihilateFermion
is_strictly_increasing() (in module
attribute), 1524 sympy.calculus.singularities), 1503
is_q_annihilator (sympy.physics.secondquant.CreateFermion
is_subdiagram() (sympy.categories.Diagram
attribute), 1525 method), 1822
is_q_creator (sympy.physics.secondquant.AnnihilateFermion
is_subgroup() (sympy.combinatorics.perm_groups.Permutati
attribute), 1524 method), 253
is_q_creator (sympy.physics.secondquant.CreateFermion
is_submodule() (sympy.polys.agca.modules.FreeModule
attribute), 1526 method), 914
is_quad_residue() (in module is_submodule() (sympy.polys.agca.modules.Module
sympy.ntheory.residue_ntheory), method), 913
336 is_submodule() (sympy.polys.agca.modules.QuotientModule
is_quadratic (sympy.polys.polyclasses.DMP method), 922
2000 Index
SymPy Documentation, Release 1.3
is_submodule() (sympy.polys.agca.modules.SubModule
isolate() (in module
method), 916 sympy.polys.numberfields), 893
is_subset() (sympy.sets.sets.Set method), IsomorphismFailed (class in
1166 sympy.polys.polyerrors), 1026
is_superset() (sympy.sets.sets.Set method), isprime() (in module
1166 sympy.ntheory.primetest), 333
is_surjective() (sympy.polys.agca.homomorphisms.ModuleHomomorphism
issubset() (sympy.sets.sets.Set method), 1166
method), 927 issuperset() (sympy.sets.sets.Set method),
is_tangent() (sympy.geometry.ellipse.Ellipse 1166
method), 641 ITE (class in sympy.logic.boolalg), 750
is_transitive() (sympy.combinatorics.perm_groups.PermutationGroup
items() (sympy.core.containers.Dict method),
method), 254 204
is_trivial (sympy.combinatorics.perm_groups.PermutationGroup
iter_q_annihilators()
attribute), 254 (sympy.physics.secondquant.NO
is_true (sympy.assumptions.ask.AssumptionKeys method), 1533
attribute), 1110 iter_q_creators() (sympy.physics.secondquant.NO
is_univariate (sympy.polys.polytools.Poly at- method), 1533
tribute), 869 iterable() (in module
is_whole_ring() (sympy.polys.agca.ideals.Ideal sympy.core.compatibility), 204
method), 920 iterate_binary() (sympy.combinatorics.subsets.Subset
is_zero (sympy.geometry.point.Point at- method), 275
tribute), 594 iterate_graycode() (sympy.combinatorics.subsets.Subset
is_zero (sympy.matrices.immutable.ImmutableDenseMatrix method), 275
attribute), 802 itercoeffs() (sympy.polys.rings.PolyElement
is_zero (sympy.polys.polyclasses.ANP at- method), 1004
tribute), 951 itermonomials() (in module
is_zero (sympy.polys.polyclasses.DMF at- sympy.polys.monomials), 893
tribute), 950 itermonoms() (sympy.polys.rings.PolyElement
is_zero (sympy.polys.polyclasses.DMP at- method), 1004
tribute), 947 iterterms() (sympy.polys.rings.PolyElement
is_zero (sympy.polys.polytools.Poly attribute), method), 1004
870
J
is_zero() (sympy.polys.agca.homomorphisms.ModuleHomomorphism
method), 928 jacobi (class in sympy.functions.special.polynomials),
is_zero() (sympy.polys.agca.ideals.Ideal 563
method), 920 jacobi_normalized() (in module
is_zero() (sympy.polys.agca.modules.FreeModule sympy.functions.special.polynomials),
method), 914 564
is_zero() (sympy.polys.agca.modules.Module jacobi_poly() (in module
method), 913 sympy.polys.orthopolys), 900
is_zero() (sympy.polys.agca.modules.QuotientModule
jacobi_symbol() (in module
method), 922 sympy.ntheory.residue_ntheory),
is_zero() (sympy.polys.agca.modules.SubModule 336
method), 917 jacobian() (sympy.diffgeom.CoordSystem
is_zero() (sympy.polys.domains.domain.Domain method), 1835
method), 933 JavascriptCodePrinter (class in
is_zero_dimensional sympy.printing.jscode), 1062
(sympy.polys.polytools.GroebnerBasis jn (class in sympy.functions.special.bessel),
attribute), 888 538
is_zero_dimensional() (in module jn_zeros() (in module
sympy.polys.polytools), 846 sympy.functions.special.bessel),
isdisjoint() (sympy.sets.sets.Set method), 539
1166 join() (sympy.physics.continuum_mechanics.beam.Beam
isign (class in sympy.codegen.fnodes), 439 method), 1789
Index 2001
SymPy Documentation, Release 1.3
2002 Index
SymPy Documentation, Release 1.3
sympy.functions.special.tensor_functions),
laplace_transform() (in module
576 sympy.integrals.transforms), 692
KroneckerDelta (class in latex() (in module sympy.printing.latex), 1072
sympy.physics.secondquant), 1518 LaTeXParsingError (class in
ksubsets() (sympy.combinatorics.subsets sympy.parsing.latex), 1500
method), 280 LatexPrinter (class in sympy.printing.latex),
Kumaraswamy() (in module sympy.stats), 1072
1237 lbound() (in module sympy.codegen.fnodes),
kwargs() (sympy.codegen.ast.Token method), 439
424 LC() (in module sympy.polys.polytools), 829
LC() (sympy.polys.polyclasses.ANP method),
L 951
l1_norm() (sympy.polys.polyclasses.DMP LC() (sympy.polys.polyclasses.DMP method),
method), 947 945
l1_norm() (sympy.polys.polytools.Poly LC() (sympy.polys.polytools.Poly method),
method), 870 848
L2 (class in sympy.physics.quantum.hilbert), lcm() (in module sympy.polys.polytools), 838
1718 lcm() (sympy.core.numbers.Number method),
Label (class in sympy.codegen.cnodes), 433 143
label (sympy.physics.quantum.state.TimeDepState lcm() (sympy.polys.domains.domain.Domain
attribute), 1745 method), 933
label (sympy.tensor.indexed.Idx attribute), lcm() (sympy.polys.domains.field.Field
1408 method), 935
label (sympy.tensor.indexed.IndexedBase at- lcm() (sympy.polys.domains.PolynomialRing
tribute), 1411 method), 939
labeller() (in module lcm() (sympy.polys.domains.RealField
sympy.physics.quantum.circuitplot), method), 942
1750 lcm() (sympy.polys.polyclasses.DMP method),
LagrangesMethod (class in 947
sympy.physics.mechanics.lagrange), lcm() (sympy.polys.polytools.Poly method),
1695 870
Lagrangian() (in module lcm_list() (in module sympy.polys.polytools),
sympy.physics.mechanics.functions), 838
1691 ldescent() (in module
laguerre (class in sympy.solvers.diophantine), 1369
sympy.functions.special.polynomials), LDLdecomposition()
571 (sympy.matrices.matrices.MatrixBase
laguerre_poly() (in module method), 763
sympy.polys.orthopolys), 900 LDLdecomposition()
Lambda (class in sympy.core.function), 181 (sympy.matrices.sparse.SparseMatrix
lambda_notation() (in module method), 792
sympy.parsing.sympy_parser), 1499 LDLsolve() (sympy.matrices.matrices.MatrixBase
LambdaPrinter (class in method), 764
sympy.printing.lambdarepr), 1072 Le (in module sympy.core.relational), 167
lambdarepr() (in module leading_expv() (sympy.polys.rings.PolyElement
sympy.printing.lambdarepr), 1072 method), 1004
lambdastr() (in module leading_monom() (sympy.polys.rings.PolyElement
sympy.utilities.lambdify), 1478 method), 1004
lambdify() (in module leading_term() (sympy.polys.rings.PolyElement
sympy.utilities.lambdify), 1479 method), 1004
LambertW (class in leadterm() (sympy.core.expr.Expr method),
sympy.functions.elementary.exponential), 133
470 left (sympy.sets.sets.Interval attribute), 1170
Laplace() (in module sympy.stats), 1238 left() (sympy.printing.pretty.stringpict.stringPict
Index 2003
SymPy Documentation, Release 1.3
2004 Index
SymPy Documentation, Release 1.3
Index 2005
SymPy Documentation, Release 1.3
492 562
LUdecomposition() (sympy.matrices.matrices.MatrixBase
mathieus (class in
method), 764 sympy.functions.special.mathieu_functions),
LUdecomposition_Simple() 560
(sympy.matrices.matrices.MatrixBase mathieusprime (class in
method), 765 sympy.functions.special.mathieu_functions),
LUdecompositionFF() 561
(sympy.matrices.matrices.MatrixBase mathml() (in module sympy.printing.mathml),
method), 765 1076
LUsolve() (sympy.matrices.matrices.MatrixBasemathml_tag() (sympy.printing.mathml.MathMLContentPrint
method), 766 method), 1076
mathml_tag() (sympy.printing.mathml.MathMLPresentationP
M method), 1076
magnitude() (sympy.physics.vector.vector.VectorMathMLContentPrinter (class in
method), 1602 sympy.printing.mathml), 1076
magnitude() (sympy.vector.vector.Vector MathMLPresentationPrinter (class in
method), 1874 sympy.printing.mathml), 1076
major (sympy.geometry.ellipse.Ellipse at- MathMLPrinterBase (class in
tribute), 641 sympy.printing.mathml), 1076
MatMul (class in sympy.matrices.expressions),
make_perm() (sympy.combinatorics.perm_groups.PermutationGroup
method), 255 806
make_routine() (in module MatPow (class in
sympy.utilities.codegen), 1448 sympy.matrices.expressions), 806
Manifold (class in sympy.diffgeom), 1832 matrix2numpy() (in module
map() (sympy.polys.domains.domain.Domain sympy.matrices.dense), 784
method), 933 matrix_fglm() (in module
mass (sympy.physics.mechanics.particle.Particle sympy.polys.fglmtools), 1024
attribute), 1683 matrix_form() (sympy.liealgebras.weyl_group.WeylGroup
mass_matrix (sympy.physics.mechanics.kane.KanesMethod method), 740
attribute), 1694 matrix_multiply_elementwise() (in module
sympy.matrices.dense), 779
mass_matrix (sympy.physics.mechanics.lagrange.LagrangesMethod
attribute), 1697 matrix_rep() (in module
sympy.physics.secondquant), 1530
mass_matrix_full (sympy.physics.mechanics.kane.KanesMethod
attribute), 1694 matrix_to_density() (in module
sympy.physics.quantum.qubit), 1761
mass_matrix_full (sympy.physics.mechanics.lagrange.LagrangesMethod
attribute), 1697 matrix_to_qubit() (in module
MatAdd (class in sympy.matrices.expressions), sympy.physics.quantum.qubit), 1760
806 matrix_to_vector() (in module sympy.vector),
match() (sympy.core.basic.Basic method), 1881
107 MatrixBase (class in
matches() (sympy.core.basic.Basic method), sympy.matrices.matrices), 762
107 MatrixError (class in
mathematica() (in module sympy.matrices.matrices), 779
sympy.parsing.mathematica), 1497 MatrixExpr (class in
mathematica_code() (in module sympy.matrices.expressions), 803
sympy.printing.mathematica), 1062 MatrixSymbol (class in
MathieuBase (class in sympy.matrices.expressions), 805
Max (class in sympy.functions.elementary.miscellaneous),
sympy.functions.special.mathieu_functions),
560 473
mathieuc (class in max (sympy.codegen.ast.FloatType at-
sympy.functions.special.mathieu_functions), tribute), 419
561 max() (sympy.combinatorics.permutations.Permutation
mathieucprime (class in method), 229
max_bmoment()
sympy.functions.special.mathieu_functions), (sympy.physics.continuum_mechanics.beam
2006 Index
SymPy Documentation, Release 1.3
Index 2007
SymPy Documentation, Release 1.3
2008 Index
SymPy Documentation, Release 1.3
Index 2009
SymPy Documentation, Release 1.3
2010 Index
SymPy Documentation, Release 1.3
Index 2011
SymPy Documentation, Release 1.3
orient_new() (sympy.vector.coordsysrect.CoordSys3D
parameter (sympy.geometry.curve.Curve at-
method), 1867 tribute), 633
orient_new_axis() (sympy.vector.coordsysrect.CoordSys3D
parameter_value() (sympy.geometry.entity.GeometryEntity
method), 1868 method), 585
orient_new_body() (sympy.vector.coordsysrect.CoordSys3D
parameter_value() (sympy.geometry.plane.Plane
method), 1868 method), 677
orient_new_quaternion() Parametric2DLineSeries (class in
(sympy.vector.coordsysrect.CoordSys3D sympy.plotting.plot), 1100
method), 1869 Parametric3DLineSeries (class in
orient_new_space() (sympy.vector.coordsysrect.CoordSys3Dsympy.plotting.plot), 1100
method), 1870 ParametricSurfaceSeries (class in
Orienter (class in sympy.vector.orienters), sympy.plotting.plot), 1101
1878 parametrize_ternary_quadratic() (in module
orientnew() (sympy.physics.vector.frame.ReferenceFrame sympy.solvers.diophantine), 1368
method), 1596 parens() (sympy.printing.pretty.stringpict.stringPict
origin (sympy.geometry.point.Point attribute), method), 1085
595 Pareto() (in module sympy.stats), 1243
orthocenter (sympy.geometry.polygon.Triangle parity() (sympy.combinatorics.permutations.Permutation
attribute), 672 method), 231
orthogonal (sympy.assumptions.ask.AssumptionKeysparse_expr() (in module
attribute), 1114 sympy.parsing.sympy_parser), 1496
orthogonal_direction parse_latex() (in module
(sympy.geometry.point.Point at- sympy.parsing.latex), 1500
tribute), 595 parse_maxima() (in module
outer() (in module sympy.parsing.maxima), 1497
sympy.physics.vector.functions), partial_velocity() (in module
1619 sympy.physics.vector.functions),
outer() (sympy.physics.vector.vector.Vector 1612
method), 1602 partial_velocity() (sympy.physics.vector.frame.ReferenceFra
outer() (sympy.vector.vector.Vector method), method), 1597
1874 partial_velocity() (sympy.physics.vector.point.Point
OuterProduct (class in method), 1608
sympy.physics.quantum.operator), Particle (class in sympy.physics.mechanics.particle),
1721 1681
OutputArgument (class in Partition (class in
sympy.utilities.codegen), 1439 sympy.combinatorics.partitions),
208
P partition (class in
P() (in module sympy.stats), 1252 sympy.functions.combinatorial.numbers),
p1 (sympy.geometry.line.LinearEntity at- 493
tribute), 608 partition (sympy.combinatorics.partitions.Partition
p1 (sympy.geometry.plane.Plane attribute), attribute), 209
677 partition() (in module
p2 (sympy.geometry.line.LinearEntity at- sympy.solvers.diophantine), 1363
tribute), 608 partitions() (in module
padded_key() (in module sympy.utilities.iterables), 1470
sympy.crypto.crypto), 343 pat_matrix() (in module
parallel_line() (sympy.geometry.line.LinearEntity sympy.physics.matrices), 1515
method), 608 Patch (class in sympy.diffgeom), 1832
parallel_plane() (sympy.geometry.plane.Plane pde_1st_linear_constant_coeff() (in module
method), 677 sympy.solvers.pde), 1325
parallel_poly_from_expr() (in module pde_1st_linear_constant_coeff_homogeneous()
sympy.polys.polytools), 828 (in module sympy.solvers.pde), 1324
pde_1st_linear_variable_coeff() (in module
2012 Index
SymPy Documentation, Release 1.3
Index 2013
SymPy Documentation, Release 1.3
2014 Index
SymPy Documentation, Release 1.3
position_x (sympy.physics.quantum.cartesian.PositionState3D
Pow() (sympy.assumptions.handlers.order.AskNegativeHand
attribute), 1717 static method), 1127
position_y (sympy.physics.quantum.cartesian.PositionState3D
Pow() (sympy.assumptions.handlers.sets.AskAntiHermitianH
attribute), 1717 static method), 1128
position_z (sympy.physics.quantum.cartesian.PositionState3D
Pow() (sympy.assumptions.handlers.sets.AskHermitianHand
attribute), 1717 static method), 1128
PositionBra3D (class in Pow() (sympy.assumptions.handlers.sets.AskImaginaryHand
sympy.physics.quantum.cartesian), static method), 1129
1717 Pow() (sympy.assumptions.handlers.sets.AskIntegerHandler
PositionKet3D (class in static method), 1129
sympy.physics.quantum.cartesian), Pow() (sympy.assumptions.handlers.sets.AskRationalHandle
1717 static method), 1129
PositionState3D (class in Pow() (sympy.assumptions.handlers.sets.AskRealHandler
sympy.physics.quantum.cartesian), static method), 1129
1717 pow() (sympy.polys.domains.domain.Domain
positive, 99 method), 933
positive (sympy.assumptions.ask.AssumptionKeys pow() (sympy.polys.polyclasses.ANP method),
attribute), 1114 951
positive_definite (sympy.assumptions.ask.AssumptionKeys
pow() (sympy.polys.polyclasses.DMF
attribute), 1115 method), 951
positive_roots() (sympy.liealgebras.type_a.TypeApow() (sympy.polys.polyclasses.DMP
method), 730 method), 948
positive_roots() (sympy.liealgebras.type_b.TypeBpow() (sympy.polys.polytools.Poly method),
method), 732 875
positive_roots() (sympy.liealgebras.type_c.TypeCpow_xin() (in module
method), 733 sympy.polys.ring_series), 1046
positive_roots() (sympy.liealgebras.type_d.TypeDpowdenest() (in module
method), 734 sympy.simplify.powsimp), 1202
positive_roots() (sympy.liealgebras.type_e.TypeEpowerset() (sympy.sets.sets.Set method),
method), 736 1166
positive_roots() (sympy.liealgebras.type_f.TypeFpowsimp() (in module
method), 737 sympy.simplify.powsimp), 1201
positive_roots() (sympy.liealgebras.type_g.TypeGpowsimp() (sympy.core.expr.Expr method),
method), 738 134
PostDecrement (class in pprint_nodes() (in module
sympy.codegen.cnodes), 433 sympy.printing.tree), 1079
postfixes() (in module PQa() (in module sympy.solvers.diophantine),
sympy.utilities.iterables), 1471 1367
PostIncrement (class in pquo() (in module sympy.polys.polytools), 830
sympy.codegen.cnodes), 433 pquo() (sympy.polys.polyclasses.DMP
postorder_traversal() (in module method), 948
sympy.utilities.iterables), 1471 pquo() (sympy.polys.polytools.Poly method),
potential_energy (sympy.physics.mechanics.particle.Particle
876
attribute), 1683 PRECEDENCE (in module
potential_energy (sympy.physics.mechanics.rigidbody.RigidBody
sympy.printing.precedence), 1083
attribute), 1686 precedence() (in module
potential_energy() (in module sympy.printing.precedence), 1083
sympy.physics.mechanics.functions), PRECEDENCE_FUNCTIONS (in module
1690 sympy.printing.precedence), 1083
Pow (class in sympy.core.power), 157 PRECEDENCE_VALUES (in module
Pow() (sympy.assumptions.handlers.calculus.AskFiniteHandler
sympy.printing.precedence), 1083
static method), 1126 PrecisionExhausted (class in
Pow() (sympy.assumptions.handlers.ntheory.AskPrimeHandler
sympy.core.evalf), 202
static method), 1127 PreDecrement (class in
Index 2015
SymPy Documentation, Release 1.3
2016 Index
SymPy Documentation, Release 1.3
Index 2017
SymPy Documentation, Release 1.3
2018 Index
SymPy Documentation, Release 1.3
Index 2019
SymPy Documentation, Release 1.3
2020 Index
SymPy Documentation, Release 1.3
Index 2021
SymPy Documentation, Release 1.3
2022 Index
SymPy Documentation, Release 1.3
Index 2023
SymPy Documentation, Release 1.3
2024 Index
SymPy Documentation, Release 1.3
Index 2025
SymPy Documentation, Release 1.3
2026 Index
SymPy Documentation, Release 1.3
1328 110
solve() (sympy.matrices.matrices.MatrixBase sorted_components()
method), 776 (sympy.tensor.tensor.TensMul
solve() (sympy.matrices.sparse.SparseMatrix method), 1429
method), 796 source (sympy.geometry.line.Ray attribute),
solve_congruence() (in module 617
sympy.ntheory.modular), 330 source() (in module sympy.utilities.source),
solve_de() (in module sympy.series.formal), 1495
1159 SpaceOrienter (class in
solve_for_reaction_loads() sympy.vector.orienters), 1880
SparseMatrix
(sympy.physics.continuum_mechanics.beam.Beam (class in
method), 1796 sympy.matrices.sparse), 792
solve_for_reaction_loads() speed (sympy.physics.optics.medium.Medium
(sympy.physics.continuum_mechanics.beam.Beam3D attribute), 1777
method), 1800 speed (sympy.physics.optics.waves.TWave at-
solve_least_squares() tribute), 1783
(sympy.matrices.matrices.MatrixBase speeds (sympy.physics.mechanics.system.SymbolicSystem
method), 776 attribute), 1702
solve_least_squares() spin() (sympy.geometry.polygon.RegularPolygon
(sympy.matrices.sparse.SparseMatrix method), 664
method), 796 split() (sympy.tensor.tensor.TensMul method),
solve_linear() (in module 1429
sympy.solvers.solvers), 1336 split_list() (in module
solve_linear_system() (in module sympy.utilities.runtests), 1491
sympy.solvers.solvers), 1337 split_super_sub() (in module
solve_linear_system_LU() (in module sympy.printing.conventions), 1082
sympy.solvers.solvers), 1338 split_symbols() (in module
sympy.parsing.sympy_parser), 1497
solve_multipliers() (sympy.physics.mechanics.lagrange.LagrangesMethod
method), 1697 split_symbols_custom() (in module
solve_poly_inequality() (in module sympy.parsing.sympy_parser), 1497
sympy.solvers.inequalities), 1372 spoly() (in module
solve_poly_system() (in module sympy.polys.groebnertools), 1024
sympy.solvers.polysys), 1346 sqf() (in module sympy.polys.polytools), 841
solve_rational_inequalities() (in module sqf_list() (in module sympy.polys.polytools),
sympy.solvers.inequalities), 1372 841
solve_triangulated() (in module sqf_list() (sympy.polys.polyclasses.DMP
sympy.solvers.polysys), 1346 method), 949
solve_undetermined_coeffs() (in module sqf_list() (sympy.polys.polytools.Poly
sympy.solvers.solvers), 1338 method), 881
solve_univariate_inequality() (in module sqf_list_include() (sympy.polys.polyclasses.DMP
sympy.solvers.inequalities), 1374 method), 949
solveset() (in module sympy.solvers.solveset), sqf_list_include() (sympy.polys.polytools.Poly
1383 method), 881
solveset_complex() (in module sqf_norm() (in module sympy.polys.polytools),
sympy.solvers.solveset), 1385 841
solveset_real() (in module sqf_norm() (sympy.polys.polyclasses.DMP
sympy.solvers.solveset), 1385 method), 949
SOPform() (in module sympy.logic.boolalg), sqf_norm() (sympy.polys.polytools.Poly
742 method), 882
sort_dims() (sympy.physics.units.dimensions.DimensionSystem
sqf_normal() (in module
static method), 1553 sympy.solvers.diophantine), 1371
sort_key() (sympy.combinatorics.partitions.Partition
sqf_part() (in module sympy.polys.polytools),
method), 209 841
sort_key() (sympy.core.basic.Basic method), sqf_part() (sympy.polys.polyclasses.DMP
Index 2027
SymPy Documentation, Release 1.3
2028 Index
SymPy Documentation, Release 1.3
sympy.functions.combinatorial.factorials),
sum_of_four_squares() (in module
486 sympy.solvers.diophantine), 1365
subgroup() (sympy.combinatorics.perm_groups.PermutationGroup
sum_of_powers() (in module
method), 265 sympy.solvers.diophantine), 1365
subgroup_search() (sympy.combinatorics.perm_groups.PermutationGroup
sum_of_squares() (in module
method), 265 sympy.solvers.diophantine), 1366
SubModule (class in sum_of_three_squares() (in module
sympy.polys.agca.modules), 915 sympy.solvers.diophantine), 1364
submodule() (sympy.polys.agca.modules.ModuleSumApprox (class in
method), 913 sympy.codegen.approximations),
submodule() (sympy.polys.agca.modules.QuotientModule 411
method), 923 summation() (in module
submodule() (sympy.polys.agca.modules.SubModule sympy.concrete.summations), 379
method), 918 sup (in module sympy.printing.pretty.pretty_symbology),
SubQuotientModule (class in 1084
sympy.polys.agca.modules), 923 sup (sympy.sets.sets.Set attribute), 1167
subresultants() (in module superposition_basis() (in module
sympy.polys.polytools), 833 sympy.physics.quantum.grover),
subresultants() (sympy.polys.polyclasses.DMP 1756
method), 949 superset (sympy.combinatorics.subsets.Subset
subresultants() (sympy.polys.polytools.Poly attribute), 279
method), 884 superset_size (sympy.combinatorics.subsets.Subset
Subroutine (class in sympy.codegen.fnodes), attribute), 280
436 support() (sympy.combinatorics.permutations.Permutation
SubroutineCall (class in method), 234
sympy.codegen.fnodes), 437 SurfaceBaseSeries (class in
Subs (class in sympy.core.function), 191 sympy.plotting.plot), 1100
subs() (sympy.core.basic.Basic method), 111 SurfaceOver2DRangeSeries (class in
subs() (sympy.physics.vector.dyadic.Dyadic sympy.plotting.plot), 1100
method), 1605 SWAP (in module
subs() (sympy.physics.vector.vector.Vector sympy.physics.quantum.gate), 1754
method), 1603 SwapGate (class in
Subset (class in sympy.combinatorics.subsets), sympy.physics.quantum.gate), 1753
274 swinnerton_dyer_poly() (in module
subset (sympy.combinatorics.subsets.Subset sympy.polys.specialpolys), 898
attribute), 279 sylow_subgroup() (sympy.combinatorics.perm_groups.Permu
subset() (sympy.polys.agca.ideals.Ideal method), 266
method), 921 symarray() (in module
subset() (sympy.polys.agca.modules.Module sympy.matrices.dense), 784
method), 913 symb_2txt (in module
subset_from_bitlist() sympy.printing.pretty.pretty_symbology),
(sympy.combinatorics.subsets.Subset 1083
class method), 279 Symbol (class in sympy.core.symbol), 137
subset_indices() (sympy.combinatorics.subsets.Subset
symbol (sympy.physics.secondquant.AntiSymmetricTensor
class method), 279 attribute), 1536
subsets() (in module Symbol() (sympy.assumptions.handlers.calculus.AskFiniteHa
sympy.utilities.iterables), 1475 static method), 1127
SubsSet (class in sympy.series.gruntz), 1139 SymbolicSystem (class in
substitute_dummies() (in module sympy.physics.mechanics.system),
sympy.physics.secondquant), 1536 1698
substitute_indices() (sympy.tensor.tensor.TensAdd
symbols() (in module sympy.core.symbol),
method), 1427 139
Sum (class in sympy.concrete.summations), symmetric (sympy.assumptions.ask.AssumptionKeys
369 attribute), 1118
Index 2029
SymPy Documentation, Release 1.3
2030 Index
SymPy Documentation, Release 1.3
Index 2031
SymPy Documentation, Release 1.3
2032 Index
SymPy Documentation, Release 1.3
T 1421
T (in module sympy.physics.quantum.gate), TensorIndex (class in sympy.tensor.tensor),
1754 1418
T (sympy.matrices.expressions.MatrixExpr TensorIndexType (class in
attribute), 804 sympy.tensor.tensor), 1416
t (sympy.physics.quantum.shor.CMod at- TensorProduct (class in sympy.diffgeom),
tribute), 1763 1839
table() (sympy.matrices.matrices.MatrixBase TensorProduct (class in
method), 777 sympy.physics.quantum.tensorproduct),
tail_degree() (sympy.polys.rings.PolyElement 1715
method), 1005 tensorproduct() (in module
tail_degrees() (sympy.polys.rings.PolyElement sympy.tensor.array), 1405
method), 1005 TensorSymmetry (class in
take() (in module sympy.utilities.iterables), sympy.tensor.tensor), 1419
1476 tensorsymmetry() (in module
tan (class in sympy.functions.elementary.trigonometric),sympy.tensor.tensor), 1420
454 TensorType (class in sympy.tensor.tensor),
tangent_lines() (sympy.geometry.ellipse.Ellipse 1420
method), 646 terminal_width() (sympy.printing.pretty.stringpict.stringPict
tanh (class in sympy.functions.elementary.hyperbolic), method), 1086
464 terms() (sympy.polys.polyclasses.DMP
targets (sympy.physics.quantum.gate.CGate method), 949
attribute), 1752 terms() (sympy.polys.polytools.Poly method),
targets (sympy.physics.quantum.gate.CNotGate 884
attribute), 1754 terms() (sympy.polys.rings.PolyElement
targets (sympy.physics.quantum.gate.Gate method), 1006
attribute), 1751 terms_gcd() (in module
targets (sympy.physics.quantum.gate.UGate sympy.polys.polytools), 836
attribute), 1752 terms_gcd() (sympy.polys.polyclasses.DMP
targets (sympy.physics.quantum.grover.OracleGate method), 949
attribute), 1756 terms_gcd() (sympy.polys.polytools.Poly
taxicab_distance() (sympy.geometry.point.Point method), 884
method), 596 termwise() (sympy.polys.polytools.Poly
taylor_term() (sympy.core.expr.Expr method), method), 884
137 test() (in module sympy.utilities.runtests),
TC() (sympy.polys.polyclasses.ANP method), 1493
951 test_derivative_numerically() (in module
TC() (sympy.polys.polyclasses.DMP method), sympy.utilities.randtest), 1487
945 TGate (class in sympy.physics.quantum.gate),
TC() (sympy.polys.polytools.Poly method), 1753
849 theano_function() (in module
TensAdd (class in sympy.tensor.tensor), 1425 sympy.printing.theanocode), 1071
TensExpr (class in sympy.tensor.tensor), 1424 TheanoPrinter (class in
TensMul (class in sympy.tensor.tensor), 1427 sympy.printing.theanocode), 1071
tensor_indices() (in module ThinLens (class in
sympy.tensor.tensor), 1419 sympy.physics.optics.gaussopt),
tensor_mul() (in module sympy.tensor.tensor), 1768
1430 threaded() (in module
tensor_product_simp() (in module sympy.utilities.decorator), 1450
sympy.physics.quantum.tensorproduct),threaded_factory() (in module
1716 sympy.utilities.decorator), 1451
tensorcontraction() (in module time (sympy.physics.quantum.state.TimeDepState
sympy.tensor.array), 1404 attribute), 1745
TensorHead (class in sympy.tensor.tensor), time_derivative() (in module
Index 2033
SymPy Documentation, Release 1.3
2034 Index
SymPy Documentation, Release 1.3
Index 2035
SymPy Documentation, Release 1.3
2036 Index
SymPy Documentation, Release 1.3
Index 2037
SymPy Documentation, Release 1.3
2038 Index