python - Numpy:矢量化滞后差异之和

Question

是否可以计算总和

在 numpy 中使用向量运算(y 是一个 Tx1 向量)，还是我必须用循环来做？谢谢。

Answer 1

你可以通过切片数组来做到这一点:

import numpy as np
y = np.array([1,2,3,4,5])
print np.sum(np.square(y[1:] - 2*y[:-1]))
# 14

其中y[1:]表示跳过第一个元素，y[:-1]表示跳过最后一个元素。