Python实现列表重复元素出现次数上限控制的函数需求

需求清晰：写一个函数，接收初始列表和重复次数上限n（默认值为2），返回新列表，其中每个元素的出现次数不超过n次，输出顺序无关紧要。

实现方案1：保留原元素顺序 #

该方法遍历原列表，仅保留每个元素的前n次出现，完全匹配示例中的输出顺序：

initial_list = [1,2,1,1,'a','b','a', 'b','b']

def cap(initial_list, n=2):
    count = {}
    result = []
    for item in initial_list:
        current_count = count.get(item, 0)
        if current_count < n:
            result.append(item)
            count[item] = current_count + 1
    return result

# 测试调用
print(cap(initial_list))  # 输出: [1, 2, 1, 'a', 'b', 'a', 'b']

实现方案2：快速生成（顺序无关） #

如果不介意输出顺序，可以先用计数器统计所有元素的出现次数，再按上限生成结果，代码更简洁：

from collections import Counter

initial_list = [1,2,1,1,'a','b','a', 'b','b']

def cap(initial_list, n=2):
    element_counts = Counter(initial_list)
    result = []
    for elem, cnt in element_counts.items():
        result.extend([elem] * min(cnt, n))
    return result

# 测试调用
print(cap(initial_list))  # 输出示例: [1, 1, 2, 'a', 'a', 'b', 'b']

两种方案均满足需求，可根据对输出顺序的要求选择使用。

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