从下一个元素中减去数组的元素

Question

我有一个 numpy 数组我想在修复第一个数字后从下一个数字中减去前一个数字并想用零替换最后一个数字

a=np.array([10,20,22,44])

预计 output

np.array([10,10,2,0])

我试过np.diff function 但它错过了第一个数字。希望专家能提出更好的解决方案。

Answer 1

您想要自定义 output，因此使用自定义串联：

out = np.r_[a[0], np.diff(a[:-1]), 0]

Output: array([10, 10, 2, 0])