NumPy - 我将如何向量化这个操作？

Question

我需要获取一个维度为(M, 4) 的数组并减去元素以返回一个(M, 2) 的数组。

例如：

[[1, 2, 3, 4],
 [5, 6, 7, 8]]

# to

[[2, 2],
 [2, 2]]

我尝试过的是：

subs = a[..., 2] - a[..., 0], a[..., 3] - a[..., 1]

但这以 2 个维度为 (M,) 的数组的方式返回。

Answer 1

您可以创建数组的视图，然后减去两个视图，例如：

a<b>[:,2:]</b> - a<b>[:,:2]</b>

例如对于一些样本输入：

>>> a
array([[5, 2, 9, 2],
       [9, 9, 7, 9]])
>>> a[:,2:] - a[:,:2]
array([[ 4,  0],
       [-2,  0]])