Skip to content

hgrecco/symbolite

Folders and files

NameName
Last commit message
Last commit date

Latest commit

058f778 · Jan 27, 2025

History

97 Commits
Jan 23, 2025
Jan 25, 2025
Dec 14, 2022
Jan 21, 2025
Jan 21, 2025
Dec 14, 2022
Jan 27, 2025
Dec 14, 2022
Jan 23, 2025
Jul 1, 2023
Jan 23, 2025
Jan 23, 2025

Repository files navigation

Package CodeStyle License PyVersion CI Lint

symbolite: a minimalistic symbolic python package


Symbolite allows you to create symbolic mathematical expressions. Just create a symbol (or more) and operate with them as you will normally do in Python.

>>> from symbolite import Symbol
>>> x = Symbol("x")
>>> y = Symbol("y")
>>> expr1 = x + 3 * y
>>> print(expr1)
x + 3 * y

An expression is just an unnamed Symbol. You can easily replace the symbols by the desired value.

>>> expr2 = expr1.subs_by_name(x=5, y=2)
>>> print(expr2)
5 + 3 * 2

The output is still a symbolic expression, which you can evaluate:

>>> expr2.eval()
11

Notice that we also got a warning (No libsl provided, defaulting to Python standard library.). This is because evaluating an expression requires a actual library implementation, name usually as libsl. The default one just uses python's math module.

You can avoid this warning by explicitely providing an libsl implementation.

>>> from symbolite.impl import libstd
>>> expr2.eval(libstd)
11

You can also import it with the right name and it will be found

>>> from symbolite.impl import libstd as libsl
>>> expr2.eval()
11

In addition to the Symbol class, there is also a Scalar and Vector classes to represent integer, floats or complex numbers, and an array of those.

>>> from symbolite import Scalar, Vector
>>> x = Scalar("x")
>>> y = Scalar("y")
>>> v = Vector("v")
>>> expr1 = x + 3 * y
>>> print(expr1)
x + 3 * y
>>> print(2 * v)
2 * v

Mathematical functions that operate on scalars are available in the scalar module.

>>> from symbolite import scalar
>>> expr3 = 3. * scalar.cos(0.5)
>>> print(expr3)
3.0 * scalar.cos(0.5)

Mathematical functions that operate on vectors are available in the vector module.

>>> from symbolite import vector
>>> expr4 = 3. * vector.sum((1, 2, 3))
>>> print(expr4)
3.0 * vector.sum((1, 2, 3))

Notice that functions are named according to the python math module. Again, this is a symbolic expression until evaluated.

>>> expr3.eval()
2.6327476856711
>>> expr4.eval()
18.0

Three other implementations are provided: NumPy, SymPy, JAX.

>>> from symbolite.impl import libnumpy
>>> expr3.eval(libsl=libnumpy)
np.float64(2.6327476856711183)
>>> from symbolite.impl import libsympy
>>> expr3.eval(libsl=libsympy)
2.6327476856711

(notice that the way that the different libraries round and display may vary)

In general, all symbols must be replaced by values in order to evaluate an expression. However, when using an implementation like SymPy that contains a Scalar object you can still evaluate.

>>> from symbolite.impl import libsympy as libsl
>>> (3. * scalar.cos(x).eval(libsl))
3.0*cos(x)

which is actually a SymPy expression with a SymPy symbol (x).

And by the way, checkout vectorize and auto_vectorize functions in the vector module.

We provide a simple way to call user defined functions.

>>> from symbolite import UserFunction
>>> def abs_times_two(x: float) -> float:
...     return 2 * abs(x)
>>> uf = UserFunction.from_function(abs_times_two)
>>> uf
UserFunction(name='abs_times_two', namespace='user')
>>> uf(-1).eval()
2

and you can register implementations for other backends:

>>> def np_abs_times_two(x: float) -> float:
...     return 2 * np.abs(x)
>>> uf.register_impl(libnumpy, np_abs_times_two)
>>> uf(-1).eval(libnumpy)
2

Installing:

pip install -U symbolite

FAQ

Q: Is symbolite a replacement for SymPy?

A: No

Q: Does it aim to be a replacement for SymPy in the future?

A: No