用户工具

站点工具


hacks:solve-x-y

3行Python代码解简单的一元一次方程

刚才看到一篇《Linear equations solver in 3 lines (Python recipe)》,用3行代码实现了解一元一次方程:

>>> solve("x - 2*x + 5*x - 46*(235-24) = x + 2")
3236.0

源码:

def solve(eq,var='x'):
    eq1 = eq.replace("=","-(")+")"
    c = eval(eq1,{var:1j})
    return -c.real/c.imag

看上去很强大,于是就解读下代码吧。

首先是第一行,它将等式进行了变形,生成了一个结果为0的算式“x - 2*x + 5*x - 46*(235-24) -( x + 2)”。

第二行用eval来执行这个算式,并将x = 1j代入算式,结果是-9708+3j。

注意x = 1j,所以这个方程就化简为“-9708+3x = 0”了,只要将-(-9708) / 3就能得到x了。

而-9708是这个复数的实部,3是这个复数的虚部,于是结果变成了“-c.real/c.imag”。

很显然,这个函数是不能解复数方程的。

顺带一提,Python 2.x的/运算会使用整数除法,导致小数部分丢失,所以要获得正确结果就应该使用Python 3.x。

参考

hacks/solve-x-y.txt · 最后更改: 2014/06/09 14:29 (外部编辑)