<p>根据文档示例,只需执行以下操作:</p>
<pre><code>from sympy import *
import math
n=3
n_atoms=8
MainMatrix=Matrix([[0 for x in range(n)] for y in range(n)])
KappaMatrix=Matrix([0 for x in range(n-1)])
MassMatrix=Matrix([0 for x in range(n)])
Kappa=3
ka_by_pi=4
for i in range(1,n):
KappaMatrix[i-1]=3
for i in range(1,n+1):
MassMatrix[i-1]=3
for i in range(2,n):
MainMatrix[i-1]= -KappaMatrix[i-2]/MassMatrix[i-1]
MainMatrix[i-1]= (KappaMatrix[i-2] + KappaMatrix[i-1])/MassMatrix[i-1]
MainMatrix[i-1]= -KappaMatrix[i-1]/MassMatrix[i-1]
MainMatrix[0]=(KappaMatrix[0]+Kappa)/MassMatrix[0]
MainMatrix[0]=-(KappaMatrix[0])/MassMatrix[0]
i=symbols('i')
MainMatrix[0]= -(Kappa/MassMatrix[0])*(math.cos(math.pi*ka_by_pi)+(i*math.sin(math.pi*ka_by_pi)))
MainMatrix[n-1]= -(Kappa/MassMatrix[n-1])*(math.cos(math.pi*ka_by_pi)-(i*math.sin(math.pi*ka_by_pi)))
MainMatrix[n-1]=-(KappaMatrix[n-2])/MassMatrix[n-1]
MainMatrix[n-1]=(KappaMatrix[n-2]+Kappa)/MassMatrix[n-1]
t=symbols('t')
for j in range(0,n_atoms):
MainMatrix[j]=MainMatrix[j]-(t**2)
print(MainMatrix)
</code></pre>