有没有办法在Python中从右到左累加？

问题描述

Python 已经有一个 itertools.accumulate 函数。不过这个函数是从左往右累加的：

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

相反，我想从右到左累积，如下所示：

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

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

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

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

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

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

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

解决方法

您只需要对 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]]
[['a','c'],['b',['c']]

,

用itertools累加，结果用[::-1]来反转不是很容易吗？

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