Python|列表中的部分子集总和
编程领域的一些经典问题来自不同的类别,其中之一是求子集的总和。当我们需要累加总和并存储连续的组总和时,这个特殊问题也很常见。让我们在Python语言中尝试不同的方法来解决这个问题。
方法 #1:使用列表理解 + sum()
列表推导可用于执行此特定任务以过滤出连续的组,并且 sum函数可用于获得过滤后的解决方案的总和。
# Python3 code to demonstrate
# Sectional subset sum in list
# using list comprehension + sum()
# initializing list
test_list = [4, 7, 8, 10, 12, 15, 13, 17, 14]
# printing original list
print("The original list : " + str(test_list))
# using list comprehension + sum()
# Sectional subset sum in list
res = [ sum(test_list[x : x + 3])
for x in range(0, len(test_list), 3)]
# printing result
print("The grouped summation list is : " + str(res))
输出 :
The original list : [4, 7, 8, 10, 12, 15, 13, 17, 14]
The grouped summation list is : [19, 37, 44]
方法#2:使用sum() + itertools.islice()
将列表切片成块的任务在这里通过 islice 方法完成,而获得总和的常规任务是通过上述方法的 sum函数完成的。
# Python3 code to demonstrate
# Sectional subset sum in list
# using itertools.islice() + sum()
import itertools
# initializing list
test_list = [4, 7, 8, 10, 12, 15, 13, 17, 14]
# printing original list
print("The original list : " + str(test_list))
# using itertools.islice() + sum()
# Sectional subset sum in list
res = [sum(list(itertools.islice(test_list, i, i + 3)))
for i in range(0, len(test_list), 3)]
# printing result
print("The grouped summation list is : " + str(res))
输出 :
The original list : [4, 7, 8, 10, 12, 15, 13, 17, 14]
The grouped summation list is : [19, 37, 44]