将 Matlab 代码转换为 Python 不可广播输出的问题

Question

我在 Matlab 中有以下代码

a= a + b(c,:);

其中“a”是 4524x3 矩阵，“b”是 1131x3，“c”是 4524x1。

在 Python 中我有

a[:]+= b[c, :]

我得到 'a' 作为 4524x4524x3 矩阵。为什么 Python 会创建这个额外的维度而不是求和？

Answer 1

尝试这样做：

a[:] += b[c.ravel(), :]

发生的事情是c 被视为二维矩阵而不是单个一维数组，这就是发生不必要广播的原因。当您需要它是 1D 时，您基本上是在尝试用 2D 数组索引矩阵。