Разница между массивами numpy разных размеров

Question

Скажем, у меня есть два двумерных массива A и B с формой: (10, 10) и (3, 3) соответственно.

Я хотел бы знать, есть ли способ вычисления A - B такогочто форма: (10, 10, 9) без использования петли.

т.е. разница каждого элемента A с каждым элементом из B.

Answer 1

Используйте внешнее вычитание, а затем измените форму -

np.subtract.outer(A,B).reshape((A.shape)+(-1,))

Или увеличьте A до 3D с единственным оттенком в качестве последнего и вычтите плоский коэффициент B -

A[...,None] - B.ravel() # or B.flat in place of B.ravel()