您可以使用字典来按照一个列表对元素进行分组,并对另一个列表中的相应条目求和。以下是一个示例代码:
def group_and_sum(keys, values):
result = {}
for key, value in zip(keys, values):
if key in result:
result[key] += value
else:
result[key] = value
return result
keys = ['A', 'B', 'A', 'C', 'B']
values = [1, 2, 3, 4, 5]
result = group_and_sum(keys, values)
print(result)
运行以上代码会输出以下结果:
{'A': 4, 'B': 7, 'C': 4}
在这个例子中,keys
列表中的元素被用作字典的键,而values
列表中的相应元素被用来进行求和。如果键已经存在于字典中,则将对应的值加上新的值;否则,在字典中创建新的键值对。最后返回结果字典。