如何创建一个充满了另一个数组元素差异的numpy数组

Question

我有一个numpy数组X.我需要创建另一个相同大小的数组（比如Y），它有元素

Y[i] = X[i+1]-X[i-1]

我可以不循环使用数组元素吗？

Answer 1

您可以创建具有移位值的新数组，然后将它们相互减去。像这样：

import numpy as np

X  = np.arange(10)
X1 = np.roll(X,-1) #right shift
X2 = np.roll(X,1)  #left shift
Y  = X1 - X2