如何在python中添加2个多维矩阵的对应元素？答案

【问题标题】：How to add corresponding elements of 2 multidimensional matrices in python?如何在python中添加2个多维矩阵的对应元素？
【发布时间】：2016-08-26 03:58:26
【问题描述】：

我有 2 个多维数组，大小均为 128X640X5。 5 是矩阵的通道数。我希望为矩阵中的每个点添加两个矩阵的各自通道值。例如，如果我们将 A 和 B 作为 2 个矩阵，我希望执行如下操作： A(x,y,0)+B(x,y,0) =A(x,y,0)。这应该在 A 和 B 中添加点 x 和 y 的第 0 个通道值，然后将其存储回 A。类似地，我也希望对其他 4 个通道执行此操作。关于如何在 python 中执行此操作的任何想法？我正在使用 python 的 numpy 数组，基本上是在处理图像处理问题。

【问题讨论】：

请告诉我们您的search and research。你试过A = A + B吗？
我在尺寸重塑方面遇到了一些问题。但它现在得到了修复。感谢您的回复。

标签： python numpy image-processing

【解决方案1】：

为了在 numpy 中添加 ndarray 的每个对应点，您可以使用 numpy 的 add 函数 (numpy.add)。

它将两个具有相同形状的ndarray的每个对应点相加。如果数组没有相同的形状，它会broadcast（改变它们的形状）以便它们可以相加。

在你的情况下，你只需写：

import numpy as np
C = np.add(A, B)

来源：
http://docs.scipy.org/doc/numpy/reference/generated/numpy.add.html

【讨论】：