如何通过特定元素获得两组元组的交互？

Question

有两组s1{(1, 'string111'), (2, 'string222')}和s2{(2, 'string333'), (3, 'string444')}。我可以通过id（元组中的第一个元素）获得这两个集合的交互吗。所以我实际上想要得到的交互是{1, 2} & {2, 3}，但是返回{2, 'string222'}。还是使用其他数据结构而不是元组集更好？你知道吗

Answer 1

为s1查找id2中的每个tuple在s2中的所有id上设置set

s1 = {(1, 'string111'), (2, 'string222')}
s2 = {(2, 'string333'), (3, 'string444')}

id2 = {x[0] for x in s2}        # all the id in s2
filtered = list(filter(lambda x: x[0] in id2, s1))  # lookup id2 and filter
print(filtered)                 # => [(2, 'string222')]

非FP版本

id2 = {x[0] for x in s2}
ret = set()
for x in s1:
    if x[0] in id2:
        ret.add(x)
print(ret)      # => {(2, 'string222')}