如何根据元素的属性将python列表拆分成片段？

# 假设有一个包含字典元素的列表，每个字典都有一个属性'category'
elements = [
    {'name': 'apple', 'category': 'fruit'},
    {'name': 'carrot', 'category': 'vegetable'},
    {'name': 'banana', 'category': 'fruit'},
    {'name': 'broccoli', 'category': 'vegetable'},
    {'name': 'orange', 'category': 'fruit'}
]

# 根据'category'属性将列表拆分成片段
fruit_elements = [element for element in elements if element['category'] == 'fruit']
vegetable_elements = [element for element in elements if element['category'] == 'vegetable']

# 打印拆分后的结果
print('水果元素：', fruit_elements)
print('蔬菜元素：', vegetable_elements)

水果元素： [{'name': 'apple', 'category': 'fruit'}, {'name': 'banana', 'category': 'fruit'}, {'name': 'orange', 'category': 'fruit'}]
蔬菜元素： [{'name': 'carrot', 'category': 'vegetable'}, {'name': 'broccoli', 'category': 'vegetable'}]