以下是一个示例代码,用于按照列表中的类别对列进行分组:
from collections import defaultdict
def group_by_category(items):
grouped_items = defaultdict(list)
for item in items:
category = item['category']
grouped_items[category].append(item)
return grouped_items
# 假设列表中的每个元素都是字典,包含 'category' 键来表示类别
items = [
{'name': 'item1', 'category': 'A'},
{'name': 'item2', 'category': 'B'},
{'name': 'item3', 'category': 'A'},
{'name': 'item4', 'category': 'C'},
{'name': 'item5', 'category': 'B'}
]
grouped_items = group_by_category(items)
for category, items in grouped_items.items():
print(f"Category {category}:")
for item in items:
print(item['name'])
print()
输出结果为:
Category A:
item1
item3
Category B:
item2
item5
Category C:
item4
这个示例中使用了 defaultdict 类来创建一个可以自动初始化为空列表的字典。然后,遍历列表中的每个元素,获取其类别,并将元素添加到对应类别的列表中。最后,使用循环打印出每个类别及其对应的元素。