Арифметика в пределах Numpy массивов

Question

Я прочитал 3 растровых изображений одинаковой формы (500 на 500) в качестве Numpy массива, и поместить их таким образом:

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

где А, В, С являются 2d Numpy массивы принадлежащих каждому изображению.

Теперь я должен вычислить следующее:

result1 = B-A 
result2 = C-B 

Затем

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

final_result должны иметь такую ​​же форму А или В или С (т.е., 500 на 500)

Как это сделать?

Answer 1

Вы можете использовать np.diff и np.max:

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

В качестве альтернативы:

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

Answer 2

B-A выполняется с использованием np.subtract(B,A).