我试图用MATLAB重现Python中的结果。但是,我似乎无法做到正确。这是正确的MATLAB代码:
nx = 5;
ny = 7;
x = linspace(0, 1, nx); dx = x(2) - x(1);
y = linspace(0, 1, ny); dy = y(2) - y(1);
onex = ones(nx, 1);
oney = ones(ny, 1);
Dx = spdiags([onex -2*onex onex], [-1 0 1], nx, nx);
Dy = spdiags([oney -2*oney oney], [-1 0 1], ny, ny);
Ix = eye(nx); Iy = eye(ny);
L = kron(Iy, Dx);
size(L) % 35 35
现在,这是Python代码:
nx = 5
ny = 7
x = linspace(0, 1, nx); dx = x[1] - x[0]
y = linspace(0, 1, ny); dy = y[1] - y[0]
onex = ones(nx)
oney = ones(ny)
Dx = sparse.dia_matrix( ([onex, -2*onex, onex], [-1,0,1] ), shape=(nx,nx))
Dy = sparse.dia_matrix( ([oney, -2*oney, oney], [-1,0,1] ), shape=(ny,ny))
Ix = eye(nx)
Iy = eye(ny)
L = kron(Iy, Dx)
L.shape # (7, 7)
到目前为止,我已经能够验证,一切都是正确的,直到L的定义。根据MATLAB kron(Iy, Dx)
(应该是kronecker产品)应该产生35X35矩阵,但Python认为应该是一个7X7矩阵。在更简单的计算中,两者都给出了正确的答案:
的Python:
kron(array(([1,2],[2,3])), [1,2])
array([[1, 2, 2, 4],
[2, 4, 3, 6]])
MATLAB
kron([1 2; 2 3], [1 2])
ans = 1 2 2 4
2 4 3 6
为什么我会得到不同的结果?
谢谢!
答案 0 :(得分:4)
使用sparse.kron
作为稀疏矩阵的Kronecker积。
numpy.kron
不处理稀疏矩阵。当给定稀疏矩阵时,它可能不会产生错误,但它返回的值将不正确。