忽略元组比较中的值但仍检索它

Question

我正在使用一个包含 (position, name) 形式的元组的集合，并且需要检查名称的集合中是否已经存在一个值，同时忽略位置。

有没有一种方法可以使用类似于value in my_set 的in 运算符，在比较期间忽略元组中的位置变量，但仍然检索它？类似于 (_, value) in my_set 或 (*, value) in my_set) 的东西，但它们不起作用，第一个返回不正确的值，第二个引发 SyntaxError。

显然，我可以使用循环或生成器理解，如value in (tup[1] for tup in my_set)，但这不会从该元组中检索位置变量，我很好奇是否有某种形式的单行理解可以做到这一点。

Answer 1

您可以使用现有数据结构（迭代集合）在 O(n) 中执行此操作，但对于 O(1)，您必须更改数据结构。您需要进行查找：

from collections import defaultdict

positions = defaultdict(list)
for position, name in my_set:
    positions[name].append(position)

现在这是一个 O(1) 操作：

name in positions

按名称检索所有内容：

for pos in positions[name]:
    ...

如果您希望它与my_set 突变保持同步，那么您需要在添加/删除my_set 的同时添加用于更新positions 的挂钩。完全重新考虑底层数据结构可能会更好，例如，首先使用字典而不是集合。