假设有以下字符串数组:
data = [['key1', 'key2', 1], ['key2', 'key1', 2], ['key1', 'key3', 3], ['key3', 'key1', 4], ['key3', 'key2', 5]]
要求按照key1和key2进行分组,并累加第三个元素的值,可以使用Python中的dict类型来实现。
from collections import defaultdict
result = defaultdict(int)
# 遍历数据
for record in data:
key1, key2, value = record
# 构造字典的键
key = (key1, key2)
# 累加值
result[key] += value
# 输出结果
for (key1, key2), value in result.items():
print(key1, key2, value)
输出结果:
key1 key2 1
key2 key1 2
key1 key3 3
key3 key1 4
key3 key2 5
上一篇:按两个键对结构体数组进行排序