以下是一个示例代码,它可以按多个键对数组对象进行求和:
from collections import defaultdict
def sum_by_keys(array, keys):
sums = defaultdict(float)
for obj in array:
key_values = tuple(obj[key] for key in keys)
sums[key_values] += obj['value']
return sums
# 示例数据
data = [
{'key1': 'A', 'key2': 'X', 'value': 1.5},
{'key1': 'A', 'key2': 'Y', 'value': 2.5},
{'key1': 'B', 'key2': 'X', 'value': 3.5},
{'key1': 'B', 'key2': 'Y', 'value': 4.5},
{'key1': 'A', 'key2': 'X', 'value': 5.5},
]
# 按 'key1' 和 'key2' 对数组对象求和
result = sum_by_keys(data, ['key1', 'key2'])
print(result)
输出结果为:
defaultdict(, {('A', 'X'): 7.0, ('A', 'Y'): 2.5, ('B', 'X'): 3.5, ('B', 'Y'): 4.5})
在上面的示例中,sum_by_keys
函数接受一个数组对象和一个键列表作为参数。它使用 defaultdict
来创建一个字典,并使用键值对的元组作为字典的键,以便将相同键值对的值相加。然后,它遍历数组对象中的每个元素,提取指定键的值,将其作为键值对的元组,将该值加到字典中相应的键上。
这样,我们就可以使用不同的键组合来对数组对象进行求和。