如何判断值是否存在于指定集合中并直接获取对应的集合？

两种常用的简化实现方案，不需要写大量if/elif分支：

方法1：遍历匹配（适合少量查询、集合数量少的场景） #

把所有待检查的集合放入一个列表，用生成器表达式遍历匹配，直接返回第一个符合要求的集合：

value = 3
s1 = {1,2,3,4,5}
s2 = {6,7,8,9,0}
s3 = {11,12,13,14,15}

# 把所有待检查的可迭代对象（set/list/tuple都支持）存入列表
all_collections = [s1, s2, s3]
# 找到第一个包含目标值的集合，找不到则返回None
target_collection = next((col for col in all_collections if value in col), None)

如果目标值可能同时存在于多个集合中，把next换成列表推导即可拿到所有符合要求的集合：

target_collections = [col for col in all_collections if value in col]

方法2：预构建值映射表（适合高频查询场景） #

如果需要频繁执行这类查询，可以提前构建「值→所属集合」的映射表，后续查询时间复杂度为O(1)，效率更高：

value = 3
s1 = {1,2,3,4,5}
s2 = {6,7,8,9,0}
s3 = {11,12,13,14,15}
all_collections = [s1, s2, s3]

# 提前一次构建映射表
value_to_collection = {}
for col in all_collections:
    for item in col:
        # 若值可能属于多个集合，此处可改为value_to_collection.setdefault(item, []).append(col)
        value_to_collection[item] = col

# 后续直接查询即可
target_collection = value_to_collection.get(value)

两种方法对set/list/tuple类型都通用，不需要修改逻辑。

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