numpy数组中的算术

时间:2014-12-21 09:39:15

标签: numpy

我已经阅读了3个相同形状(500 x 500)的光栅图像作为numpy数组,并以这种方式放置它们:

rasters = np.array(A,B,C)

其中A,B,C是属于每个图像的2d numpy数组。

现在我必须计算以下内容:

result1 = B-A
result2 = C-B

然后,

final_result = np.max([result1,result2],axis = 0)

final_result应具有相同的A或B或C形状(即500乘500)

我该怎么做?

2 个答案:

答案 0 :(得分:3)

您可以使用np.diffnp.max

np.max(np.diff(rasters, axis=0), axis=0)

可替换地:

np.max(rasters[1:] - rasters[:-1], axis=0)

答案 1 :(得分:1)

B-A是使用np.subtract(B,A)完成的。