Python package simplifies algebraic equations
Programming Snapshot – SymPy

© Lead Image © Andrea Danti, 123RF.com
Whether he's filling the bathtub with water or routing electricity through resistors – Mike Schilli juggles mathematical formulas with the assistance of the Python SymPy package.
I still remember a traumatic event as an elementary school student: A weekend newspaper had set a logic puzzle for kids, the solution of which it promised to publish in the next issue a week later. It involved a bathtub with two taps: one of which filled the tub in 10 minutes, the other in 15. The question was how long would it take to fill the tub if both taps were turned on all the way.
As a little boy, I was absolutely sure that 10 plus 15 equals 25, which is 25 minutes. My father laughed and suggested that couldn't be true, because two taps would fill the bathtub faster than one alone. The next weekend, I was initially triumphant, because there it was – printed in black and white in the following issue – the confirmation that 25 minutes was the correct solution.
But disillusionment followed one week later: After receiving angry reader comments, the newspaper had to admit that it had made a mistake, because it does not take 25 minutes, but only 6 minutes, to fill the tub with both taps. I nearly fell off my chair and decided at that point to become a famous columnist peddling logic puzzles.
Faucets and Resistors
It was only much later, in college, during a lecture on circuit technology, that I came across a similar problem, which somehow translated nicely to a solution to the bathtub problem: parallel arrangement of resistors in a circuit. The electrical engineer measures the resistance of a conductor in ohms, and the higher the value, the more it slows down the flow of electrical current.
Now how do you calculate the equivalent resistance of a parallel circuit with two resistors R1 and R2 as shown in Figure 1? The solution involves the currents I1 and I2 flowing through each resistor. They add up to the total current I after the two branches are reunited. The measured voltage is the same everywhere, that is V, and according to Ohm's law V = R*I. Ergo, with individual currents I1 and I2 adding up to I, the result comes out as V/R1 + V/R2.
If we now replace the two parallel resistors R1 and R2 with a substitute resistor R, we also get I = V/R for the total current, so this gets us: V/R = V/R1 + V/R2. The voltage V can be canceled out to give 1/R = 1/R1 + 1/R2 (Figure 2). After transformation, the following result appears:
R = (R1*R2)/(R1+R2)
Today, computer programs perform the computational work instead of the engineer doing it manually. Listing 1 [1] shows the development of the formula using SymPy, a symbolic algebra package in Python. It can be used to define symbols, which the package later leaves intact when evaluating formulas, instead of immediately replacing variables with values and determining the formula's numerical result. With the simplify()
function, SymPy can simplify expressions using the rules of algebra – just as any mathematically skilled person would do.
Listing 1
parallel.py
01 #!/usr/bin/env python3 02 from sympy import simplify, symbols, pprint 03 r, r1, r2, v, i1, i2 = symbols("r r1 r2 v i1 i2") 04 i1 = v / r1 05 i2 = v / r2 06 r = v / (i1 + i2) 07 pprint(simplify(r))
Thus, line 3 in Listing 1 defines a whole slew of symbols. For the two resistors of the circuit and their equivalent resistance, there are r1
, r2
, and r
. For the applied voltage, there is v
; for the two current components, there are i1
and i2
. The formulas in lines 4 to 6 apply Ohm's law and sum the partial currents to receive at the total current.
The resulting formula for the equivalent resistance r
comes to light when the script is called using the pprint()
(pretty print) function (Figure 3). SymPy has obviously realized that the voltage v
can be canceled out of the fraction, and the result no longer depends on it.
If you use Python 3, you can install packages like sympy and matplotlib, which I'll be using in a moment to illustrate results, easily with the command:
pip3 install <package>
In order to avoid pulling the carpet out from under the feet of other Python scripts on the same host, I prefer doing that in Python's virtual environment [2].
From Circuit to Bathtub
This formula also works for the bathtub problem mentioned earlier: In the numerator, the filling times per tap get multiplied (10*15), and the denominator holds their sum (10+15). According to Adam Riese, this comes out to w150/25 (i.e., six minutes), just like in the example solution in the Sunday paper.
By the way, you can also think of the solution as follows: After one minute, the first tap has filled the tub to one tenth and the second to one fifteenth. Both together fill in one minute 1/10 + 1/15 = 3/30 + 2/30 = 5/30 = 1/6 of the tub, so you can get into the bath after six minutes.
Borderline
What happens if both resistors in the parallel circuit drop down to 0 ohms – that is, they do not slow down the current flow at all, or, in the case of a bathtub with Niagara Falls-style faucets, fill the tub in next to no time? Intuitively, it is clear that this kind of mega-faucet would fill the tub practically as fast when connected in parallel as when operated individually. But if you set the values for (R1*R2)/(R1+R2) in the formula (R1*R2)/(R2)/(R1+R2) to zero and pass the construct to a Python program for computing, you will experience a fatal error: Computers steadfastly refuse to perform divisions by zero, because it is mathematically undefined.
In the present case, however, both the numerator and the denominator contain a value tending towards zero, which sometimes yields interesting (because finite) results. Mind you, a true zero as denominator defies mathematical definition, but the limit for R1 and R2 approaching zero can definitely be computed.
SymPy provides the limit()
function, which takes a symbolic formula, a symbol (such as r1
) and a limit (here
). Listing 2 defines r1 = r2
beforehand, meaning that both variables in the limit()
function tend towards zero. The result for the equivalent resistor, as the values for r1
and r2
approach zero, is that the script returns, as expected, a value of zero (Listing 3).
Listing 2
limit.py
01 #!/usr/bin/env python3 02 from sympy import limit, symbols 03 r1, r2, r = symbols("r1 r2 r") 04 05 r = (r1 * r2) / (r1 + r2) 06 r1 = r2 07 08 # r1/2->0 09 print(limit(r, r1, 0)) 10 11 # 1/x with x->0 12 x = symbols("x") 13 print(limit(1/x, x, 0))
Listing 3
Limit Value
$ ./limit.py 0 oo
The second test case, starting from line 12 in Listing 2, illustrates with an example what happens to another formula, 1/x
, when x
tends towards zero. The output in Listing 3 shows here that the result of the formula tends to infinity in this case, which is what the ASCII output oo
tries to illustrate.
Buy this article as PDF
(incl. VAT)
Buy Linux Magazine
Direct Download
Read full article as PDF:
Price $2.95
Subscribe to our Linux Newsletters
Find Linux and Open Source Jobs
Subscribe to our ADMIN Newsletters
Support Our Work
Linux Magazine content is made possible with support from readers like you. Please consider contributing when you've found an article to be beneficial.
News
-
Armbian 23.05 is Now Available
Based on Debian 12, the latest version of the ARM/RISC-V distribution is now available to download and install.
-
Linux Mint Finally Receiving Support for Gestures
If you use the Linux Mint Cinnamon desktop, you'll be thrilled to know that 21.2 is getting support for gestures on touchscreen devices and touchpads.
-
An All-Snap Version of Ubuntu is In The Works
Along with the standard deb version of the open-source operating system, Canonical will release an-all snap version.
-
Mageia 9 Beta 2 Ready for Testing
The latest beta of the popular Mageia distribution now includes the latest kernel and plenty of updated applications.
-
KDE Plasma 6 Looks to Bring Basic HDR Support
The KWin piece of KDE Plasma now has HDR support and color management geared for the 6.0 release.
-
Bodhi Linux 7.0 Beta Ready for Testing
The latest iteration of the Bohdi Linux distribution is now available for those who want to experience what's in store and for testing purposes.
-
Changes Coming to Ubuntu PPA Usage
The way you manage Personal Package Archives will be changing with the release of Ubuntu 23.10.
-
AlmaLinux 9.2 Now Available for Download
AlmaLinux has been released and provides a free alternative to upstream Red Hat Enterprise Linux.
-
An Immutable Version of Fedora Is Under Consideration
For anyone who's a fan of using immutable versions of Linux, the Fedora team is currently considering adding a new spin called Fedora Onyx.
-
New Release of Br OS Includes ChatGPT Integration
Br OS 23.04 is now available and is geared specifically toward web content creation.