Python中有没有一种从右到左的累积方法



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']

您只需要对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']]

您可以尝试类似的东西

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

返回

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

使用itertools累加并在结果处使用[::-1]进行反转不是很容易吗?

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

最新更新