如何在 numpy 中將 [a,b,c] 轉換為 [a+b+c, b+c, c]？

Question

如果我有一個數組 [a,b,c]，如何以最有效的方式將其轉換為 [a+b+c, b+c, c]？

(a, b, c) 是浮點數。

謝謝！

Answer 1

您可以使用np.cumsum ：

import numpy as np

a = np.array([1, 2, 3])
print(np.cumsum(a[::-1])[::-1])
# Outputs [6 5 3]

Answer 2

在反轉數組上使用numpy.cumsum() ，然后再次反轉它。 cumsum([a,b,c])返回[a,a+b,a+b+c] 。

import numpy as np

x = [3,7,14]

y = np.cumsum(x[::-1])[::-1]

print(y)