2

我有以下问题：我想将一个较小的矩阵 M 与一个较大的矩阵 N 相加，从 N 中的 i,j 开始。

这是代码：

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
   ];

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

提前致谢。

4

1 回答 1

0

太好了，您自己发现了错误。

为了同样的目的，我制作了一个更优雅、更健壮的模块，使用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}}

在 Mathematica 中，通常可以避免使用 for-lopes、Listable 函数、map、apply 等。

希望这能启发你。最好的祝福。

于 2014-12-08T03:09:45.323 回答