Vectorize numpy 1D array 函数减去下一个元素并添加前一个元素？

Question

我有一个简单的一维 numpy 数组，如下所示：

np.array([2,1,3,5,1])

如何对一个函数进行矢量化，该函数对于每个元素减去下一个元素并添加上一个元素？在上述情况下，我希望得到一些回报：

[?, 1-3+2, 3-5+1, 5-1+3, ?]
=[?, 0, -1, 7, ?]

，在哪里？意味着我不关心元素的价值。此操作将在具有 100 万个元素（双精度类型）的一维 np.array 中运行。

Answer 1

正如评论中所建议的，切片将满足您的需求：

a=np.array([2,1,3,5,1])
print(a[1:-1]+a[:-2]-a[2:])

结果将是包含您关心的值的数组 [0,-1,7]。