下面是一个示例代码,演示了如何按照ID将多行转换为一行:
from collections import defaultdict
# 示例输入数据
data = [
{"id": 1, "value": "A"},
{"id": 1, "value": "B"},
{"id": 2, "value": "C"},
{"id": 2, "value": "D"},
{"id": 2, "value": "E"},
{"id": 3, "value": "F"}
]
# 使用defaultdict创建一个字典,key为ID,value为一个空列表
result = defaultdict(list)
# 遍历输入数据,并将每个元素的value添加到对应ID的列表中
for item in data:
result[item["id"]].append(item["value"])
# 将字典转换为列表,每个元素为一个字典,包含ID和对应的value列表
output = [{"id": key, "values": values} for key, values in result.items()]
# 输出结果
for item in output:
print(item)
运行以上代码,将得到以下输出:
{'id': 1, 'values': ['A', 'B']}
{'id': 2, 'values': ['C', 'D', 'E']}
{'id': 3, 'values': ['F']}
以上代码中,使用了defaultdict
来创建一个默认值为列表的字典,这样可以方便地将相同ID的value添加到对应的列表中。然后,使用列表推导式将字典转换为列表,每个元素为一个字典,包含ID和对应的value列表。最后,遍历输出结果,打印每个字典。