查找值属于多个集合中的哪个

Question

我有几组值，需要检查给定值位于其中哪些值中，并返回该组的名称。

value = 'a'
set_1 = {'a', 'b', 'c'}
set_2 = {'d', 'e', 'f'}
set_3 = {'g', 'h', 'i'}
set_4 = {'a', 'e', 'i'}

我想检查值是否存在于集合1-3中，而方法中不包括set_4，并返回集合名称。 所以像这样：

find_set(value in set_1, set_2, set_3)

应该回来

set_1

也许一些整齐的lambda函数？ 我试过了

w = next(n for n,v in filter(lambda t: isinstance(t[1],set), globals().items()) if value in v)

从Find中查找值是否存在于多个列表中，但是该方法将检查所有本地/全局集。 这在这里行不通，因为值可以存在于其中几个中。 我需要能够指定要查看的集合。

Answer 1

不要使用会在globals中挖掘的丑陋的hacklamdmbda，这样您就可以获得名称。 几个星期后，包括您自己在内的任何人都可能会读懂您的代码：-)。

您希望能够为已定义的集合命名，这就是为什么我们有字典的原因。 用您的集合制作字典，然后您可以创建方便/可读的集合/列表理解，以紧凑的可读方式获得所需内容：

>>> d = {'set_1': set_1, 'set_2': set_2, 'set_3': set_3, 'set_4': set_4}

要捕获'a'所在的所有集合：

>>> {name for name, items in d.items() if 'a' in items}
{'set_1', 'set_4'}

要排除某些名称，请在if添加另一个required子句以进行过滤：

>>> {name for name, items in d.items() if 'a' in items and name != 'set_4'}
{'set_1'}

当然，您可以将其分解为一个函数，并且很高兴将来遇到它时能够理解它：

def find_sets(val, *excludes, d=d):
    return {n for n, i in d.items() if val in i and n not in excludes}

这与以前的行为类似。 d=d可能不是您想要的方式，为此使用一些**d语法可能会更好。

---

如果只想获取第一个值，请从函数中返回next(comprehension) ，如下所示：

def find_sets(val, *excludes, d=d):
    return next((n for n, i in d.items() if val in i and n not in excludes), '')

''表示如果未实际找到任何元素，则返回默认值，即当使用不存在的值调用时，将返回一个空字符串（可根据您的喜好进行更改）：

>>> find_sets('1')
''