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]