python - 在 Python 中对数组进行滚动求和

Question

我得到一个数组 X，我被要求返回另一个数组 Y，其中 y[t] = x[t] + x[t-1] 和 y[0] = 0 ，而不在 Python 中使用 for 循环。

我能想到的是使用滚动求和，但我不确定这是否是最快的方法，因为我需要将 x 转换为数据帧/序列才能滚动工作。有没有更快的方法来做到这一点？谢谢！

df = pd.DataFrame(X).rename(columns={0: 'X'})
df['Y'] = df['X'].rolling(2).sum().fillna(0)
Y = df['Y'].values

score 1 · Accepted Answer

如果您正在寻找单线可能无法提供最佳性能

from toolz.itertoolz import sliding_window
[0]+map(sum,list(sliding_window(2, [1, 2, 3, 4]))) # [0, 3, 5, 7]

score 1 · Accepted Answer

您可以使用列表理解和zip：

x = [1, 2, 3, 4]
y = [0] + [c + n for c, n in zip(x, x[1:])]
print(y)

输出

[0, 3, 5, 7]

这种方法依赖于内置函数，因此无需导入 pandas 等外部模块。

score 0 · Accepted Answer

好吧，您可以使用numpy，尽管这仍然会将列表转换为数组。

y = x + np.roll(x, 1)
y[0] = 0

这是快速、简短、相当透明的，并且不（明确）使用for循环。

您也可以使用map，这与列表推导几乎相同，并且不需要任何外部库。

y = 0 + map(lambda (a, b): a+b, zip(x, x[1:]))

在 Python3 中这不起作用，相反，您需要编写：

y = [0] + list(map(lambda a: a[0]+a[1], zip(x, x[1:])))

或者

y = [0] + list(map(sum, zip(x, x[1:])))

3 回答 3