numpy除法问题

Question

我正在尝试使用numpy除法对数组执行除法，我有两个数组，我将其称为：

log_norm_images = np.divide(diff_images, b_0)

我得到错误：

operands could not be broadcast together with shapes (96,96,55,64) (96,96,55).

这些分别是ndarray的形状。

现在，在我的python shell中，我进行了以下测试：

 x = np.random.rand((100, 100, 100))
 y = np.random.rand((100, 100))

和

np.divide(x, y)

运行没有任何错误。 我不确定为什么这样做，而不是我的情况。

Answer 1

您正在尝试将4-D阵列与3-D阵列一起广播。 根据NumPy的广播行为，只有对于每个对应的维度，维度相等或其中之一为1时，此操作才会成功。这就是它不匹配的原因：

Your 4-D array:  96 x 96 x 55 x 64
Your 3-D array:       96 x 96 x 55
                           ^     ^
                           Mismatching dimensions

如果填充/重塑3-D数组（假设不再是3-DI）以明确具有形状（96、96、55、1），则您的操作可能会起作用。 然后看起来像：

Your 4-D array:  96 x 96 x 55 x 64
Your 3-D array:  96 x 96 x 55 x 1
                                 ^
                                This is acceptable for the broadcast behavior

指向SciPy / NumPy文档的链接对此进行了更详细的介绍：

http://docs.scipy.org/doc/numpy/user/basics.broadcasting.html