Python-前缀总和列表

列表是有序且可更改的集合。在Python中,列表用方括号括起来。您通过引用索引号访问列表项。负索引表示从末尾开始,-1表示最后一项。您可以通过指定范围的起点和终点来指定索引范围。指定范围时,返回值将是包含指定项目的新列表。

示例

# using list comprehension + sum() + list slicing
# initializing list
test_list = [3, 4, 1, 7, 9, 1]
# printing original list
print("The original list : " + str(test_list))
# using list comprehension + sum() + list slicing
# prefix sum list
res = [sum(test_list[ : i + 1]) for i in range(len(test_list))]
# print result
print("The prefix sum list is : " + str(res))

输出结果

The original list : [3, 4, 1, 7, 9, 1]
The prefix sum list is : [3, 7, 8, 15, 24, 25]