在python中操作numpy数组的元素

Question

我有：

in = array([4, 3, 2, 1,
-3, -2, -1,  0])

如何从下一个元素中减去上一个元素，如下所示：

out = array([(3-4), (2-3), (1-2),
(-3-1), (-2-(-3)), (-1-(-2)), (0-1)]
out = array([-1,-1,-1,-4,1,1,-1])

Answer 1

您可以简单地使用np.diff():

import numpy as np
arr = np.array([4, 3, 2, 1, -3, -2, -1, 0])
print(np.diff(arr)) # [-1 -1 -1 -4  1  1  1]

np.diff()返回您刚才描述的numpy数组：两个相邻元素之间的差异。