Python中保留嵌套列表内每个子集合指定数量元素的方法

首先，你需要明确每个子集合要保留的元素数量（比如示例中每个子集合都保留3个），可以通过定义数量列表对应每个子集合的保留数，再遍历处理每个集合。

实现代码 #

original_data = [{'512', '530', '788'}, {'112', '401', '276', '280', '787', '555'}, {'284', '041', 'V64.0', 'E8798', 'V15.8', '205'}]
# 每个子集合对应的保留元素数量，顺序与原列表一致
keep_counts = [3, 3, 3]

result = []
for s, count in zip(original_data, keep_counts):
    # 集合无序，转列表取前count个元素后转回集合
    result.append(set(list(s)[:count]))

print(result)

关键说明 #

• Python的集合是无序的，每次运行取到的元素可能和示例不完全一致。如果需要固定保留特定元素，得先给集合排序或明确筛选规则。
• 若要统一保留所有子集合的前N个元素，直接把keep_counts改成全为N的列表即可，或者不用列表，直接在循环中使用固定值。

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