Вычитание каждого n-го элемента матрицы из каждого i-го элемента из другой матрицы

Question

У меня есть две матрицы

A = [a,b,c,d,e,f,g,h,i,j,k,l]
B = [w,x,y,z]

Я хочу вычесть как [a,e,i] из w и [b,,f,j] из x таким образом. Все эти буквы являются поплавками. Результаты будут в массиве.

Не могли бы вы помочь как вычесть каждый n -й элемент матрицы из каждого i -ого элемента из другой матрицы?

Answer 1

Вы можете сначала изменить свой массив в матрицу 4 × 3, затем вычесть его, а затем изменить его обратно. Например:

(A.reshape(-1, 4) - B).flatten()

Например:

>>> A
array([8, 7, 1, 4, 6, 2, 5, 4, 6, 8, 8, 1])
>>> B
array([7, 4, 2, 1])
>>> (A.reshape(-1,4) - B).flatten()
array([ 1,  3, -1,  3, -1, -2,  3,  3, -1,  4,  6,  0])