[英]Python - finding same elements in three lists (ignoring empty list)
如果有任何方法可以找到三个列表的共同元素,而忽略了三个列表中的空列表,那么我很想知道。 例如,我知道:
a = ['a', 'b', 'c', 'd']
b = ['a', 'v', 'd', 'g']
v = ['d']
>>> set(a).intersection(b, v)
{'d'}
但我想知道是否有办法做到这一点:
a = ['a', 'b', 'c', 'd']
b = ['a', 'v', 'd', 'g']
v = []
>>> comparison_method(a, b, v)
{'a', 'd'}
或者,如果3个列表中的2个为空,则只返回不是的列表。
使用过滤器然后设置交集:
set.intersection(*map(set,filter(None, [a,[],[]])))
O / P: set(['a', 'c', 'b', 'd'])
set.intersection(*map(set,filter(None, [a,b,[]])))
O / P: set(['a', 'd'])
set.intersection(*map(set,filter(None, [a,b,v])))
O / P: set(['d'])
正如jme所说,这是一个更好的解决方案
set.intersection(*(set(x) for x in [a, b, v] if x))
只过滤掉所有具有len
的列表(即长度不为零)并使用set-intersection
-
>>>a = ['a', 'b', 'c', 'd']
>>>b = ['a', 'v', 'd', 'g']
>>>v=[]
>>>input_list = [a,v,b]
>>>result = reduce(set.intersection,map(set,filter(len,input_list)))
>>>set(['a', 'd'])
当然,非常类似于此 ,但只是在运行intersection
测试之前过滤掉空列表:
def comparison_method(*args):
sets = [set(arg) for arg in args if arg]
if not sets:
return []
result = sets[0]
for s in sets[1:]:
result = result.intersection(s)
return result
a = ['a', 'b', 'c', 'd']
b = ['a', 'v', 'd', 'g']
v = []
>>> comparison_method(a, b, v)
{'a', 'd'}
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.