Python有没有办法从右往左累加？

Question

Python 已经有一个 itertools.accumulate 函数。但是，这个函数从左到右累加：

>>> list(itertools.accumulate([[x] for x in range(5)]))
[[0], [0, 1], [0, 1, 2], [0, 1, 2, 3], [0, 1, 2, 3, 4]]

相反，我想从右到左累积，像这样：

>>> list(accumulate_from_right([[x] for x in range(5)]))
[[0, 1, 2, 3, 4], [1, 2, 3, 4], [2, 3, 4], [3, 4], [4]]

我目前的解决方案（仅适用于列表）非常低效且丑陋：

>>> list(x[::-1] for x in accumulate([y] for y in reversed(range(5))))[::-1]
[[0, 1, 2, 3, 4], [1, 2, 3, 4], [2, 3, 4], [3, 4], [4]]

从右到左累积列表的最佳方法是什么？

编辑：我使用范围作为示例。我希望能够将此方法应用于任何嵌套列表。

这是另一个不使用范围的示例：

>>> list(accumulate_from_right(['a', 'b', 'c']))
['abc', 'bc', 'c']

Answer 1

您只需要对 accumulate_from_right:

的输入进行一次传递

def accumulate_from_right(vals):
   return [vals[i:] for i in range(len(vals))]

print(accumulate_from_right(list(range(5))))
print(accumulate_from_right(['a', 'b', 'c']))

输出：

[[0, 1, 2, 3, 4], [1, 2, 3, 4], [2, 3, 4], [3, 4], [4]]
[['a', 'b', 'c'], ['b', 'c'], ['c']]

Answer 2

你可以试试

[list(range(i, 5)) for i in range(5)]

returns

[[0, 1, 2, 3, 4], [1, 2, 3, 4], [2, 3, 4], [3, 4], [4]]

Answer 3

用itertools累加，在结果处用[::-1]反转是不是很简单？

list(itertools.accumulate([[x] for x in range(5)]))[::-1]

Python有没有办法从右往左累加？

Is there a way to accumulate from right to left in Python?

python

python-3.x

itertools

accumulate