Python|自定义索引范围总和
开发和有时机器学习应用程序需要以自定义方式将列表拆分为较小的列表,即在必须执行拆分然后求和的某些值上。这是一个非常有用的实用程序,值得了解。让我们讨论可以执行此任务的某些方式。
方法 #1:使用列表理解 + zip() + sum()
通过结合列表理解和 zip() 的强大功能,可以完成此任务。在此我们压缩列表的开头和结尾,然后在列表到达时继续对其进行切片并从中删除新列表。求和的任务是使用 sum() 执行的。
# Python3 code to demonstrate
# Custom Index Range Summation
# using list comprehension + zip() + sum()
# initializing string
test_list = [1, 4, 5, 6, 7, 3, 5, 9, 2, 4]
# initializing split index list
split_list = [2, 5, 7]
# printing original list
print ("The original list is : " + str(test_list))
# printing original split index list
print ("The original split index list : " + str(split_list))
# using list comprehension + zip() + sum()
# Custom Index Range Summation
res = [sum(test_list[i : j]) for i, j in zip([0] +
split_list, split_list + [None])]
# printing result
print ("The splitted lists summation are : " + str(res))
输出 :
The original list is : [1, 4, 5, 6, 7, 3, 5, 9, 2, 4]
The original split index list : [2, 5, 7]
The splitted lists summation are : [5, 18, 8, 15]
方法#2:使用itertools.chain() + zip() + sum()
列表理解函数执行的获取拆分块的任务也可以使用链函数完成。当我们希望处理更大的列表时,这更有用,因为这种方法更有效。求和的任务是使用 sum() 执行的。
# Python3 code to demonstrate
# Custom Index Range Summation
# using itertools.chain() + zip()
from itertools import chain
# initializing string
test_list = [1, 4, 5, 6, 7, 3, 5, 9, 2, 4]
# initializing split index list
split_list = [2, 5, 7]
# printing original list
print ("The original list is : " + str(test_list))
# printing original split index list
print ("The original split index list : " + str(split_list))
# using itertools.chain() + zip()
# Custom Index Range Summation
temp = zip(chain([0], split_list), chain(split_list, [None]))
res = list(sum(test_list[i : j]) for i, j in temp)
# printing result
print ("The splitted lists summations are : " + str(res))
输出 :
The original list is : [1, 4, 5, 6, 7, 3, 5, 9, 2, 4]
The original split index list : [2, 5, 7]
The splitted lists summation are : [5, 18, 8, 15]