模块，Mathematica

Question

我有以下问题:我想把一个较小的矩阵M加到一个较大的矩阵N上，从i开始，j在N中。

代码如下：

PutMintoN[M_, Q_, i_, j_] := Module[{Mrow, Mcol},
   {Mrow, Mcol} = Dimensions[M];
   For[k = 1, k <= Mrow, k++,
    For[q = 1, q <= Mcol, q++,
     Q[[i + k - 1, j + q - 1]] = 
      Q[[i + k - 1, j + q - 1]] + M[[k, q]]]];
   Q
   ];   

问题似乎不在算法中，而是在模块中，因为如果我将内部代码复制到外部，它就会工作。

提前谢谢。

Answer 1

太好了，你自己找到了错误。

为了同样的目的，我制作了一个更优雅和健壮的模块，通过使用ArrayPad，使M与N具有相同的维度，而不仅仅是将M添加到N。它甚至可以工作，如果i或j超出N的维度，这对您的原始模块来说是一个问题。

putMintoN[M_, N_, i_, j_] := Module[{Mrow, Mcol, Nrow, Ncol, mn},
  {Mrow, Mcol} = Dimensions[M]; {Nrow, Ncol} = Dimensions[N];
  mn = {{Min[i - 1, Nrow], Min[(Nrow - Mrow) - i + 1, Nrow]}, 
        {Min[j - 1, Ncol], Min[(Ncol - Mcol) - j + 1, Ncol]}};
  ArrayPad[M, mn] + N]

测试：

IN: putMintoN[{{x, y, p}, {z, w, q}}, {{a, b, c}, {d, e, f}, {g, h, i}, {j, k, l}}, 2, 1]
OUT: {{a, b, c}, {d + x, e + y, f + p}, {g + z, h + w, i + q}, {j, k, l}}

希望这篇文章能激励你。诚挚的问候。