Kronecker产品在Python和Matlab中

时间:2013-06-11 02:45:00

标签: python matlab numpy scipy

我试图用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

为什么我会得到不同的结果?

谢谢!

1 个答案:

答案 0 :(得分:4)

使用sparse.kron作为稀疏矩阵的Kronecker积。

numpy.kron不处理稀疏矩阵。当给定稀疏矩阵时,它可能不会产生错误,但它返回的值将不正确。