从另一个矩阵的第ith个元素减去一个矩阵的第n个元素

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])