如何在python中添加2个多维矩阵的相应元素?

时间:2016-08-26 03:58:26

标签: python numpy image-processing

我有2个多维数组,大小都是128X640X5。 5是矩阵的通道数。我希望为矩阵中的每个点添加两个矩阵的相应通道值。例如,如果我们将A和B作为2个矩阵,我希望进行如下操作: A(x,y,0)+ B(x,y,0)= A(x,y,0)。这应该在A和B中添加点x和y的第0个通道值,然后将其存储回A. Similiary我也希望为其他4个通道执行此操作。有关如何在python中执行此操作的任何想法?我正在使用numpy数组的python,主要处理图像处理问题。

1 个答案:

答案 0 :(得分:0)

为了在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