Python如何按指定顺序对列表元素进行分组排列

需求为按照groups列表定义的固定顺序，对items列表的元素做分组重排，同组元素保留原列表中的出现相对位置；需要规避遍历原列表时直接增删元素的不安全写法，同时优化现有两层循环实现的性能。
现有示例代码通过两层循环实现需求，时间复杂度为O(len(groups) * len(items))，当两个列表长度较大时执行效率偏低。

1. 单次遍历分桶法（推荐通用场景） #

全程不修改原items列表，仅做一次遍历即可完成元素分组，时间复杂度为O(len(groups) + len(items))，性能远高于双层循环，同时支持保留原始元素引用（适配元素为自定义对象的场景）。

groups = ["d", "c", "a", "b"]
items = ["a", "b", "b", "c", "d", "a"]

# 预创建每个分组对应的存储桶
group_bucket = {g: [] for g in groups}
# 单次遍历原列表完成元素分桶
for item in items:
    if item in group_bucket:
        group_bucket[item].append(item)
# 按groups定义的顺序拼接所有桶的内容
result = []
for g in groups:
    result.extend(group_bucket[g])

print(result)
# 输出：['d', 'c', 'a', 'a', 'b', 'b']

如果items中存在未在groups里定义的元素，可以单独新增一个桶收纳这类元素，按需决定是丢弃还是追加到结果末尾。

2. 计数统计法（高性能场景） #

如果元素为可哈希的简单值（字符串、数字等），可以用计数统计的方式进一步减少内存占用，执行效率最高：

from collections import Counter

groups = ["d", "c", "a", "b"]
items = ["a", "b", "b", "c", "d", "a"]

item_count = Counter(items)
result = []
for g in groups:
    result.extend([g] * item_count.get(g, 0))

print(result)
# 输出：['d', 'c', 'a', 'a', 'b', 'b']

3. 自定义key排序法（简洁场景） #

如果追求代码简洁，且列表长度在千级以内，可以直接利用Python稳定排序的特性，通过自定义优先级key实现需求，代码可读性最高：

groups = ["d", "c", "a", "b"]
items = ["a", "b", "b", "c", "d", "a"]

# 构建分组值到排序优先级的映射，优先级值越小排序越靠前
priority_map = {g: idx for idx, g in enumerate(groups)}
# 未定义在groups中的元素会统一排到结果末尾
result = sorted(items, key=lambda x: priority_map.get(x, len(groups)))

print(result)
# 输出：['d', 'c', 'a', 'a', 'b', 'b']

• 通用场景优先选分桶法，无额外依赖，逻辑清晰，性能足够，适配绝大多数业务场景
• 处理超大规模简单值列表时选计数统计法，性能最优
• 短列表快速实现选自定义key排序法，代码量最少，维护成本低

内容的提问来源于stack exchange，提问作者hebat