python热点数据排序

问题描述

遇到一个需求，关于数据排序，我们的数据有些是热点数据，希望排列在最前面，其他的数据要根据字段名进行大写字母排序

代码实现

# 示例数据
data = [
    {'name': 'apple', 'type': 'fruit'},
    {'name': 'banana', 'type': 'fruit'},
    {'name': 'carrot', 'type': 'vegetable'},
    {'name': 'date', 'type': 'fruit'},
    {'name': 'eggplant', 'type': 'vegetable'},
]

# 指定的排序顺序
custom_order = ['banana', 'apple']

# 自定义排序函数
def custom_sort(item):
    # 获取当前元素的名称
    name = item['name']
    # 如果名称在自定义排序中，返回它在自定义列表中的索引；否则返回一个较大的值以保证它在后面
    return (0 if name in custom_order else 1, 
            custom_order.index(name) if name in custom_order else name)

# 排序
sorted_data = sorted(data, key=custom_sort)

# 输出结果
for item in sorted_data:
    print(item)