在元组列表中查找重叠元素?

时间:2018-05-09 14:29:57

标签: python list for-loop tuples intersection

根据我对交叉函数的理解,它找到列表中元素之间的完全重叠。例如:

tup_1 = [(1,2,3),(4,5,6)]
tup_2 = [(4,5,6)]
ol_tup = set(tup_1).intersection(tup_2)
print ol_tup

会产生:

set([(4, 5, 6)])

但是,假设我的元组列表设置如下:

tup_1 = [(1,2,3),(4,5,5)]
tup_2 = [(4,5,6)]

tup_1中第二元组的2个元素与tup_2中的第1个元组重叠。如果我想让python返回这两个元组:(4,5,5)和(4,5,6),有没有比嵌套for循环更简单的方法(下面)?

for single_tuple_1 in tup_1:
    for single_tuple_2 in tup_2:
        if single_tuple_1[0] == single_tuple_2[0] and single_tuple_1[1] == single_tuple_2[1]:
            print single_tuple_1,single_tuple_2

编辑:

对于这种情况,假设顺序很重要并假设元组包含5个元素:

tup_1 = [(1,2,3,4,5),(4,5,6,7,8),(11,12,13,14,15)]
tup_2 = [(1,2,3,4,8),(4,5,1,7,8),(11,12,13,14,-5)]

我想找到在各自的前4个元素中相互交叉的元组。所以结果应该是:

[(1,2,3,4,5),(1,2,3,4,8),(11,12,13,14,15),(11,12,13,14,-5)]

如何更改代码以适应这种情况?

2 个答案:

答案 0 :(得分:3)

如果要返回所有“重叠”元组对,则无法比较所有对,即二次算法。但是你可以使用列表理解使代码更优雅,CIDR range用于组合,sum>>> tup_1 = [(1,2,3),(4,5,5),(7,8,9)] >>> tup_2 = [(4,5,6),(0,5,5),(9,8,7)] >>> [(a, b) for (a, b) in itertools.product(tup_1, tup_2) ... if sum(1 for ai, bi in zip(a, b) if ai == bi) >= 2] [((4, 5, 5), (4, 5, 6)), ((4, 5, 5), (0, 5, 5))] 用于比较:

a

注意:这将检查两个元组是否在至少两个位置具有相同的元素,即顺序问题。如果订单无关紧要,您可以将bset转换为(1,1,2)并检查其交叉点的大小,但这可能会因重复数字而失败,即{{1 }和(1,1,3)只是1而不是2

如果你只想匹配前两个,前两个和后两个元素,你可以在一致的分离中比较元组的切片:

>>> [(a, b) for (a, b) in itertools.product(tup_1, tup_2)
...         if a[:2] == b[:2]]
[((4, 5, 5), (4, 5, 6))]

>>> [(a, b) for (a, b) in itertools.product(tup_1, tup_2)
...         if a[:2] == b[:2] or a[-2:] == b[-2:]]
[((4, 5, 5), (4, 5, 6)), ((4, 5, 5), (0, 5, 5))]

答案 1 :(得分:2)

这是使用列表理解的一种方式。写入的逻辑检查至少2个元素的重叠。

请注意,如果没有重叠,您将留下tup_2的一个元素,但这可以简单地识别。

from itertools import chain

tup_1 = [(1,2,3),(4,5,5)]
tup_2 = [(4,5,6)]

y = sorted(tup_2[0])
res = [i for i in chain(tup_1, tup_2) if
       sum(i==j for i, j in zip(sorted(i), y)) > 1]

print res

[(4, 5, 5), (4, 5, 6)]