python生成三对角矩阵_使用Python的三对角矩阵

我看了过去关于三对角的问题，但似乎没有人遇到我遇到的问题。我试图用，非均匀泊松方程的三对角刚度矩阵scipy.sparse.spdiags但似乎并没有因此收到矩阵。在def Poisson_Stiffness(x0):

N = len(x0) - 1 #THE AMOUNT OF ELEMENTS (NOT THE AMOUNT OF POINTS) x0, x1, ... , x_N

h = np.zeros(N)

a = np.zeros(N+1)

b = np.zeros(N)

for i in range(N):

h[i] = x0[i+1] - x0[i] #Length of each nonuniform element

a[0] = 1/h[0]

for i in range(1,N):

a[i] = 1/h[i] + 1/h[i-1] #Main Diagonal of stiffness matrix

a[N] = 1/h[N-1]

for i in range(N):

b[i] = -1/h[i] #Upper and lower diagonal of stiffness matrix.

Tridiagonal_Data = np.array([[a],[b],[b]])

Positions = [0, 1, -1]

Stiffness_Matrix = scipy.sparse.spdiags(Tridiagonal_Data,Positions,N+1,N+1)

print Stiffness_Matrix

因此，当x0=[0,0.3,0.4,0.7,1]时，得到的刚度矩阵