np.subtract 沿轴

Question

有没有一种简单的方法可以沿二维数组的轴进行减法？

a=np.array([[610,550],[400,360]])
np.sum(a,axis=0) # Results in  array([1010, 910])

np.subtract(a[0],a[1]) # Results in array([210, 190])
np.subtract(a,axis=0) # Errors out obviously

Answer 1

np.subtract 减去两个数组的对应元素。沿轴减去两个数组是没有意义的，因为它发生在每个元素上，无论其相对于其邻居的位置如何。请注意，在您的示例中，np.sum 只接受一个数组，而不是两个。

要查找单个数组的连续元素之间的差异，请使用 np.diff，它完全按照您的预期接受 axis 参数

np.diff(a, axis=0)

您可以通过在沿所需轴切成1: 和:-1 的数组上调用减法来手动实现diff。在您的示例中：

a[1:] - a[:-1]