python加速运算_加速sympy矩阵运算

我写了一个代码，它使用sympy来建立一个矩阵和一个向量。这两个元素是sympy符号。然后我把矩阵求逆，然后把逆矩阵和向量相乘。这应该是一个具有n个变量的线性方程组的通用解算器。我对这些线性方程的符号解法感兴趣。

问题是我的代码太慢了。

例如，对于n=4，它大约需要30秒，但是对于n=7，我到目前为止还没有解决它，代码运行了一整晚(8小时)，早上还没有完成。

这是我的密码。在from sympy import *

import pprint

MM = Matrix(niso,1, lambda i,j:var('MM_%s' % (i+1) ))

MA = Matrix (niso,1, lambda i,j:var('m_%s%s' % ('A', chr(66+i)) ) )

MX = Matrix (niso,1, lambda i,j:var('m_%s%s'% (chr(66+i), 'A')))

RB = Matrix(niso,1, lambda i,j:var('R_%s%s' % ('A'+chr(66+i),i+2)))

R = Matrix (niso, niso-1, lambda i,j: var('R_%s%d' % (chr(65+i) , j+2 )))

K= Matrix(niso-1,1, lambda i,j:var('K_%d' % (i+2) ) )

C= Matrix(niso-1,1, lambda i,j:var('A_%d' % i))

A = Matrix(niso-1,niso-1, lambda i,j:var('A_%d' % i))

b = Matrix(niso-1,1, lambda i,j:var('A_%d' % i))

for i in range(0,niso-1):

for j in range(0,niso-1):

A[i,j]=MM[j+1,0]*(Add(Mul(R[0,j],1/MA[i,0]/(RB[i,0]-R[0,i])))+R[i+1,j]/MX[i,0]/(-RB[i,0]+R[0,i]))

for i in range(0,niso-1):

b[i,0]=MM[0,0]*(Add(Mul(1,1/MA[i,0]/(RB[i,0]-R[0,i])))+1/MX[i,0]/(-RB[i,0]+R[0,i]))

A_in = Inverse(A)

if niso <= 4:

X =simplify(A_in*b)

if niso > 4:

X = A_in*b

pprint(X)

有没有办法加快速度？在