如何在数组的一部分与数组的一部分之间进行操作？

Question

import numpy as np
a=np.array([2,3,4,6,7,4,5,3,2,1,9,8,7,6,4,2])
## a has 16 elements
b=np.array([1,5,3,7])
""" i want this:
 a[:4] - b[1], a[4:8]-b[2], a[8:12]-b[3], a[12:16]-b[4] in one array
and it should look like
c=numpy.array[1,2,3,5,2,-1,0,-2....]

这只是我的问题的一个简单示例。所以不能使用组来完成。我需要一个循环来解决这个问题。

Answer 1

您可以重塑您的数组a，对其进行操作，然后将其展平。

In [24]: import numpy as np

In [25]: a=np.array([2,3,4,6,7,4,5,3,2,1,9,8,7,6,4,2])

In [26]: b=np.array([1,5,3,7])

In [27]: (a.reshape(4,4).T - b).T.ravel()
Out[27]: array([ 1,  2,  3,  5,  2, -1,  0, -2, -1, -2,  6,  5,  0, -1, -3, -5])