我有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,主要处理图像处理问题。
答案 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