如何用列表推导式筛选包含指定元组列表全部元素的元组项?
元组列表的全匹配筛选问题
需求与问题
需要从元组列表x中筛选出同时包含目标元组列表y里所有元组的元素,但当前的列表推导式返回了重复且不符合预期的结果;单个元组作为匹配条件时能正常工作,不清楚多元组全匹配时的问题所在。
错误代码示例
x = [((1,1),(1,1),(1,2)), ((2,1),(1,3),(2,9)), ((2,1),(2,3),(2,9))] y = [(2,1),(1,3)] [i for i in x for k in y if k in i]
- 当前结果:
[((2, 1), (1, 3), (2, 9)), ((2, 1), (1, 3), (2, 9)), ((2, 1), (2, 3), (2, 9))] - 期望结果:
[((2, 1), (1, 3), (2, 9))]
单个元组匹配的正确示例
x = [((1,1),(1,1),(1,2)), ((2,1),(1,3),(2,9)), ((2,1),(2,3),(2,9))] y = (2,1) [i for i in x if y in i]
- 结果:
[((2, 1), (1, 3), (2, 9)), ((2, 1), (2, 3), (2, 9))]
问题原因
原来的列表推导式是双重循环逻辑:遍历x中的每个i,再遍历y中的每个k,只要k在i里就把i加入结果。这会导致两个问题:
- 只要满足
y中任意一个元组的匹配,i就会被加入,无法实现全匹配要求; - 当
i满足多个k的匹配时,会被重复添加到结果中。
解决方法
使用all()函数判断y中的所有元组是否都存在于i中,修改后的列表推导式如下:
x = [((1,1),(1,1),(1,2)), ((2,1),(1,3),(2,9)), ((2,1),(2,3),(2,9))] y = [(2,1),(1,3)] [i for i in x if all(k in i for k in y)]
结果验证
运行上述代码会得到预期结果:[((2, 1), (1, 3), (2, 9))]
原理说明
all()函数会遍历传入的迭代器(这里是k in i for k in y),只有当迭代器中所有条件都为True时,才返回True。也就是说,只有当i同时包含y中的每一个元组时,才会被筛选出来,完美满足全匹配的需求,同时不会出现重复元素。
内容的提问来源于stack exchange,提问作者Rajib Lochan Sarkar
相关产品推荐
相关产品推荐

