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]表示跳过最后一个元素。